Added \Iterator or \Generator support for slice callback return items

AshleyDawson · Jan 17, 2019 · b824bcb · b824bcb
1 parent de3bb0c
commit b824bcb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/AshleyDawson/SimplePagination/Paginator.php b/lib/AshleyDawson/SimplePagination/Paginator.php
@@ -144,6 +144,9 @@ public function paginate($currentPageNumber = 1)
         } else {
             $items = $sliceCallback($offset, $this->itemsPerPage, $pagination);
         }
+        if ($items instanceof \Iterator) {
+            $items = iterator_to_array($items);   
+        }
         $afterQueryCallback($this, $pagination);
 
         $previousPageNumber = null;