In long-running processes, allows checking the database connection by…

… pinging it, thus avoiding exceptions. Additionally, it provides the flexibility to configure the frequency of the check to be performed in the case of long-running processes.

src/Connection.php

@@ -29,6 +29,7 @@
 use Doctrine\Deprecations\Deprecation;
 use LogicException;
 use SensitiveParameter;
+use Symfony\Bridge\Doctrine\ConnectionLossAwareHandler;
 use Throwable;
 use Traversable;
 
@@ -166,6 +167,12 @@ class Connection
 
     private SchemaManagerFactory $schemaManagerFactory;
 
+    private static bool $isChecking = false;
+
+    private static ?int $lastCheckedAt = null;
+
+    private static ?int $checkFrequency = null;
+
     /**
      * Initializes a new instance of the Connection class.
      *
@@ -186,6 +193,7 @@ public function __construct(
         ?Configuration $config = null,
         ?EventManager $eventManager = null
     ) {
+        self::$checkFrequency = $params['check_connection_frequency'] ?? null;
         $this->_driver = $driver;
         $this->params  = $params;
 
@@ -371,6 +379,21 @@ public function connect()
         );
 
         if ($this->_conn !== null) {
+            if (! class_exists(ConnectionLossAwareHandler::class) || null === self::$checkFrequency) {
+                return false;
+            }
+
+            if (! self::$isChecking) {
+                if (null === self::$lastCheckedAt || time() - self::$lastCheckedAt >= self::$checkFrequency) {
+                    self::$isChecking = true;
+
+                    ConnectionLossAwareHandler::reconnectOnFailure($this);
+
+                    self::$lastCheckedAt = time();
+                    self::$isChecking = false;
+                }
+            }
+
             return false;
         }