fix(coroutines): added entity manager resetter, which is decoupled fr…

…om the stability checker, so it can be overridden easily

src/Bridge/Doctrine/DoctrineProcessor.php

@@ -5,6 +5,7 @@
 namespace K911\Swoole\Bridge\Doctrine;
 
 use K911\Swoole\Bridge\Doctrine\DBAL\ConnectionKeepAliveResetter;
+use K911\Swoole\Bridge\Doctrine\ORM\EntityManagerResetter;
 use K911\Swoole\Bridge\Symfony\Bundle\DependencyInjection\CompilerPass\StatefulServices\CompileProcessor;
 use K911\Swoole\Bridge\Symfony\Bundle\DependencyInjection\CompilerPass\StatefulServices\Proxifier;
 use K911\Swoole\Bridge\Symfony\Bundle\DependencyInjection\ContainerConstants;
@@ -53,11 +54,12 @@ public function process(ContainerBuilder $container, Proxifier $proxifier): void
             throw new \UnexpectedValueException('Cannot obtain array of doctrine connections.');
         }
 
+        $this->createEntityManagerResetterDefinition($container);
         $this->prepareConnectionsForProxification($container, $connectionSvcIds);
 
         foreach ($entityManagers as $emName => $emSvcId) {
             $emDef = $container->findDefinition($emSvcId);
-            $tagParams = [];
+            $tagParams = ['resetter' => EntityManagerResetter::class];
             $limit = $this->getLimitFromEntityManagerConnection($container, $emDef);
 
             if (null !== $limit) {
@@ -72,6 +74,13 @@ public function process(ContainerBuilder $container, Proxifier $proxifier): void
         $this->fixDebugDataHolderResetter($container, $proxifier);
     }
 
+    private function createEntityManagerResetterDefinition(ContainerBuilder $container): void
+    {
+        $resetterDef = new Definition(EntityManagerResetter::class);
+        $resetterDef->setClass(EntityManagerResetter::class);
+        $container->setDefinition(EntityManagerResetter::class, $resetterDef);
+    }
+
     private function overrideEmConfigurator(ContainerBuilder $container, Definition $emDef): void
     {
         $configuratorCallback = $emDef->getConfigurator();

src/Bridge/Doctrine/ORM/EntityManagerResetter.php

@@ -0,0 +1,20 @@
+<?php
+
+declare(strict_types=1);
+
+namespace K911\Swoole\Bridge\Doctrine\ORM;
+
+use Doctrine\Persistence\ObjectManager;
+use K911\Swoole\Bridge\Symfony\Container\Resetter;
+
+final class EntityManagerResetter implements Resetter
+{
+    public function reset(object $service): void
+    {
+        if (!$service instanceof ObjectManager) {
+            throw new \UnexpectedValueException(\sprintf('Invalid service - expected %s, got %s', ObjectManager::class, \get_class($service)));
+        }
+
+        $service->clear();
+    }
+}

src/Bridge/Doctrine/ORM/EntityManagerStabilityChecker.php

@@ -5,18 +5,15 @@
 namespace K911\Swoole\Bridge\Doctrine\ORM;
 
 use Doctrine\ORM\EntityManager;
+use Doctrine\ORM\EntityManagerInterface;
 use K911\Swoole\Bridge\Symfony\Container\StabilityChecker;
 
 final class EntityManagerStabilityChecker implements StabilityChecker
 {
     public function isStable(object $service): bool
     {
-        if (!$service instanceof EntityManager) {
-            throw new \UnexpectedValueException(\sprintf('Invalid service - expected %s, got %s', EntityManager::class, \get_class($service)));
-        }
-
-        if ($service->isOpen()) {
-            $service->clear(); // clear the em when not used any more in context, so RAM can be freed early
+        if (!$service instanceof EntityManagerInterface) {
+            throw new \UnexpectedValueException(\sprintf('Invalid service - expected %s, got %s', EntityManagerInterface::class, \get_class($service)));
         }
 
         return $service->isOpen();