迭代器和生成器

迭代器

迭代器实际是一个实现了Iterator的类,可以用foreach进行遍历。
例如:

<?php
class Sample implements Iterator{
    private $curIndex=0;
    private $items=null;
    public function __construct($_items) {
        $this->items = $_items;
    }
    public function current (){
        echo "current\n";
        return $this->items[$this->curIndex];
    }

    public function key (){
        echo "key\n";
        return $this->curIndex;
    }

    public function next (){      
           echo "next\n"; 
           $this->curIndex++;
    }

    public function rewind (){
            $this->curIndex = 0;     
    }

    public function send ( $value ){
       if($value == "stop"){
           $this->curIndex = null;
       }
    }

    public function valid (){
        echo "valid\n";
        return  isset($this->items[$this->curIndex]);
    }
}

$sample = new Sample([1,2,3]);
foreach ($sample as $k =>$v){
    
}

输出 valid current key next
可以看到foreach 是先调用valid判断迭代器是否有效,然后再调用current获取当前值,同时调用next移动key到指向下一个值(输出key是因为 $k=>$v的缘故)。

生成器

让我们先看一下官方文档

生成器提供了一种更容易的方法来实现简单的对象迭代,相比较定义类实现 Iterator 接口的方式,性能开销和复杂性大大降低。
生成器允许你在 foreach 代码块中写代码来迭代一组数据而不需要在内存中创建一个数组, 那会使你的内存达到上限,或者会占据可观的处理时间。
相反,你可以写一个生成器函数,就像一个普通的自定义函数一样, 和普通函数只返回一次不同的是, 生成器可以根据需要 yield 多次,以便生成需要迭代的值。
PHP 将会在每次需要值的时候调用生成器函数,并在产生一个值之后保存生成器的状态,这样它就可以在需要产生下一个值的时候恢复调用状态。

下面是php官方文档中的示例

<?php
function gen_one_to_three() {
    for ($i = 1; $i <= 3; $i++) {
        //注意变量$i的值在不同的yield之间是保持传递的。
        yield $i;
    }
}

$generator = gen_one_to_three();
foreach ($generator as $value) {
    echo "$value\n";
}
var_dump($generator);  //实际上是Generator对象

如上,若把3修改成10000,对于$generator实际上没有区别,它只是保存了一个当前值(当然还有相关的内部状态,这里是为了简化),并没有产生10000个数。
从中可以看出生成器的优势在于减少内存的使用,在需要时才生成对应的值。

查看php文档,我们可以看到Generator实际也是Iterator的具体实现,yield调用时就是返回的Generator对象。
那么怎么理解迭代器和生成器的关系呢?
其实,生成器是迭代器的实现+yield,产生了生成器对象。

我们也可以自己定义一个类似yield的函数,如下:

function myYeild(){
    $args = func_get_args();
    return new Sample($args);
}

$generator = myYeild(1,2,3);
foreach ($generator as $value) {
    echo "$value\n";
}

注意,我们的myYeild,是不能和php内置的yeild那么使用的,因为yeild会保存调用上下文,临时离开,并没有return。
这里只是类比一下。
既然yeild可以把普通的对象包装成generator,那么我们的iterator通过yeild也可以像Generator一样吗?
答案有点悲伤,yeild是把传入的值作为参数生成Generator实例,它并不知道我们的iterator。不过这样设计也是合理的,
以防我们自己的iterator不靠谱。

实际使用场合

  • 数据库遍历
    可以结合游标,遍历数据库时,不需要一次返回所有数据,而是每次取一行。
class AllUser implements \Iterator
{
    protected $index = 0;
    protected $data = [];

    public function __construct()
    {
        $link = mysqli_connect('192.168.0.91', 'root', '123', 'xxx');
        $rec = mysqli_query($link, 'select id from doc_admin');
        $this->data = mysqli_fetch_all($rec, MYSQLI_ASSOC);
    }

    //1 重置迭代器
    public function rewind()
    {
        $this->index = 0;
    }
    //2 验证迭代器是否有数据
    public function valid()
    {
        return $this->index < count($this->data);
    }

    //3 获取当前内容
    public function current()
    {
        $id = $this->data[$this->index];
        return User::find($id);
    }

    //4 移动key到下一个
    public function next()
    {
        return $this->index++;
    }


    //5 迭代器位置key
    public function key()
    {
        return $this->index;
    }
}

//实现迭代遍历用户表
$users = new AllUser();
//可实时修改
foreach ($users as $user){
    $user->add_time = time();
    $user->save();
}

  • 文件遍历
    一次读取一行
  • 实现Iterator接口,让普通类可以使用foreach遍历。
  • 协程,参见鸟哥则这篇文章

注意:可以在生成器的函数前加"&",可以使用引用。在函数里直接return会终止生成器。

posted @ 2017-04-03 21:58  半山th  阅读(305)  评论(0编辑  收藏  举报