long running process doctrine connection listener

DependencyInjection/DoctrineExtension.php

@@ -532,6 +532,10 @@ protected function ormLoad(array $config, ContainerBuilder $container)
             $container->removeDefinition('doctrine.mapping_import_command');
         }
 
+        if (!$container->hasParameter('kernel.runtime_mode') || !$container->hasParameter('kernel.runtime_mode.worker')) {
+            $container->removeDefinition('doctrine.orm.listeners.doctrine_connection_listener');
+        }
+
         $entityManagers = [];
         foreach (array_keys($config['entity_managers']) as $name) {
             /** @psalm-suppress InvalidArrayOffset */

Resources/config/orm.xml

@@ -273,5 +273,11 @@
 
             <tag name="console.command" command="doctrine:mapping:import" />
         </service>
+        <service id="doctrine.orm.listeners.doctrine_connection_listener" class="Symfony\Bridge\Doctrine\Listener\DoctrineConnectionListener">
+            <argument type="service" id="doctrine" />
+            <argument type="service" id="service_container" />
+            <argument type="service" id="event_dispatcher" />
+            <tag name="kernel.event_subscriber" />
+        </service>
     </services>
 </container>

Tests/DoctrineConnectionListenerTest.php

@@ -0,0 +1,139 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Doctrine\Bundle\DoctrineBundle\Tests;
+
+use Doctrine\DBAL\Connection;
+use Doctrine\DBAL\Exception as DBALException;
+use Doctrine\DBAL\Platforms\AbstractPlatform;
+use Doctrine\ORM\EntityManagerInterface;
+use Doctrine\Persistence\ManagerRegistry;
+use PHPUnit\Framework\TestCase;
+use ProxyManager\Proxy\LazyLoadingInterface;
+use Symfony\Bridge\Doctrine\Event\ForceKernelRebootEvent;
+use Symfony\Bridge\Doctrine\Listener\DoctrineConnectionListener;
+use Symfony\Component\DependencyInjection\Container;
+use Symfony\Component\EventDispatcher\EventDispatcher;
+use Symfony\Component\HttpKernel\Event\RequestEvent;
+
+/**
+ * Based on https://github.com/Baldinof/roadrunner-bundle/blob/3.x/src/Integration/Doctrine/DoctrineORMMiddleware.php
+ */
+class DoctrineConnectionListenerTest extends TestCase
+{
+    public const CONNECTION_NAME = 'doctrine.connection';
+    public const MANAGER_NAME    = 'doctrine.manager';
+
+    private $managerRegistryMock;
+    private $connectionMock;
+    private $container;
+    private $dispatcher;
+    private $listener;
+    private $event;
+
+    public function setUp(): void
+    {
+        $platform = $this->createMock(AbstractPlatform::class);
+        $platform->method('getDummySelectSQL')->willReturn('SELECT 1');
+
+        $this->managerRegistryMock = $this->createMock(ManagerRegistry::class);
+        $this->connectionMock = $this->createMock(Connection::class);
+        $this->connectionMock->method('getDatabasePlatform')->willReturn($platform);
+
+        $this->container = new Container();
+        $this->container->set(self::CONNECTION_NAME, $this->connectionMock);
+
+        $this->managerRegistryMock->method('getConnectionNames')->willReturn([self::CONNECTION_NAME]);
+        $this->managerRegistryMock->method('getManagerNames')->willReturn([self::MANAGER_NAME]);
+
+        $this->dispatcher = new EventDispatcher();
+        $this->event = $this->createMock(RequestEvent::class);
+        $this->listener = new DoctrineConnectionListener($this->managerRegistryMock, $this->container, $this->dispatcher);
+    }
+
+    public function testSkipNotInitializedConnections()
+    {
+        $this->container->set(self::CONNECTION_NAME, null);
+
+        $this->connectionMock->expects($this->never())->method('isConnected');
+        $this->connectionMock->expects($this->never())->method('executeQuery');
+        $this->connectionMock->expects($this->never())->method('close');
+        $this->connectionMock->expects($this->never())->method('connect');
+
+        $this->listener->onKernelRequest($this->event);
+    }
+
+    public function testSkipWhenNotConnected(): void
+    {
+        $this->connectionMock->method('isConnected')->willReturn(false);
+        $this->connectionMock->expects($this->never())->method('executeQuery');
+        $this->connectionMock->expects($this->never())->method('close');
+        $this->connectionMock->expects($this->never())->method('connect');
+
+        $this->listener->onKernelRequest($this->event);
+    }
+
+    public function testItClosesNotPingableConnection(): void
+    {
+        $this->connectionMock->expects($this->once())->method('executeQuery')->will($this->throwException(new DBALException()));
+
+        $this->connectionMock->method('isConnected')->willReturn(true);
+        $this->connectionMock->expects($this->once())->method('close');
+        $this->connectionMock->expects($this->never())->method('connect');
+
+        $this->listener->onKernelRequest($this->event);
+    }
+
+    public function testItDoesNotClosePingableConnection(): void
+    {
+        $this->connectionMock->expects($this->once())->method('executeQuery');
+        $this->connectionMock->method('isConnected')->willReturn(true);
+        $this->connectionMock->expects($this->never())->method('close');
+        $this->connectionMock->expects($this->never())->method('connect');
+
+        $this->listener->onKernelRequest($this->event);
+    }
+
+    public function testItForcesRebootOnClosedManagerWhenMissingProxySupport()
+    {
+        $rebootForced = false;
+
+        $this->dispatcher->addListener(ForceKernelRebootEvent::class, function () use (&$rebootForced) {
+            $rebootForced = true;
+        });
+
+        $manager = $this->createMock(EntityManagerInterface::class);
+        $manager->expects($this->once())->method('isOpen')->willReturn(false);
+
+        $this->container->set(self::MANAGER_NAME, $manager);
+
+        $this->listener->onKernelRequest($this->event);
+
+        $this->assertTrue($rebootForced, 'A ForceKernelRebootEvent should have been dispatched');
+    }
+
+    public function testItSkipsLazyEntityManagers()
+    {
+        $rebootForced = false;
+
+        $this->dispatcher->addListener(ForceKernelRebootEvent::class, function () use (&$rebootForced) {
+            $rebootForced = true;
+        });
+
+        $manager = $this->createMock(LazyEntityManager::class);
+        $manager->expects($this->never())->method('isOpen');
+
+        $this->container->set(self::MANAGER_NAME, $manager);
+
+        $this->listener->onKernelRequest($this->event);
+        $this->assertFalse($rebootForced, 'A ForceKernelRebootEvent should not have been dispatched');
+    }
+}
+
+/**
+ * @internal Allow mock of multiple interface
+ */
+interface LazyEntityManager extends EntityManagerInterface, LazyLoadingInterface
+{
+}