Allow the eager loader to take an array of models as well as a collec…

…tion.
Gignite · Jul 19, 2012 · 34659c8 · 34659c8
1 parent b743314
commit 34659c8
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/classes/TheCure/Optimisers/EagerLoader.php b/classes/TheCure/Optimisers/EagerLoader.php
@@ -80,7 +80,7 @@ private function appendRelationAndIDs(
 	}
 
 	private function getRelationshipsAndIDsFromCollection(
-		ModelCollection $collection)
+		\TheCure\Collections\Collection $collection)
 	{
 		$attributes = array();
 		$accessor = new TransferObjectAccessor;
@@ -142,9 +142,15 @@ private function getRelationshipsAndIDsFromCollection(
 	 */
 	public function loadRelations(
 		Container $container,
-		ModelCollection $collection,
+		$collection,
 		array $relationNames = array())
 	{
+
+		if (is_array($collection))
+		{
+			$collection = new \TheCure\Collections\Collection($collection);
+		}
+
 		$ids = $this->getRelationshipsAndIDsFromCollection($collection);
 
 		$eagerLoaded = array();