Allow change the hydratation mode on ResultSet

Kdyby · Oct 20, 2013 · dfc5b0e · dfc5b0e
1 parent 2c06421
commit dfc5b0e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/Kdyby/Doctrine/ResultSet.php b/src/Kdyby/Doctrine/ResultSet.php
@@ -152,12 +152,15 @@ public function getTotalCount()
 
 
 	/**
-	 * @throws \Kdyby\Doctrine\QueryException
+	 * @param int $hydrationMode
+	 * @throws QueryException
 	 * @return \ArrayIterator
 	 */
-	public function getIterator()
+	public function getIterator($hydrationMode = ORM\AbstractQuery::HYDRATE_OBJECT)
 	{
 		try {
+			$this->query->setHydrationMode($hydrationMode);
+
 			if ($this->query->getMaxResults() > 0 || $this->query->getFirstResult() > 0) {
 				return $this->getPaginatedQuery()->getIterator();
 			}