对于PHP 的生成器，用过的都说好！

Student_Li
262 阅读3分钟

或许你在 PHP 开发过程中已经见过它，甚至已经用过几次，但如果你还没有真正了解过它的强大功能，那么这篇文章会带你走进生成器的世界，看看为什么“用过的都说好”！

一、什么是生成器？

首先，生成器是 PHP5.5 引入的一种轻量级的协程实现方式。它允许我们可以在迭代过程中暂停执行，稍后再继续。生成器的语法特别简洁，但却可以大幅优化代码的执行效率，尤其是在处理大量数据时，生成器的优势尤为明显。

二、生成器的基本用法

生成器和普通的函数有些相似，但不同的是它不是一次性返回所有数据，而是可以“按需”提供数据。当生成器的 yield 关键字被调用时，函数会返回一个值并暂停执行，下次需要时再继续执行。

1. 基本示例

让我们从一个简单的例子开始：


<?php

function numberGenerator() {
    for ($i = 1; $i <= 5; $i++) {
        yield $i;
    }
}

$generator = numberGenerator();
foreach ($generator as $number) {
    echo $number . PHP_EOL;
}

解释：

  • yield：yield 的作用类似于 return，但不同的是它不会终止函数，而是暂停函数的执行，并在下一次调用时从暂停的地方继续执行。
  • numberGenerator：这是一个简单的生成器函数，它会生成从 1 到 5 的数字。

输出：

1
2
3
4
5

在这个例子中，生成器每次 yield 一个数字，然后函数暂停，等待下次调用 foreach 时继续。

三、生成器 VS 普通函数

你可能会问：“这和普通的返回数组有什么区别？”

区别在于性能。在需要处理非常大的数据集时，生成器可以显著降低内存使用。假设你要生成 100 万个数字，使用普通的数组返回方式，会消耗大量内存，而生成器则是按需生成数据，内存占用小很多。

2. 使用普通数组的方式

<?php

function generateNumbers() {
    $numbers = [];
    for ($i = 1; $i <= 1000000; $i++) {
        $numbers[] = $i;
    }
    return $numbers;
}

$numbers = generateNumbers();
foreach ($numbers as $number) {
    echo $number . PHP_EOL;
}

这种方法会将 100 万个数字一次性加载到内存中，对服务器资源的消耗非常大，容易造成性能瓶颈。

3. 使用生成器的方式

<?php

function generateNumbers() {
    for ($i = 1; $i <= 1000000; $i++) {
        yield $i;
    }
}

$numbers = generateNumbers();
foreach ($numbers as $number) {
    echo $number . PHP_EOL;
}

使用生成器则不会一次性将 100 万个数字加载到内存中，而是按需生成，极大降低了内存使用。

四、生成器的高级用法

生成器不仅仅局限于返回简单的值，它还可以通过 send() 方法将数据传递回生成器，并可以结合 key() 和 current() 方法获取更多的信息。

4. 双向通信的生成器
<?php

function taskGenerator() {
    $result = (yield 'Task 1 started');
    echo "Result of Task 1: " . $result . PHP_EOL;

    $result = (yield 'Task 2 started');
    echo "Result of Task 2: " . $result . PHP_EOL;
}

$generator = taskGenerator();
echo $generator->current() . PHP_EOL; // 输出 "Task 1 started"
$generator->send('Task 1 completed'); // 输出 "Result of Task 1: Task 1 completed"
echo $generator->current() . PHP_EOL; // 输出 "Task 2 started"
$generator->send('Task 2 completed'); // 输出 "Result of Task 2: Task 2 completed"

解释：

  • send()：通过 send() 方法，我们可以将数据从外部传递回生成器内部。
  • current()：获取当前生成的值。
  • yield：生成器通过 yield 暂停执行并等待外部传入的数据。

五、实际应用场景

那么生成器的实际应用场景是什么呢？下面列举几个常见的场景：

1. 读取大文件

在 PHP 中，处理大文件时，如果直接将文件内容全部读入内存可能导致内存溢出问题，而生成器可以一行一行地读取文件，极大降低内存使用。

<?php

function readFileLineByLine($filePath) {
    $file = fopen($filePath, 'r');
    while (!feof($file)) {
        yield fgets($file);
    }
    fclose($file);
}

$fileGenerator = readFileLineByLine('large_file.txt');
foreach ($fileGenerator as $line) {
    echo $line;
}
2. 数据库分页

处理数据库分页时，通常我们需要从数据库中批量提取数据，生成器可以让我们逐行处理查询结果，而不需要一次性加载所有数据。

<?php

function fetchDataFromDatabase(PDO $pdo) {
    $stmt = $pdo->query('SELECT * FROM users');
    while ($row = $stmt->fetch(PDO::FETCH_ASSOC)) {
        yield $row;
    }
}

$pdo = new PDO('mysql:host=localhost;dbname=test', 'root', '');
$dataGenerator = fetchDataFromDatabase($pdo);
foreach ($dataGenerator as $row) {
    print_r($row);
}

六、总结

生成器为 PHP 开发者提供了一种轻量级处理大量数据的方式。它通过 按需生成 数据，避免了将大量数据一次性加载到内存中，从而大大提高了内存效率。在处理大文件、大量数据库记录或是复杂算法时，生成器都可以显著优化性能。

生成器的优势不仅仅在于性能，它还让我们的代码更加优雅，避免了大量的临时变量和冗余代码。希望这篇文章能够帮助你深入了解生成器的强大功能，让你的 PHP 开发更加高效！

avatar
文章
阅读
粉丝