Add ArrayCollection::chunk() function

getgrav · Sep 15, 2017 · 588c509 · 588c509
1 parent 37b5818
commit 588c509
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 0 deletions.
diff --git a/system/src/Grav/Framework/Collection/AbstractLazyCollection.php b/system/src/Grav/Framework/Collection/AbstractLazyCollection.php
@@ -42,6 +42,15 @@ public function shuffle()
         return $this->collection->shuffle();
     }
 
+    /**
+     * {@inheritDoc}
+     */
+    public function chunk($size)
+    {
+        $this->initialize();
+        return $this->collection->chunk($size);
+    }
+
     /**
      * {@inheritDoc}
      */

diff --git a/system/src/Grav/Framework/Collection/ArrayCollection.php b/system/src/Grav/Framework/Collection/ArrayCollection.php
@@ -50,6 +50,17 @@ public function shuffle()
         return $this->createFrom(array_replace(array_flip($keys), $this->toArray()));
     }
 
+    /**
+     * Split collection into chunks.
+     *
+     * @param int $size     Size of each chunk.
+     * @return array
+     */
+    public function chunk($size)
+    {
+        return array_chunk($this->toArray(), $size, true);
+    }
+
     /**
      * Implementes JsonSerializable interface.
      *

diff --git a/system/src/Grav/Framework/Collection/CollectionInterface.php b/system/src/Grav/Framework/Collection/CollectionInterface.php
@@ -30,4 +30,12 @@ public function reverse();
      * @return static
      */
     public function shuffle();
+
+    /**
+     * Split collection into chunks.
+     *
+     * @param int $size     Size of each chunk.
+     * @return array
+     */
+    public function chunk($size);
 }