Fix set total & filtered records count

Fix #1453 #1454 #2050 #2609

Author: yajra

src/CollectionDataTable.php

@@ -91,7 +91,7 @@ public static function create($source)
      */
     public function count(): int
     {
-        return ($this->collection->count() > $this->totalRecords) ? $this->totalRecords : $this->collection->count();
+        return $this->collection->count();
     }
 
     /**
@@ -112,8 +112,6 @@ public function columnSearch(): void
                 continue;
             }
 
-            $this->isFilterApplied = true;
-
             $regex = $this->request->isRegex($i);
             $keyword = $this->request->columnKeyword($i);
 
@@ -187,16 +185,6 @@ public function make($mDataSupport = true): JsonResponse
         }
     }
 
-    /**
-     * Count total items.
-     *
-     * @return int
-     */
-    public function totalCount(): int
-    {
-        return $this->totalRecords ?: $this->collection->count();
-    }
-
     /**
      * Get results.
      *
@@ -254,8 +242,6 @@ protected function globalSearch(string $keyword): void
         $keyword = $this->config->isCaseInsensitive() ? Str::lower($keyword) : $keyword;
 
         $this->collection = $this->collection->filter(function ($row) use ($keyword) {
-            $this->isFilterApplied = true;
-
             $data = $this->serialize($row);
             foreach ($this->request->searchableColumnIndex() as $index) {
                 $column = $this->getColumnName($index);

src/DataTableAbstract.php

@@ -72,16 +72,23 @@ abstract class DataTableAbstract implements DataTable
     /**
      * Total records.
      *
-     * @var int
+     * @var int|null
      */
-    protected int $totalRecords = 0;
+    protected ?int $totalRecords = null;
+
+    /**
+     * Flag to skip total records count query.
+     *
+     * @var bool
+     */
+    protected bool $skipTotalRecords = false;
 
     /**
      * Total filtered records.
      *
-     * @var int
+     * @var int|null
      */
-    protected int $filteredRecords = 0;
+    protected ?int $filteredRecords = null;
 
     /**
      * Auto-filter flag.
@@ -109,17 +116,10 @@ abstract class DataTableAbstract implements DataTable
         'DT_RowAttr' => [],
     ];
 
-    /**
-     * [internal] Track if any filter was applied for at least one column.
-     *
-     * @var bool
-     */
-    protected bool $isFilterApplied = false;
-
     /**
      * Custom ordering callback.
      *
-     * @var ?callable
+     * @var callable|null
      */
     protected $orderCallback = null;
 
@@ -453,7 +453,7 @@ public function with($key, $value = ''): static
      * @param  callable  $value
      * @return $this
      */
