long running process doctrine connection listener

doctrine · Apr 16, 2024 · 36a7031 · 36a7031
1 parent f966567
commit 36a7031
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/DependencyInjection/Configuration.php b/DependencyInjection/Configuration.php
@@ -114,6 +114,8 @@ private function addDbalSection(ArrayNodeDefinition $node): void
                 ->children()
                     ->scalarNode('default_connection')->end()
                 ->end()
+                ->children()
+                    ->integerNode('check_connection_timing')->end()
                 ->fixXmlConfig('type')
                 ->children()
                     ->arrayNode('types')
@@ -223,6 +225,7 @@ private function getDbalConnectionsNode(): ArrayNodeDefinition
                 ->scalarNode('server_version')->end()
                 ->scalarNode('driver_class')->end()
                 ->scalarNode('wrapper_class')->end()
+                ->scalarNode('check_connection_frequency')->defaultValue(30)->end()
                 ->booleanNode('keep_slave')
                     ->setDeprecated(
                         'doctrine/doctrine-bundle',

diff --git a/DependencyInjection/DoctrineExtension.php b/DependencyInjection/DoctrineExtension.php
@@ -174,6 +174,12 @@ protected function dbalLoad(array $config, ContainerBuilder $container)
             $config['default_connection'] = reset($keys);
         }
 
+        if (! $container->hasParameter('kernel.runtime_mode') && ! $container->hasParameter('kernel.runtime_mode.worker')) {
+            $container->removeDefinition('doctrine.listeners.doctrine_connection_listener');
+        } else {
+            $container->getDefinition('doctrine.orm.listeners.doctrine_connection_listener')->setArgument(1, $config['check_connection_timing']);
+        }
+
         $this->defaultConnection = $config['default_connection'];
 
         $container->setAlias('database_connection', sprintf('doctrine.dbal.%s_connection', $this->defaultConnection));
@@ -252,6 +258,12 @@ protected function loadDbalConnection($name, array $connection, ContainerBuilder
             $connection['profiling_collect_schema_errors'],
         );
 
+        if (isset($connection['check_connection_timing'])) {
+            if ($container->hasParameter('kernel.runtime_mode') || $container->hasParameter('kernel.runtime_mode.worker')) {
+                $configuration->addMethodCall('setCheckConnectionTiming', [$connection['check_connection_timing']]);
+            }
+        }
+
         if (isset($connection['auto_commit'])) {
             $configuration->addMethodCall('setAutoCommit', [$connection['auto_commit']]);
         }

diff --git a/Resources/config/dbal.xml b/Resources/config/dbal.xml
@@ -101,6 +101,12 @@
             <tag name="controller.service_arguments" />
         </service>
 
+        <service id="doctrine.listeners.doctrine_connection_listener" class="Symfony\Bridge\Doctrine\Listener\ConnectionListener">
+            <argument type="service" id="service_container" />
+            <argument /> <!-- check timing -->
+            <tag name="kernel.event_subscriber" />
+        </service>
+
         <service id="doctrine.dbal.default_schema_manager_factory" class="Doctrine\DBAL\Schema\DefaultSchemaManagerFactory" />
         <service id="doctrine.dbal.legacy_schema_manager_factory" class="Doctrine\DBAL\Schema\LegacySchemaManagerFactory" />