61 lines
1.1 KiB
PHP
61 lines
1.1 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
namespace Illuminate\Container;
|
||
|
|
|
||
|
|
use Countable;
|
||
|
|
use IteratorAggregate;
|
||
|
|
|
||
|
|
class RewindableGenerator implements Countable, IteratorAggregate
|
||
|
|
{
|
||
|
|
/**
|
||
|
|
* The generator callback.
|
||
|
|
*
|
||
|
|
* @var callable
|
||
|
|
*/
|
||
|
|
protected $generator;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* The number of tagged services.
|
||
|
|
*
|
||
|
|
* @var callable|int
|
||
|
|
*/
|
||
|
|
protected $count;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Create a new generator instance.
|
||
|
|
*
|
||
|
|
* @param callable $generator
|
||
|
|
* @param callable|int $count
|
||
|
|
* @return void
|
||
|
|
*/
|
||
|
|
public function __construct(callable $generator, $count)
|
||
|
|
{
|
||
|
|
$this->count = $count;
|
||
|
|
$this->generator = $generator;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Get an iterator from the generator.
|
||
|
|
*
|
||
|
|
* @return mixed
|
||
|
|
*/
|
||
|
|
public function getIterator()
|
||
|
|
{
|
||
|
|
return ($this->generator)();
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Get the total number of tagged services.
|
||
|
|
*
|
||
|
|
* @return int
|
||
|
|
*/
|
||
|
|
public function count()
|
||
|
|
{
|
||
|
|
if (is_callable($count = $this->count)) {
|
||
|
|
$this->count = $count();
|
||
|
|
}
|
||
|
|
|
||
|
|
return $this->count;
|
||
|
|
}
|
||
|
|
}
|