feature #98 Total count JobExecution matched by query

Author: guich25

src/batch-doctrine-dbal/src/DoctrineDBALJobExecutionStorage.php

@@ -7,6 +7,7 @@
 use Doctrine\DBAL\Connection;
 use Doctrine\DBAL\Driver\Result;
 use Doctrine\DBAL\Exception as DBALException;
+use Doctrine\DBAL\Query\QueryBuilder;
 use Doctrine\DBAL\Schema\Comparator;
 use Doctrine\DBAL\Schema\Schema;
 use Doctrine\DBAL\Types\Types;
@@ -147,33 +148,11 @@ public function list(string $jobName): iterable
 
     public function query(Query $query): iterable
     {
-        $queryParameters = [];
-        $queryTypes = [];
-
         $qb = $this->connection->createQueryBuilder();
         $qb->select('*')
             ->from($this->table);
 
-        $names = $query->jobs();
-        if (\count($names) > 0) {
-            $qb->andWhere($qb->expr()->in('job_name', ':jobNames'));
-            $queryParameters['jobNames'] = $names;
-            $queryTypes['jobNames'] = Connection::PARAM_STR_ARRAY;
-        }
-
-        $ids = $query->ids();
-        if (\count($ids) > 0) {
-            $qb->andWhere($qb->expr()->in('id', ':ids'));
-            $queryParameters['ids'] = $ids;
-            $queryTypes['ids'] = Connection::PARAM_STR_ARRAY;
-        }
-
-        $statuses = $query->statuses();
-        if (\count($statuses) > 0) {
-            $qb->andWhere($qb->expr()->in('status', ':statuses'));
-            $queryParameters['statuses'] = $statuses;
-            $queryTypes['statuses'] = Connection::PARAM_INT_ARRAY;
-        }
+        [$queryParameters, $queryTypes] = $this->addWheres($query, $qb);
 
         switch ($query->sort()) {
             case Query::SORT_BY_START_ASC:
@@ -196,6 +175,18 @@ public function query(Query $query): iterable
         yield from $this->queryList($qb->getSQL(), $queryParameters, $queryTypes);
     }
 
+    public function count(Query $query): int
+    {
+        $qb = $this->connection->createQueryBuilder();
+        $qb->select('count(*)')
+            ->from($this->table);
+
+        /** @var int $result */
+        $result = $this->connection->executeQuery($qb->getSQL(), ...$this->addWheres($query, $qb))->fetchOne();
+
+        return $result;
+    }
+
     private function getSchema(): Schema
     {
         $schema = new Schema();
@@ -325,4 +316,36 @@ private function getNormalizer(): JobExecutionRowNormalizer
 
         return $this->normalizer;
     }
+
+    /**
+     * @return array<int, array<string, array<int|string>|int>>
+     */
+    private function addWheres(Query $query, QueryBuilder $qb): array
+    {
+        $queryParameters = [];
+        $queryTypes = [];
+
+        $names = $query->jobs();
+        if (\count($names) > 0) {
+            $qb->andWhere($qb->expr()->in('job_name', ':jobNames'));
+            $queryParameters['jobNames'] = $names;
+            $queryTypes['jobNames'] = Connection::PARAM_STR_ARRAY;
+        }
+
+        $ids = $query->ids();
+        if (\count($ids) > 0) {
+            $qb->andWhere($qb->expr()->in('id', ':ids'));
+            $queryParameters['ids'] = $ids;
+            $queryTypes['ids'] = Connection::PARAM_STR_ARRAY;
+        }
+
+        $statuses = $query->statuses();
+        if (\count($statuses) > 0) {
+            $qb->andWhere($qb->expr()->in('status', ':statuses'));
+            $queryParameters['statuses'] = $statuses;
+            $queryTypes['statuses'] = Connection::PARAM_INT_ARRAY;
+        }
+
+        return [$queryParameters, $queryTypes];
+    }
 }

src/batch-doctrine-dbal/tests/DoctrineDBALJobExecutionStorageTest.php

@@ -282,6 +282,7 @@ public function testQuery(QueryBuilder $queryBuilder, array $expectedCouples): v
         $this->loadFixtures($storage);
 
         self::assertExecutions($expectedCouples, $storage->query($queryBuilder->getQuery()));
+        self::assertSame(\count($expectedCouples), $storage->count($queryBuilder->getQuery()));
     }
 
     public function queries(): Generator

src/batch/src/Storage/FilesystemJobExecutionStorage.php

@@ -139,6 +139,14 @@ public function query(Query $query): iterable
         return \array_slice($candidates, $query->offset(), $query->limit());
     }
 
+    public function count(Query $query): int
+    {
+        /** @var JobExecution[] $result */
+        $result = $this->query($query);
+
+        return \count($result);
+    }
+
     private function buildFilePath(string $jobName, string $executionId): string
     {
         return \implode(DIRECTORY_SEPARATOR, [$this->directory, $jobName, $executionId]) .

src/batch/src/Storage/QueryableJobExecutionStorageInterface.php

@@ -17,4 +17,9 @@ interface QueryableJobExecutionStorageInterface extends ListableJobExecutionStor
      * @return iterable|JobExecution[]
      */
     public function query(Query $query): iterable;
+
+    /**
+     * Execute query against stored job executions, and return count result.
+     */
+    public function count(Query $query): int;
 }

src/batch/tests/Storage/FilesystemJobExecutionStorageTest.php

@@ -150,6 +150,7 @@ public function testQueryWithProvider(QueryBuilder $query, array $expectedCouple
         );
 
         self::assertExecutions($expectedCouples, $storage->query($query->getQuery()));
+        self::assertEquals(\count($expectedCouples), $storage->count($query->getQuery()));
     }
 
     public function query(): \Generator