Merge pull request #25 from AppsDevTeam/overwritable-count-expr

Overwritable count expr
AppsDevTeam · Nov 2, 2022 · faa403b · faa403b
2 parents b748af6 + e686c59
commit faa403b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/QueryObject.php b/src/QueryObject.php
@@ -528,7 +528,7 @@ final public function count(): int
 	{
 		$qb = $this->createQueryBuilder(false);
 
-		$qb->select('COUNT(e.id)');
+		$qb->select('COUNT(' . $this->getCountExpr() . ')');
 
 		if ($qb->getDQLPart('groupBy')) {
 			$paginator = new Doctrine\ORM\Tools\Pagination\Paginator($qb);
@@ -539,6 +539,11 @@ final public function count(): int
 
 		return (int) $query->getSingleScalarResult();
 	}
+
+	protected function getCountExpr(): string
+	{
+		return $this->entityAlias . '.id';
+	}
 
 	final public function getResultSet(int $page, int $itemsPerPage): ResultSet
 	{