Make query for filteredRecords when totalRecords was manually set

Author: JurianArie

src/QueryDataTable.php

@@ -25,6 +25,11 @@ class QueryDataTable extends DataTableAbstract
      */
     protected bool $prepared = false;
 
+    /**
+     * Flag to check if the total records count query has been performed.
+     */
+    protected bool $performedTotalRecordsCount = false;
+
     /**
      * Query callback for custom pagination using limit without offset.
      *
@@ -157,6 +162,20 @@ public function prepareQuery(): static
         return $this;
     }
 
+    /**
+     * Count total items.
+     */
+    public function totalCount(): int
+    {
+        if ($this->totalRecords !== null) {
+            return $this->totalRecords;
+        }
+
+        $this->performedTotalRecordsCount = true;
+
+        return $this->totalRecords = $this->count();
+    }
+
     /**
      * Counts current query.
      */
@@ -253,7 +272,7 @@ protected function filterRecords(): void
 
         // If no modification between the original query and the filtered one has been made
         // the filteredRecords equals the totalRecords
-        if ($this->query == $initialQuery) {
+        if ($this->query == $initialQuery && $this->performedTotalRecordsCount) {
             $this->filteredRecords ??= $this->totalRecords;
         } else {
             $this->filteredCount();