-    public function withQuery($key, callable $value): static
+    public function withQuery(string $key, callable $value): static
     {
         $this->appends[$key] = $value;
 
@@ -492,7 +492,7 @@ public function blacklist(array $blacklist): static
      * @param  string|array  $whitelist
      * @return $this
      */
-    public function whitelist($whitelist = '*'): static
+    public function whitelist(array|string $whitelist = '*'): static
     {
         $this->columnDef['whitelist'] = $whitelist;
 
@@ -505,7 +505,7 @@ public function whitelist($whitelist = '*'): static
      * @param  bool  $state
      * @return $this
      */
-    public function smart($state = true): static
+    public function smart(bool $state = true): static
     {
         $this->config->set('datatables.search.smart', $state);
 
@@ -518,7 +518,7 @@ public function smart($state = true): static
      * @param  bool  $state
      * @return $this
      */
-    public function startsWithSearch($state = true): static
+    public function startsWithSearch(bool $state = true): static
     {
         $this->config->set('datatables.search.starts_with', $state);
 
@@ -531,7 +531,7 @@ public function startsWithSearch($state = true): static
      * @param  bool  $multiTerm
      * @return $this
      */
-    public function setMultiTerm($multiTerm = true): static
+    public function setMultiTerm(bool $multiTerm = true): static
     {
         $this->config->set('datatables.search.multi_term', $multiTerm);
 
@@ -544,20 +544,35 @@ public function setMultiTerm($multiTerm = true): static
      * @param  int  $total
      * @return $this
      */
-    public function setTotalRecords($total): static
+    public function setTotalRecords(int $total): static
     {
+        $this->skipTotalRecords();
         $this->totalRecords = $total;
 
         return $this;
     }
 
+    /**
+     * Skip total records and set the recordsTotal equals to recordsFiltered.
+     * This will improve the performance by skipping the total count query.
+     *
+     * @return $this
+     */
+    public function skipTotalRecords(): static
+    {
+        $this->skipTotalRecords = true;
+        $this->totalRecords = 0;
+
+        return $this;
+    }
+
     /**
      * Set filtered records manually.
      *
      * @param  int  $total
      * @return $this
      */
-    public function setFilteredRecords($total): static
+    public function setFilteredRecords(int $total): static
     {
         $this->filteredRecords = $total;
 
@@ -651,7 +666,6 @@ abstract protected function defaultOrdering(): void;
     public function filter(callable $callback, $globalSearch = false): static
     {
         $this->autoFilter = $globalSearch;
-        $this->isFilterApplied = true;
         $this->filterCallback = $callback;
 
         return $this;
@@ -721,7 +735,7 @@ protected function filterRecords(): void
 
         $this->columnSearch();
         $this->searchPanesSearch();
-        $this->filteredRecords = $this->isFilterApplied ? $this->filteredCount() : $this->totalRecords;
+        $this->filteredCount();
     }
 
     /**
@@ -778,14 +792,28 @@ protected function searchPanesSearch(): void
         // Add support for search pane.
     }
 
+    /**
+     * Count total items.
+     *
+     * @return int
+     */
+    public function totalCount(): int
+    {
+        if ($this->skipTotalRecords) {
+            return $this->totalRecords;
+        }
+
+        return $this->totalRecords = $this->count();
+    }
+
     /**
      * Count filtered items.
      *
      * @return int
      */
     protected function filteredCount(): int
     {
-        return $this->filteredRecords ?: $this->count();
+        return $this->filteredRecords ??= $this->count();
     }
 
     /**

src/QueryDataTable.php

@@ -41,13 +41,6 @@ class QueryDataTable extends DataTableAbstract
      */
     protected $limitCallback = null;
 
-    /**
-     * Flag to skip total records count query.
-     *
-     * @var bool
-     */
-    protected bool $skipTotalRecords = false;
-
     /**
      * Flag to keep the select bindings.
      *
@@ -103,9 +96,7 @@ public static function canCreate($source): bool
     public function make($mDataSupport = true): JsonResponse
     {
         try {
-            $this->prepareQuery();
-
-            $results = $this->results();
+            $results = $this->prepareQuery()->results();
             $processed = $this->processResults($results, $mDataSupport);
             $data = $this->transform($results, $processed);
 
@@ -117,8 +108,10 @@ public function make($mDataSupport = true): JsonResponse
 
     /**
      * Prepare query by executing count, filter, order and paginate.
+     *
+     * @return $this
      */
-    protected function prepareQuery(): void
+    protected function prepareQuery(): static
     {
         if (! $this->prepared) {
             $this->totalRecords = $this->totalCount();
@@ -131,22 +124,8 @@ protected function prepareQuery(): void
         }
 
         $this->prepared = true;
-    }
 
-    /**
-     * Count total items.
-     *
-     * @return int
-     */
-    public function totalCount(): int
-    {
-        if ($this->skipTotalRecords) {
-            $this->isFilterApplied = true;
-
-            return 1;
-        }
-
-        return $this->totalRecords ?: $this->count();
+        return  $this;
     }
 
     /**
@@ -215,19 +194,6 @@ public function results(): Collection
         return $this->query->get();
     }
 
-    /**
-     * Skip total records and set the recordsTotal equals to recordsFiltered.
-     * This will improve the performance by skipping the total count query.
-     *
-     * @return $this
-     */
-    public function skipTotalRecords(): static
-    {
-        $this->skipTotalRecords = true;
-
-        return $this;
-    }
-
     /**
      * Keep the select bindings.
      *
@@ -268,8 +234,6 @@ public function columnSearch(): void
                 $keyword = $this->getColumnSearchKeyword($index);
                 $this->compileColumnSearch($index, $column, $keyword);
             }
-
-            $this->isFilterApplied = true;
         }
     }
 
@@ -628,26 +592,9 @@ protected function searchPanesSearch(): void
             } else {
                 $this->query->whereIn($column, $values);
             }
-
-            $this->isFilterApplied = true;
         }
     }
 
-    /**
-     * Count filtered items.
-     *
-     * @return int
-     */
-    protected function filteredCount(): int
-    {
-        $this->filteredRecords = $this->filteredRecords ?: $this->count();
-        if ($this->skipTotalRecords) {
-            $this->totalRecords = $this->filteredRecords;
-        }
-
-        return $this->filteredRecords;
-    }
-
     /**
      * Resolve callback parameter instance.
      *
@@ -769,8 +716,6 @@ protected function globalSearch(string $keyword): void
                     } else {
                         $this->compileQuerySearch($query, $column, $keyword);
                     }
-
-                    $this->isFilterApplied = true;
                 });
         });
     }