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.

Author: alli83

src/Connection.php

@@ -42,6 +42,7 @@
 use function is_string;
 use function key;
 use function sprintf;
+use function time;
 
 /**
  * A database abstraction-level connection that implements features like transaction isolation levels,
@@ -99,6 +100,12 @@ class Connection implements ServerVersionProvider
 
     private SchemaManagerFactory $schemaManagerFactory;
 
+    private bool $isChecking = false;
+
+    private ?int $lastCheckedAt = null;
+
+    private ?int $heartbeat;
+
     /**
      * Initializes a new instance of the Connection class.
      *
@@ -115,6 +122,8 @@ public function __construct(
         protected Driver $driver,
         ?Configuration $config = null,
     ) {
+        $this->heartbeat = $params['check_connection_frequency'] ?? null;
+
         $this->_config    = $config ?? new Configuration();
         $this->params     = $params;
         $this->autoCommit = $this->_config->getAutoCommit();
@@ -210,7 +219,23 @@ public function createExpressionBuilder(): ExpressionBuilder
     protected function connect(): DriverConnection
     {
         if ($this->_conn !== null) {
-            return $this->_conn;
+            $isTimeToCheck = $this->lastCheckedAt === null || time() - $this->lastCheckedAt >= $this->heartbeat;
+            $noCheckNeeded = $this->heartbeat === null || $this->isChecking;
+
+            if ($noCheckNeeded || ! $isTimeToCheck) {
+                return $this->_conn;
+            }
+
+            $this->isChecking = true;
+
+            $isAvailable = $this->reconnectOnFailure();
+
+            $this->lastCheckedAt = time();
+            $this->isChecking    = false;
+
+            if ($isAvailable) {
+                return $this->_conn;
+            }
         }
 
         try {
@@ -1371,4 +1396,15 @@ private function handleDriverException(
 
         return $exception;
     }
+
+    private function reconnectOnFailure(): bool
+    {
+        try {
+            $this->executeQuery($this->getDatabasePlatform($this)->getDummySelectSQL());
+
+            return true;
+        } catch (ConnectionLost) {
+            return false;
+        }
+    }
 }