Introduce proxy class name resolvers (#145)

doctrine · Feb 17, 2021 · 12a05f8 · 12a05f8
1 parent a18b272
commit 12a05f8
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 5 deletions.
diff --git a/lib/Doctrine/Persistence/Mapping/AbstractClassMetadataFactory.php b/lib/Doctrine/Persistence/Mapping/AbstractClassMetadataFactory.php
@@ -58,6 +58,9 @@ abstract class AbstractClassMetadataFactory implements ClassMetadataFactory
     /** @var ReflectionService|null */
     private $reflectionService = null;
 
+    /** @var ProxyClassNameResolver|null */
+    private $proxyClassNameResolver = null;
+
     /**
      * Sets the cache driver used by the factory to cache ClassMetadata instances.
      *
@@ -140,6 +143,11 @@ public function getAllMetadata()
         return $metadata;
     }
 
+    public function setProxyClassNameResolver(ProxyClassNameResolver $resolver): void
+    {
+        $this->proxyClassNameResolver = $resolver;
+    }
+
     /**
      * Lazy initialization of this stuff, especially the metadata driver,
      * since these are not needed at all when a metadata cache is active.
@@ -467,12 +475,26 @@ protected function getCacheKey(string $realClassName): string
      */
     private function getRealClass(string $class): string
     {
-        $pos = strrpos($class, '\\' . Proxy::MARKER . '\\');
-
-        if ($pos === false) {
-            return $class;
+        if ($this->proxyClassNameResolver === null) {
+            $this->createDefaultProxyClassNameResolver();
         }
 
-        return substr($class, $pos + Proxy::MARKER_LENGTH + 2);
+        return $this->proxyClassNameResolver->resolveClassName($class);
+    }
+
+    private function createDefaultProxyClassNameResolver(): void
+    {
+        $this->proxyClassNameResolver = new class implements ProxyClassNameResolver {
+            public function resolveClassName(string $className): string
+            {
+                $pos = strrpos($className, '\\' . Proxy::MARKER . '\\');
+
+                if ($pos === false) {
+                    return $className;
+                }
+
+                return substr($className, $pos + Proxy::MARKER_LENGTH + 2);
+            }
+        };
     }
 }
diff --git a/lib/Doctrine/Persistence/Mapping/ProxyClassNameResolver.php b/lib/Doctrine/Persistence/Mapping/ProxyClassNameResolver.php
@@ -0,0 +1,12 @@
+<?php
+
+namespace Doctrine\Persistence\Mapping;
+
+interface ProxyClassNameResolver
+{
+    /**
+     * @psalm-param class-string $className
+     * @psalm-return class-string
+     */
+    public function resolveClassName(string $className): string;
+}