Make manager registry resettable

doctrine · Dec 17, 2019 · d798dee · d798dee
1 parent b612f15
commit d798dee
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/Registry.php b/Registry.php
@@ -7,11 +7,12 @@
 use Psr\Container\ContainerInterface;
 use Symfony\Bridge\Doctrine\ManagerRegistry;
 use Symfony\Bridge\Doctrine\RegistryInterface;
+use Symfony\Contracts\Service\ResetInterface;
 
 /**
  * References all Doctrine connections and entity managers in a given Container.
  */
-class Registry extends ManagerRegistry implements RegistryInterface
+class Registry extends ManagerRegistry implements RegistryInterface, ResetInterface
 {
     /**
      * @param string[] $connections
@@ -164,4 +165,11 @@ public function getEntityManagerForClass($class)
 
         return $this->getManagerForClass($class);
     }
+
+    public function reset() : void
+    {
+        foreach ($this->getManagerNames() as $managerName) {
+            $this->resetManager($managerName);
+        }
+    }
 }
diff --git a/Resources/config/dbal.xml b/Resources/config/dbal.xml
@@ -63,6 +63,7 @@
             <argument>%doctrine.entity_managers%</argument>
             <argument>%doctrine.default_connection%</argument>
             <argument>%doctrine.default_entity_manager%</argument>
+            <tag name="kernel.reset" method="reset" />
         </service>
         <service id="Doctrine\Persistence\ManagerRegistry" alias="doctrine" public="false" />
         <service id="Doctrine\Common\Persistence\ManagerRegistry" alias="doctrine" public="false" />

diff --git a/composer.json b/composer.json
@@ -32,6 +32,7 @@
         "jdorn/sql-formatter": "^1.2.16",
         "symfony/cache": "^3.4.30|^4.3.3",
         "symfony/config": "^3.4.30|^4.3.3",
+        "symfony/service-contracts": "^1.1.1",
         "symfony/console": "^3.4.30|^4.3.3",
         "symfony/dependency-injection": "^3.4.30|^4.3.3",
         "symfony/doctrine-bridge": "^3.4.30|^4.3.3",