symfony
diff --git a/‎src/Symfony/Bridge/Doctrine/Middleware/DoctrineConnectionDriver.php
Copy file name to clipboard
+89Lines changed: 89 additions & 0 deletions b/‎src/Symfony/Bridge/Doctrine/Middleware/DoctrineConnectionDriver.php
Copy file name to clipboard
+89Lines changed: 89 additions & 0 deletions
diff --git a/‎src/Symfony/Bridge/Doctrine/Middleware/DoctrineConnectionMiddleware.php
Copy file name to clipboard
+32Lines changed: 32 additions & 0 deletions b/‎src/Symfony/Bridge/Doctrine/Middleware/DoctrineConnectionMiddleware.php
Copy file name to clipboard
+32Lines changed: 32 additions & 0 deletions
@@ -0,0 +1,89 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Bridge\Doctrine\Middleware;
+
+use Doctrine\DBAL\Connection;
+use Doctrine\DBAL\Driver;
+use Doctrine\DBAL\Driver\Connection as DriverConnection;
+use Doctrine\DBAL\Driver\Middleware\AbstractDriverMiddleware;
+use Doctrine\DBAL\Exception as DBALException;
+use Doctrine\ORM\EntityManagerInterface;
+use Doctrine\Persistence\ManagerRegistry;
+use Symfony\Component\DependencyInjection\ContainerInterface;
+
+class DoctrineConnectionDriver extends AbstractDriverMiddleware
+{
+    public function __construct(private Driver $driver, private readonly ManagerRegistry $managerRegistry, private readonly ContainerInterface $container)
+    {
+        parent::__construct($driver);
+    }
+
+    public function connect(array $params): DriverConnection
+    {
+        $connectionServices = $this->managerRegistry->getConnectionNames();
+
+        foreach ($connectionServices as $connectionServiceName) {
+            if (!$this->container->initialized($connectionServiceName)) {
+                continue;
+            }
+
+            $connection = $this->container->get($connectionServiceName);
+
+            if (!$connection instanceof Connection) {
+                continue;
+            }
+
+            if ($connection->isConnected()) {
+                $this->pingConnection($connection);
+            }
+
+            $managerNames = $this->managerRegistry->getManagerNames();
+
+            foreach ($managerNames as $managerName) {
+                if (!$this->container->initialized($managerName)) {
+                    continue;
+                }
+
+                $manager = $this->container->get($managerName);
+
+                if (!$manager instanceof EntityManagerInterface) {
+                    continue;
+                }
+
+                if (!$manager->isOpen()) {
+                    $this->managerRegistry->resetManager($managerName);
+                }
+            }
+        }
+
+        return parent::connect($params);
+    }
+
+    private function pingConnection(Connection $connection): void
+    {
+        try {
+            $this->executeDummySql($connection);
+        } catch (DBALException $e) {
+            $connection->close();
+            // Attempt to reestablish the lazy connection by sending another query.
+            $this->executeDummySql($connection);
+        }
+    }
+
+    /**
+     * @throws DBALException
+     */
+    private function executeDummySql(Connection $connection): void
+    {
+        $connection->executeQuery($connection->getDatabasePlatform()->getDummySelectSQL());
+    }
+}
@@ -0,0 +1,32 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Bridge\Doctrine\Middleware;
+
+use Doctrine\DBAL\Driver;
+use Doctrine\DBAL\Driver\Middleware;
+use Doctrine\Persistence\ManagerRegistry;
+use Symfony\Component\DependencyInjection\ContainerInterface;
+
+/**
+ * Based on https://github.com/Baldinof/roadrunner-bundle/blob/3.x/src/Integration/Doctrine/DoctrineORMMiddleware.php.
+ */
+class DoctrineConnectionMiddleware implements Middleware
+{
+    public function __construct(private ManagerRegistry $managerRegistry, private ContainerInterface $container)
+    {
+    }
+
+    public function wrap(Driver $driver): Driver
+    {
+        return new DoctrineConnectionDriver($driver, $this->managerRegistry, $this->container);
+    }
+}