Add a find version to allow to override it in future for performance …

…optimizations
tomaskocfelda · Jul 23, 2012 · 1cda09d · 1cda09d
1 parent f1ac389
commit 1cda09d
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/src/DoctrineORMModule/Hydrator/DoctrineEntity.php b/src/DoctrineORMModule/Hydrator/DoctrineEntity.php
@@ -105,11 +105,11 @@ public function hydrate(array $data, $object)
     protected function toOne($valueOrObject, $target)
     {
         if (is_numeric($valueOrObject)) {
-            return $this->objectManager->find($target, $valueOrObject);
+            return $this->find($target, $valueOrObject);
         }
 
         $identifiers = $this->metadata->getIdentifierValues($valueOrObject);
-        return $this->objectManager->find($target, $identifiers);
+        return $this->find($target, $identifiers);
     }
 
     /**
@@ -126,14 +126,24 @@ protected function toMany($valueOrObject, $target)
         $values = array();
         foreach($valueOrObject as $value) {
             if (is_numeric($value)) {
-                $values[] = $this->objectManager->find($target, $value);
+                $values[] = $this->find($target, $value);
                 continue;
             }
 
             $identifiers = $this->metadata->getIdentifierValues($valueOrObject);
-            $values[] = $this->objectManager->find($target, $identifiers);
+            $values[] = $this->find($target, $identifiers);
         }
 
         return $values;
     }
+
+    /**
+     * @param  string    $target
+     * @param  int|array $identifiers
+     * @return object
+     */
+    protected function find($target, $identifiers)
+    {
+        return $this->objectManager->find($target, $identifiers);
+    }
 }