| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960 |
- <?php
- namespace Illuminate\Container;
- use Countable;
- use IteratorAggregate;
- use Traversable;
- 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
- */
- public function __construct(callable $generator, $count)
- {
- $this->count = $count;
- $this->generator = $generator;
- }
- /**
- * Get an iterator from the generator.
- *
- * @return \Traversable
- */
- public function getIterator(): Traversable
- {
- return ($this->generator)();
- }
- /**
- * Get the total number of tagged services.
- *
- * @return int
- */
- public function count(): int
- {
- if (is_callable($count = $this->count)) {
- $this->count = $count();
- }
- return $this->count;
- }
- }
|