PHP 生成器：让您的代码更清晰、更高效

admin 2023-12-01 487 阅读 0评论

PHP 新手在处理大型数据集或数组时，可能会遇到内存耗尽错误。这是一个常见的问题，如果不加以解决，可能会导致应用程序崩溃：

Fatal error: Allowed memory size of xxxxxx bytes exhausted (tried to allocate xxxxx bytes) in your_script.php on line xx

PHP 内存耗尽错误表示您的脚本尝试分配超过允许的内存量。为了解决此问题，PHP 提供了生成器。生成器是一种特殊的函数，可让您迭代数据集，而无需将所有数据加载到内存中。这对于处理大型数据集非常有效。

什么是生成器？

生成器是一种迭代数据集的有效方式，因为它只在需要时才将数据加载到内存中。这对于处理大型数据集尤其有用，因为它可以帮助您避免内存耗尽错误。

生成器如何工作？

生成器只在需要时才将数据加载到内存中，这使得它们在处理大型数据集时非常高效。这是通过使用 yield 关键字来实现的。

使用生成器：分步指南

1. 生成器基本语法

要创建生成器，您可以使用 yield 语句定义一个函数。该函数将返回一个生成器对象，该对象可以用于迭代数据集。

以下是一个使用 yield 语句创建生成器的例子：

function numberGenerator($start = 1, $end = 10) {
    for ($i = $start; $i <= $end; $i++) {
        yield $i;
    }
}

// 使用生成器：
foreach (numberGenerator() as $number) {
    echo $number . "<br>";
}

2. 将常规函数转换为生成器

假设您有一个返回数组的函数：

function getNumbers($start = 1, $end = 10) {
    $numbers = [];
    for ($i = $start; $i <= $end; $i++) {
        $numbers[] = $i;
    }
    return $numbers;
}

使用较大 $end 值的此函数可能会导致内存问题，因为它会将整个序列一次性加载到内存中。要将其转换为生成器，只需将 yield 语句添加到循环中即可：

function getNumbersGenerator($start = 1, $end = 10) {
    for ($i = $start; $i <= $end; $i++) {
        yield $i;
    }
}

3. 生成键值对

您还可以生成键值对：

function keyValueGenerator() {
    yield 'a' => 1;
    yield 'b' => 2;
    yield 'c' => 3;
}

将数据发送回生成器

生成器不仅仅用于输出数据，还可以用于接收输入。这可以通过调用生成器的 send() 方法来实现。

function receiveDataGenerator() {
    while (true) {
        $data = yield;
        if ($data === 'stop') {
            return; // 退出生成器
        }
        echo "Received: $data<br>";
    }
}

$generator = receiveDataGenerator();
$generator->send('Hello');
$generator->send('World');
$generator->send('stop');

从生成器返回值

虽然生成器主要用于生成值，但它们也可以返回最终值。这可以通过在生成器函数中使用 return 语句来实现：

function countdownGenerator($start) {
    for ($i = $start; $i >= 0; $i--) {
        yield $i;
    }
    return 'Blastoff!';
}

$gen = countdownGenerator(5);
foreach ($gen as $value) {
    echo "$value<br>";
}
echo $gen->getReturn();// 输出“Blastoff！”

实际用例

生成器在处理大文件或数据流等场景中非常有用。例如，以下代码将逐行读取一个大文件：

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

foreach (readFileLineByLine('large_file.txt') as $line) {
    echo $line . "<br>";
}