Merge pull request #534 from utopia-php/feat-connection-checker

Add custom connection checker

Author: abnegate

composer.lock

@@ -0,0 +1,38 @@
+<?php
+
+namespace Utopia\Database;
+
+use Swoole\Database\DetectsLostConnections;
+
+class Connection
+{
+    /**
+     * @var array<string>
+     */
+    protected static array $errors = [
+        'Max connect timeout reached'
+    ];
+
+    /**
+     * Check if the given throwable was caused by a database connection error.
+     *
+     * @param \Throwable $e
+     * @return bool
+     */
+    public static function hasError(\Throwable $e): bool
+    {
+        /** @phpstan-ignore-next-line can't find static method */
+        if (DetectsLostConnections::causedByLostConnection($e)) {
+            return true;
+        }
+
+        $message = $e->getMessage();
+        foreach (static::$errors as $needle) {
+            if (\mb_strpos($message, $needle) !== false) {
+                return true;
+            }
+        }
+
+        return false;
+    }
+}

src/Database/Connection.php

@@ -2,7 +2,6 @@
 
 namespace Utopia\Database;
 
-use Swoole\Database\DetectsLostConnections;
 use Utopia\CLI\Console;
 
 /**
@@ -45,8 +44,7 @@ public function __call(string $method, array $args): mixed
         try {
             return $this->pdo->{$method}(...$args);
         } catch (\Throwable $e) {
-            /** @phpstan-ignore-next-line can't find static method */
-            if (DetectsLostConnections::causedByLostConnection($e)) {
+            if (Connection::hasError($e)) {
                 Console::warning('[Database] Lost connection detected. Reconnecting...');
                 $this->reconnect();
                 return $this->pdo->{$method}(...$args);