Persisting and removing Traverable collections

doctrine · Apr 9, 2014 · f10167b · f10167b
1 parent 17344d1
commit f10167b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/Doctrine/Search/SearchManager.php b/lib/Doctrine/Search/SearchManager.php
@@ -205,7 +205,9 @@ public function find($entityName, $id)
      */
     public function persist($objects)
     {
-        if(!is_array($objects)) $objects = array($objects);
+        if(!is_array($objects) && !$objects instanceof \Traversable) {
+            $objects = array($objects);
+        }
         foreach($objects as $object) {
             if (!is_object($object)) {
                 throw new UnexpectedTypeException($object, 'object');
@@ -223,7 +225,9 @@ public function persist($objects)
      */
     public function remove($objects)
     {
-        if(!is_array($objects)) $objects = array($objects);
+        if(!is_array($objects) && !$objects instanceof \Traversable) {
+            $objects = array($objects);
+        }
         foreach($objects as $object) {
             if (!is_object($object)) {
                 throw new UnexpectedTypeException($object, 'object');