PHP - 迭代两次泛型可迭代

时间:2017-04-24 16:46:02

标签: php iterator generator iterable

在PHP 7.1中,有一个新的iterable psudo类型,用于抽象数组和Traversable个对象。

假设我的代码中有一个如下所示的类:

class Foo
{
    private $iterable;

    public function __construct(iterable $iterable)
    {
        $this->iterable = $iterable;
    }

    public function firstMethod()
    {
        foreach ($this->iterable as $item) {...}
    }

    public function secondMethod()
    {
        foreach ($this->iterable as $item) {...}
    }
}

这项工作正常,$iterable是一个数组或Iterator,除非$iterableGenerator。事实上,在这种情况下,调用firstMethod()然后调用secondMethod()会产生以下Exception: Cannot traverse an already closed generator

有没有办法避免这个问题?

1 个答案:

答案 0 :(得分:2)

发电机不能倒带。如果要避免此问题,则必须创建新的生成器。如果您创建一个实现IteratorAggregate的对象,则可以自动完成此操作:

class Iter implements IteratorAggregate
{
    public function getIterator()
    {
        foreach ([1, 2, 3, 4, 5] as $i) {
            yield $i;
        }
    }
}

然后只需将此对象的实例作为迭代器传递:

$iter = new Iter();
$foo = new Foo($iter);
$foo->firstMethod();
$foo->secondMethod();

输出:

1
2
3
4
5
1
2
3
4
5