Paginator total override

laravel/framework#46410
mongodb · Jul 5, 2024 · fd67dac · fd67dac
1 parent e79b0ae
commit fd67dac
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/src/Relations/EmbedsMany.php b/src/Relations/EmbedsMany.php
@@ -4,6 +4,7 @@
 
 namespace MongoDB\Laravel\Relations;
 
+use Closure;
 use Illuminate\Database\Eloquent\Collection;
 use Illuminate\Database\Eloquent\Model;
 use Illuminate\Pagination\LengthAwarePaginator;
@@ -19,6 +20,7 @@
 use function is_array;
 use function method_exists;
 use function throw_if;
+use function value;
 
 class EmbedsMany extends EmbedsOneOrMany
 {
@@ -289,21 +291,22 @@ protected function associateExisting($model)
     }
 
     /**
-     * @param  int|null $perPage
-     * @param  array    $columns
-     * @param  string   $pageName
-     * @param  int|null $page
+     * @param int|Closure      $perPage
+     * @param array|string     $columns
+     * @param string           $pageName
+     * @param int|null         $page
+     * @param Closure|int|null $total
      *
      * @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
      */
-    public function paginate($perPage = null, $columns = ['*'], $pageName = 'page', $page = null)
+    public function paginate($perPage = null, $columns = ['*'], $pageName = 'page', $page = null, $total = null)
     {
         $page    = $page ?: Paginator::resolveCurrentPage($pageName);
-        $perPage = $perPage ?: $this->related->getPerPage();
-
         $results = $this->getEmbedded();
         $results = $this->toCollection($results);
-        $total   = $results->count();
+        $total   = value($total) ?? $results->count();
+        $perPage = $perPage ?: $this->related->getPerPage();
+        $perPage = $perPage instanceof Closure ? $perPage($total) : $perPage;
         $start   = ($page - 1) * $perPage;
 
         $sliced = $results->slice(