Fix potential issue with proxies

algolia · Dec 7, 2015 · 642cab5 · 642cab5
1 parent 9c25661
commit 642cab5
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/Indexer/Indexer.php b/Indexer/Indexer.php
@@ -113,12 +113,17 @@ public function getMetaDataLoader()
         return new AnnotationLoader();
     }
 
+    private function removeProxy($class)
+    {
+        /* Avoid proxy class form symfony */
+        return str_replace("Proxies\\__CG__\\", "", $class);
+    }
+
     private function get_class($entity)
     {
         $class = get_class($entity);
 
-        /* Avoid proxy class form symfony */
-        $class = str_replace("Proxies\\__CG__\\", "", $class);
+        $class = $this->removeProxy($class);
 
         return $class;
     }
@@ -140,6 +145,7 @@ public function discoverEntity($entity_or_class, EntityManager $em)
             $class = $this->get_class($entity);
         } else {
             $class = $em->getRepository($entity_or_class)->getClassName();
+            $class = $this->removeProxy($class);
             $reflClass = new \ReflectionClass($class);
 
             if ($reflClass->isAbstract()) {

diff --git a/Mapping/Loader/AnnotationLoader.php b/Mapping/Loader/AnnotationLoader.php
@@ -40,13 +40,21 @@ protected function getAnnotationReader()
         return self::$annotationReader;
     }
 
+    private function removeProxy($class)
+    {
+        /* Avoid proxy class form symfony */
+        return str_replace("Proxies\\__CG__\\", "", $class);
+    }
+
     /**
      * @return Description
      */
     public function getMetaData($entity, EntityManager $em)
     {
         $class = get_class($entity);
 
+        $class = $this->removeProxy($class);
+
         $description = new Description($class);
 
         $refl = new \ReflectionClass($entity);