PHP迭代器和生成器用法实例分析

更新时间：2020-04-08 23:03:11 作者：startmvc

本文实例讲述了PHP迭代器和生成器用法。分享给大家供大家参考，具体如下：迭代器迭代器

本文实例讲述了PHP迭代器和生成器用法。分享给大家供大家参考，具体如下：

迭代器

迭代器实际是一个实现了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();
}