add support for manual setting of filtered count

patrickwarner · May 16, 2018 · 73ad493 · 73ad493
1 parent b2d9df3
commit 73ad493
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/src/DataTableAbstract.php b/src/DataTableAbstract.php
@@ -413,6 +413,19 @@ public function setTotalRecords($total)
         return $this;
     }
 
+    /**
+     * Set total records manually.
+     *
+     * @param int $total
+     * @return $this
+     */
+    public function setFilteredRecords($total)
+    {
+        $this->filteredRecords = $total;
+
+        return $this;
+    }
+
     /**
      * Skip pagination as needed.
      *
@@ -556,7 +569,7 @@ protected function filterRecords()
         }
 
         $this->columnSearch();
-        $this->filteredRecords = $this->isFilterApplied ? $this->count() : $this->totalRecords;
+        $this->filteredRecords = $this->isFilterApplied ? $this->filteredCount() : $this->totalRecords;
     }
 
     /**

diff --git a/src/QueryDataTable.php b/src/QueryDataTable.php
@@ -121,6 +121,16 @@ public function totalCount()
         return $this->totalRecords ? $this->totalRecords : $this->count();
     }
 
+    /**
+     * Count total items.
+     *
+     * @return int
+     */
+    public function filteredCount()
+    {
+        return $this->filteredRecords ? $this->filteredRecords : $this->count();
+    }
+
     /**
      * Counts current query.
      *
@@ -148,7 +158,6 @@ protected function prepareCountQuery()
         if (! $this->isComplexQuery($builder)) {
             $row_count = $this->wrap('row_count');
             $builder->select($this->connection->raw("'1' as {$row_count}"));
-            $builder->setBindings([], 'select');
         }
 
         return $builder;