Attempt to reconnect after goneaway error

Author: peldax

composer.json

@@ -15,6 +15,7 @@
     "infinityloop-dev/utils": "^2.1",
     "nette/utils": "^3.0",
     "nette/database": "^3.1",
+    "nette/di": "^3.0",
     "nette/robot-loader": "^3.3",
     "symfony/console": "^5.2"
   },
@@ -43,5 +44,11 @@
     "psr-4": {
       "CoolBeans\\Tests\\": "tests/"
     }
+  },
+  "config": {
+    "allow-plugins": {
+      "dealerdirect/phpcodesniffer-composer-installer": true,
+      "infection/extension-installer": true
+    }
   }
 }

src/Bridge/Nette/Connection.php

@@ -0,0 +1,23 @@
+<?php
+
+declare(strict_types = 1);
+
+namespace CoolBeans\Bridge\Nette;
+
+final class Connection extends \Nette\Database\Connection
+{
+    public function query(string $sql, ...$params): \Nette\Database\ResultSet
+    {
+        try {
+            return parent::query($sql, $params);
+        } catch (\PDOException $e) {
+            if ($e->getCode() === 'HY000' && \str_contains($e->getMessage(), 'gone away')) {
+                $this->reconnect();
+
+                return parent::query($sql, $params); // retry
+            }
+
+            throw $e;
+        }
+    }
+}

src/Bridge/Nette/CoolbeansExtension.php

@@ -0,0 +1,19 @@
+<?php
+
+declare(strict_types = 1);
+
+namespace CoolBeans\Bridge\Nette;
+
+final class CoolbeansExtension extends \Nette\DI\CompilerExtension
+{
+	public function beforeCompile() : void
+	{
+		$builder = $this->getContainerBuilder();
+
+        foreach ($builder->findByType(\Nette\Database\Connection::class) as $definition) {
+            \assert($definition instanceof \Nette\DI\Definitions\ServiceDefinition);
+
+            $definition->setFactory(new \Nette\DI\Definitions\Statement(Connection::class, $definition->getFactory()->arguments));
+        }
+	}
+}