diff options
Diffstat (limited to 'vendor/guzzle/guzzle/src/Guzzle/Iterator/ChunkedIterator.php')
-rw-r--r-- | vendor/guzzle/guzzle/src/Guzzle/Iterator/ChunkedIterator.php | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/vendor/guzzle/guzzle/src/Guzzle/Iterator/ChunkedIterator.php b/vendor/guzzle/guzzle/src/Guzzle/Iterator/ChunkedIterator.php new file mode 100644 index 0000000..d76cdd4 --- /dev/null +++ b/vendor/guzzle/guzzle/src/Guzzle/Iterator/ChunkedIterator.php @@ -0,0 +1,56 @@ +<?php + +namespace Guzzle\Iterator; + +/** + * Pulls out chunks from an inner iterator and yields the chunks as arrays + */ +class ChunkedIterator extends \IteratorIterator +{ + /** @var int Size of each chunk */ + protected $chunkSize; + + /** @var array Current chunk */ + protected $chunk; + + /** + * @param \Traversable $iterator Traversable iterator + * @param int $chunkSize Size to make each chunk + * @throws \InvalidArgumentException + */ + public function __construct(\Traversable $iterator, $chunkSize) + { + $chunkSize = (int) $chunkSize; + if ($chunkSize < 0 ) { + throw new \InvalidArgumentException("The chunk size must be equal or greater than zero; $chunkSize given"); + } + + parent::__construct($iterator); + $this->chunkSize = $chunkSize; + } + + public function rewind() + { + parent::rewind(); + $this->next(); + } + + public function next() + { + $this->chunk = array(); + for ($i = 0; $i < $this->chunkSize && parent::valid(); $i++) { + $this->chunk[] = parent::current(); + parent::next(); + } + } + + public function current() + { + return $this->chunk; + } + + public function valid() + { + return (bool) $this->chunk; + } +} |