From 3ec6231425206d16f91e37b266cf4f65b06b883b Mon Sep 17 00:00:00 2001
From: insidestyles <insidestyles@gmail.com>
Date: Wed, 24 Apr 2019 19:23:10 +0000
Subject: [PATCH] [WIP] ping/close connection middleware services

---
 Resources/config/messenger.xml                   | 16 ++++++++++++++++
 .../DoctrineExtensionTest.php                    |  2 ++
 2 files changed, 18 insertions(+)

diff --git a/Resources/config/messenger.xml b/Resources/config/messenger.xml
index 9bc713672..6ba2ea126 100644
--- 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
index f7bf28624..aed5642a9 100644
--- a/Tests/DependencyInjection/DoctrineExtensionTest.php
+++ b/Tests/DependencyInjection/DoctrineExtensionTest.php
@@ -662,6 +662,8 @@ public function testMessengerIntegration()
         $extension->load([$config], $container);
 
         $this->assertNotNull($middlewarePrototype = $container->getDefinition('messenger.middleware.doctrine_transaction'));
+        $this->assertNotNull($middlewarePrototype = $container->getDefinition('messenger.middleware.doctrine_ping_connection'));
+        $this->assertNotNull($middlewarePrototype = $container->getDefinition('messenger.middleware.doctrine_close_connection'));
         $this->assertCount(1, $middlewarePrototype->getArguments());
     }