PHP生成器实战：使用yield关键字优化大数组遍历内存占用

先说一个核心判断：当PHP脚本需要遍历数十万甚至上百万元素时，频繁遭遇内存溢出或响应迟缓，十有八九是因为传统方式一次性把全部数据加载进了内存。要解决这个问题，PHP生成器（yield关键字）是目前实践中最优雅、也最有效的方案。

下面就来拆解一下，如何用yield从根上优化大数组遍历的内存占用。

一、重构数据生成逻辑为生成器函数

最直接的思路，就是把原本返回完整数组的函数改造成生成器。核心变化在于：数据不再一次性预加载，而是按需产出，内存占用从线性增长直接压缩到恒定级别。

操作步骤并不复杂：

• 首先，把原函数里所有return $array的语句全部删掉。
• 然后，在循环体内，把原先向数组追加元素的代码（比如$arr[] = $item）直接替换为yield $item。
• 正式一点，最好给函数加上返回类型声明，写清楚: Generator。
• 最后，调用这个函数时，不要再接收结果数组了，直接把它放进foreach循环里，数据自然就一条条吐出来。

二、用xrange生成器替代range函数

说到遍历数值范围，range() 的机制是立刻创建一个完整的整数数组放进内存。规模一上来，比如 range(1, 1000000)，内存瞬间就爆了。

换个思路，自定义一个xrange生成器。它只记录起始、结束和步长三个状态量，每次迭代只计算并产生一个值，内存开销几乎可以忽略不计。

实现代码也很干净：定义一个function xrange($start, $end, $step = 1): Generator，里面写一个for ($i = $start; $i <= $end; $i += $step)循环，每轮执行yield $i。然后业务代码里的foreach (range(1, 1000000) as $n)就变成foreach (xrange(1, 1000000) as $n)，效果一模一样，内存占用天差地别。

三、封装数据库查询为逐行生成器

处理PDO或MySQLi查询时，很多人习惯用fetchAll()一下子把全部记录拉回来，这是最典型的“内存杀手”。正确的做法是游标式单行获取，配合yield逐行产出。

具体来说：执行完$stmt->execute()后，不调fetchAll()。直接进入while ($row = $stmt->fetch(PDO::FETCH_ASSOC))循环，每拿到一行，就执行yield $row。外部消费时用foreach (getUsersGenerator() as $user)一行一行处理，处理完一行，内存里的那一行就被释放了。百万级记录，内存稳定在一个很小的范围内。

四、处理大文件时逐行yield文本行

读取GB级日志或CSV文件，如果用了file()或file_get_contents()，结果只有一个：内存爆炸。换用fopen + fgets配合yield，内存占用直接降到KB级别。

套路如下：在生成器函数里用$handle = fopen($filename, 'r')打开文件，然后用while (($line = fgets($handle)) !== false)逐行读取。每读一行，就yield trim($line)。处理完了，在函数末尾fclose($handle)。整个过程，内存占用始终是KB级，读1GB和读10GB没区别。

五、递归结构展开时用yield替代递归数组拼接

遍历无限级分类、嵌套JSON这类树形结构时，传统递归容易陷入两个困境：一是栈溢出，二是递归过程中不断拼接数组导致内存堆积。yield能把深度优先遍历转变为惰性流式产出，压力瞬间释放。

关键点在于：生成器函数接收一个节点参数，先yield $node当前节点，然后检查它有没有子节点。如果有，对每个子节点递归调用自身。注意，递归调用处必须使用yield from tra verseTree($child)，而不是用array_merge()或赋值拼接。同时要确保每条分支路径最终都以yield或yield from结束，绝不返回数组。