From fe6ed17f91565ffef8fba72af086038ba3d8dddc Mon Sep 17 00:00:00 2001
From: Guillaume <g.alabre@thecodingmachine.com>
Date: Wed, 31 Jul 2019 17:02:42 +0200
Subject: [PATCH 1/2] InnerResultIterator: Improve count() performance of
 non-executed query

---
 src/AlterableResultIterator.php |  3 +++
 src/InnerResultIterator.php     | 27 ++++++---------------------
 2 files changed, 9 insertions(+), 21 deletions(-)

diff --git a/src/AlterableResultIterator.php b/src/AlterableResultIterator.php
index 2ddef8a9..a07b705f 100644
--- a/src/AlterableResultIterator.php
+++ b/src/AlterableResultIterator.php
@@ -222,6 +222,9 @@ public function take($offset, $limit)
      */
     public function count()
     {
+        if ($this->resultIterator instanceof \Countable && $this->alterations->count() === 0) {
+            return $this->resultIterator->count();
+        }
         return count($this->toArray());
     }
 
diff --git a/src/InnerResultIterator.php b/src/InnerResultIterator.php
index d280a75e..3abcf01c 100644
--- a/src/InnerResultIterator.php
+++ b/src/InnerResultIterator.php
@@ -102,6 +102,7 @@ protected function executeQuery(): void
         $this->fetchStarted = true;
     }
 
+    private $count = null;
     /**
      * Counts found records (this is the number of records fetched, taking into account the LIMIT and OFFSET settings).
      *
@@ -113,30 +114,14 @@ public function count()
             return $this->count;
         }
 
-        if ($this->tdbmService->getConnection()->getDatabasePlatform() instanceof MySqlPlatform) {
+        if ($this->fetchStarted && $this->tdbmService->getConnection()->getDatabasePlatform() instanceof MySqlPlatform) {
             // Optimisation: we don't need a separate "count" SQL request in MySQL.
-            return $this->getRowCountViaRowCountFunction();
-        } else {
-            return $this->getRowCountViaSqlQuery();
-        }
-    }
-
-    private $count = null;
-
-    /**
-     * Get the row count from the rowCount function (only works with MySQL)
-     */
-    private function getRowCountViaRowCountFunction(): int
-    {
-        if (!$this->fetchStarted) {
-            $this->executeQuery();
+            $this->count = $this->statement->rowCount();
+            return $this->count;
         }
-
-        $this->count = $this->statement->rowCount();
-        return $this->count;
+        return $this->getRowCountViaSqlQuery();
     }
 
-
     /**
      * Makes a separate SQL query to compute the row count.
      * (not needed in MySQL)
@@ -147,7 +132,7 @@ private function getRowCountViaSqlQuery(): int
 
         $this->logger->debug('Running count SQL request: '.$countSql);
 
-        $this->count = $this->tdbmService->getConnection()->fetchColumn($countSql, $this->parameters);
+        $this->count = (int) $this->tdbmService->getConnection()->fetchColumn($countSql, $this->parameters);
         return $this->count;
     }
 

From 65f0e4b7ec65e99b2b6145b264e1d020fe993a02 Mon Sep 17 00:00:00 2001
From: Guillaume <g.alabre@thecodingmachine.com>
Date: Wed, 31 Jul 2019 18:07:05 +0200
Subject: [PATCH 2/2] InnerResultIterator: test count after fetch

---
 tests/TDBMDaoGeneratorTest.php | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/tests/TDBMDaoGeneratorTest.php b/tests/TDBMDaoGeneratorTest.php
index 507dbf18..11f9df99 100644
--- a/tests/TDBMDaoGeneratorTest.php
+++ b/tests/TDBMDaoGeneratorTest.php
@@ -961,6 +961,17 @@ public function testPageJsonEncode(): void
         $this->assertCount(1, $msgDecoded);
     }
 
+    /**
+     * @depends testDaoGeneration
+     */
+    public function testInnerResultIteratorCountAfterFetch(): void
+    {
+        $userDao = new TestUserDao($this->tdbmService);
+        $users = $userDao->getUsersByLoginStartingWith('j')->take(0, 4);
+        $users->toArray(); // We force to fetch
+        $this->assertEquals(3, $users->count());
+    }
+
     /**
      * @depends testDaoGeneration
      */