Merge pull request #21 from jvasseur/lazy-orm-query-result

Make ORMQueryResult lazy

src/Porpaginas/Doctrine/ORM/ORMQueryResult.php

@@ -39,6 +39,11 @@ class ORMQueryResult implements Result
      */
     private $result;
 
+    /**
+     * @var int|null
+     */
+    private $count;
+
     public function __construct($query, $fetchCollection = true)
     {
         if ($query instanceof QueryBuilder) {
@@ -82,9 +87,11 @@ public function take($offset, $limit)
      */
     public function count()
     {
-        $this->loadQuery();
+        if ($this->count === null) {
+            $this->count = count(new Paginator($this->query, $this->fetchCollection));
+        }
 
-        return count($this->result);
+        return $this->count;
     }
 
     /**
@@ -93,16 +100,12 @@ public function count()
      * @return Iterator
      */
     public function getIterator()
-    {
-        $this->loadQuery();
-
-        return new ArrayIterator($this->result);
-    }
-
-    private function loadQuery()
     {
         if ($this->result === null) {
             $this->result = $this->query->getResult();
+            $this->count = count($this->result);
         }
+
+        return new ArrayIterator($this->result);
     }
 }