ping/close connection middleware services

doctrine · Apr 28, 2019 · c0020d5 · c0020d5
1 parent 2fb807c
commit c0020d5
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/Resources/config/messenger.xml b/Resources/config/messenger.xml
@@ -12,6 +12,22 @@
             <argument type="service" id="doctrine" />
         </service>
 
+        <!--
+        The following service isn't prefixed by the "doctrine.orm" namespace in order for end-users to just use
+        the "doctrine_ping_connection" shortcut in message buses middleware config
+        -->
+        <service id="messenger.middleware.doctrine_ping_connection" class="Symfony\Bridge\Doctrine\Messenger\DoctrinePingConnectionMiddleware" abstract="true" public="false">
+            <argument type="service" id="doctrine" />
+        </service>
+
+        <!--
+        The following service isn't prefixed by the "doctrine.orm" namespace in order for end-users to just use
+        the "doctrine_close_connection" shortcut in message buses middleware config
+        -->
+        <service id="messenger.middleware.doctrine_close_connection" class="Symfony\Bridge\Doctrine\Messenger\DoctrineCloseConnectionMiddleware" abstract="true" public="false">
+            <argument type="service" id="doctrine" />
+        </service>
+
         <!--
         The following service isn't tagged as transport factory because the class may not exist.
         The tag is added conditionally in DoctrineExtension.

diff --git a/Tests/DependencyInjection/DoctrineExtensionTest.php b/Tests/DependencyInjection/DoctrineExtensionTest.php
@@ -663,6 +663,10 @@ public function testMessengerIntegration()
 
         $this->assertNotNull($middlewarePrototype = $container->getDefinition('messenger.middleware.doctrine_transaction'));
         $this->assertCount(1, $middlewarePrototype->getArguments());
+        $this->assertNotNull($middlewarePrototype = $container->getDefinition('messenger.middleware.doctrine_ping_connection'));
+        $this->assertCount(1, $middlewarePrototype->getArguments());
+        $this->assertNotNull($middlewarePrototype = $container->getDefinition('messenger.middleware.doctrine_close_connection'));
+        $this->assertCount(1, $middlewarePrototype->getArguments());
     }
 
     /**