fix(coroutines): container modifier load method override fix for gene…

…rated production service container
pixelfederation · Feb 6, 2023 · 522e5ba · 522e5ba
1 parent 62afa05
commit 522e5ba
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/src/Bridge/Symfony/Container/ContainerModifier.php b/src/Bridge/Symfony/Container/ContainerModifier.php
@@ -4,6 +4,7 @@
 
 namespace K911\Swoole\Bridge\Symfony\Container;
 
+use Symfony\Component\DependencyInjection\Container;
 use ZEngine\Reflection\ReflectionClass;
 
 final class ContainerModifier
@@ -73,6 +74,36 @@ private static function overrideCreateProxy(BlockingContainer $container, Reflec
     }
 
     private static function overrideLoad(BlockingContainer $container, ReflectionClass $reflContainer): void
+    {
+        $loadRefl = $reflContainer->getMethod('load');
+
+        if (Container::class == $loadRefl->getDeclaringClass()->getName()) {
+            self::overrideOriginalContainerLoad($container, $reflContainer);
+
+            return;
+        }
+
+        self::overrideGeneratedLoad($container, $reflContainer);
+    }
+
+    private static function overrideOriginalContainerLoad(BlockingContainer $container, ReflectionClass $reflContainer): void
+    {
+        $loadRefl = $reflContainer->getMethod('load');
+        $reflContainer->addMethod('loadOriginal', $loadRefl->getClosure($container));
+        $loadRefl->redefine(function (string $file) {
+            $lock = self::$locking->acquire($file);
+
+            try {
+                $return = $this->loadOriginal($file);
+            } finally {
+                $lock->release();
+            }
+
+            return $return;
+        });
+    }
+
+    private static function overrideGeneratedLoad(BlockingContainer $container, ReflectionClass $reflContainer): void
     {
         $loadRefl = $reflContainer->getMethod('load');
         $reflContainer->addMethod('loadOriginal', $loadRefl->getClosure($container));