本文实例讲述了PHP迭代器和生成器用法。分享给大家供大家参考,具体如下:
迭代器实际是一个实现了Iterator的类,可以用foreach进行遍历。
例如:
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官方文档中的示例
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遍历。