Add withCount method to Criteria (#561)

andersao · Dec 21, 2019 · 684c46a · 684c46a
1 parent 65b5fa2
commit 684c46a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/Prettus/Repository/Criteria/RequestCriteria.php b/src/Prettus/Repository/Criteria/RequestCriteria.php
@@ -44,6 +44,7 @@ public function apply($model, RepositoryInterface $repository)
         $orderBy = $this->request->get(config('repository.criteria.params.orderBy', 'orderBy'), null);
         $sortedBy = $this->request->get(config('repository.criteria.params.sortedBy', 'sortedBy'), 'asc');
         $with = $this->request->get(config('repository.criteria.params.with', 'with'), null);
+        $withCount = $this->request->get(config('repository.criteria.params.withCount', 'withCount'), null);
         $searchJoin = $this->request->get(config('repository.criteria.params.searchJoin', 'searchJoin'), null);
         $sortedBy = !empty($sortedBy) ? $sortedBy : 'asc';
 
@@ -163,6 +164,11 @@ public function apply($model, RepositoryInterface $repository)
             $model = $model->with($with);
         }
 
+        if ($withCount) {
+            $withCount = explode(';', $withCount);
+            $model = $model->withCount($withCount);
+        }
+
         return $model;
     }