Fix recursive function calls using set strategy and a repositoryMethod

doctrine · Aug 28, 2015 · c40709e · c40709e
1 parent f88ad5c
commit c40709e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/lib/Doctrine/ODM/MongoDB/Persisters/DocumentPersister.php b/lib/Doctrine/ODM/MongoDB/Persisters/DocumentPersister.php
@@ -778,9 +778,14 @@ public function createReferenceManyInverseSideQuery(PersistentCollection $collec
     private function loadReferenceManyWithRepositoryMethod(PersistentCollection $collection)
     {
         $cursor = $this->createReferenceManyWithRepositoryMethodCursor($collection);
+        $mapping = $collection->getMapping();        
         $documents = $cursor->toArray(false);
-        foreach ($documents as $document) {
-            $collection->add($document);
+        foreach ($documents as $key => $obj) {
+            if (CollectionHelper::isHash($mapping['strategy'])) {
+                $collection->set($key, $obj);
+            } else {
+                $collection->add($obj);
+            }
         }
     }