PHP中yield关键字的使用

PHP版本>=5.5

原理:yield关键字会生成一个Generator类的对象,PHP通过Generator实例计算出下一次迭代的值,再次返回一个Generator对象并停止循环(即循环一次执行一次)。

理解:使用在for/foreach/while循环内部,用来返回循环结构体本次生成的元素。yield会记录结构体本次循环所处的位置,下次执行循环时从上次的位置开始执行,再次生成一个元素。所以yield返回的数组内永远只有一个元素,所以叫做生成器。

目的:节省内存,防止内存溢出。

yield节省内存场景:

(1)在for/foreach/while循环结构体中,生成上亿个元素,在内存中都只有一个元素。

(2)在fgets读取文件中,每次都读取一行数据到内存中,大文件可以像小文件一样读取。

(3)在mysql读取数据中,每次都获取一行数据到内存中,就算几十万的数据也不会占用过多内存。

场景(1):

php 复制代码
<?php
/**
 * 使用 yield关键字创建数组
 * @param $number
 * @return Generator
 */
function createRangeYield($number): Generator
{
    for ($i = 0; $i <= $number; $i++) {
        yield time();

        if (($i % 20000) == 0) {
            // 内存使用以 MB 为单位
            echo "$i\t" . round(memory_get_usage() / 1024 / 1024, 2) . ' MB' . PHP_EOL;
        }
    }
}

/**
 * 直接生成数组
 * @param $number
 * @return array
 */
function createRange($number): array
{
    $data = [];
    for ($i = 0; $i <= $number; $i++) {
        $data[] = time();

        if (($i % 20000) == 0) {
            // 内存使用以 MB 为单位
            echo "$i\t" . round(memory_get_usage() / 1024 / 1024, 2) . ' MB' . PHP_EOL;
        }
    }
    return $data;
}

$time1 = microtime(true) * 1000;
echo "常规函数循环 10W次内存使用量\n";
$result = createRange(100000); // 这里调用上面创建的函数
foreach ($result as $value) {
    # code...
}
$time2 = microtime(true) * 1000;
echo "耗时:".( round($time2 - $time1,3)). "毫秒\n";

$time1 = microtime(true) * 1000;
echo "\nyield函数循环 10W次内存使用量\n";
$result = createRangeYield(100000); // 这里调用上面创建的函数
foreach ($result as $value) {
    # code...
}
$time2 = microtime(true) * 1000;
echo "耗时:".( round($time2 - $time1,3)). "毫秒\n";

执行结果分析:

场景(2):

php 复制代码
<?php
header("content-type:text/html;charset=utf-8");
/**
 * 使用 yield关键字读取大文件
 * @return Generator
 */
function readTxtYield(): Generator
{
    $handle = fopen("D:/ttt.txt", 'rb');

    $i = 0;
    while (feof($handle) === false) {
        yield fgets($handle);

        if (($i % 200000) == 0) {
            // 内存使用以 MB 为单位
            echo "$i\t" . round(memory_get_usage() / 1024 / 1024, 2) . ' MB' . PHP_EOL;
        }

        $i++;
    }

    fclose($handle);
}

/**
 * 常规函数读取大文件
 * @return array
 */
function readTxt(): array
{
    $handle = fopen("D:/ttt.txt", 'rb');

    $i = 0;
    $lines = [];
    while (feof($handle) === false) {
        $lines[] = fgets($handle);

        if (($i % 200000) == 0) {
            // 内存使用以 MB 为单位
            echo "$i\t" . round(memory_get_usage() / 1024 / 1024, 2) . ' MB' . PHP_EOL;
        }

        $i++;
    }

    fclose($handle);
    return $lines;
}

$time1 = microtime(true) * 1000;
echo "常规函数读取 100W行文件内存使用量\n";
foreach (readTxt() as $key => $value) {
    # code...
}
$time2 = microtime(true) * 1000;
echo "耗时:".( round($time2 - $time1,3)). "毫秒\n";

$time1 = microtime(true) * 1000;
echo "\nyield函数读取 100W行文件内存使用量\n";
foreach (readTxtYield() as $key => $value) {
    # code...
}
$time2 = microtime(true) * 1000;
echo "耗时:".( round($time2 - $time1,3)). "毫秒\n";

执行结果分析:

场景(3):

php 复制代码
<?php

const DB_NAME = 'yibai_purchase';// 数据库名称
const DB_HOST = '192.168.73.73';
const DB_USERNAME = 'dev_purchase';
const DB_PASSWORD = 'yb123456';
const LIMIT = 50;
const DSN = 'mysql:host=' . DB_HOST . ';dbname=' . DB_NAME;


/**
 * 使用 yield关键字 读取数据库数据
 * @return Generator
 */
function readSqlDataYield(): Generator
{
    $i = 0;
    $lines = [];

    $options = [
        PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC,
    ];
    $connectObj = new PDO(DSN, DB_USERNAME, DB_PASSWORD, $options);
    $query_tables = "SELECT * FROM yibai_purchase.pur_supplier_payment_record WHERE 1=1 limit 100000";
    $stmt = $connectObj->query($query_tables);
    while ($row = $stmt->fetch(PDO::FETCH_ASSOC)) {
        yield $row;

        if (($i % 20000) == 0) {
            // 内存使用以 MB 为单位
            echo "$i\t" . round(memory_get_usage() / 1024 / 1024, 2) . ' MB' . PHP_EOL;
        }

        $i++;
    }
}


/**
 * 常规函数读取数据库数据
 * @return array
 */
function readSqlData(): array
{
    $i = 0;
    $lines = [];

    $options = [
        PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC,
    ];
    $connectObj = new PDO(DSN, DB_USERNAME, DB_PASSWORD, $options);
    $query_tables = "SELECT * FROM yibai_purchase.pur_supplier_payment_record WHERE 1=1 limit 100000";
    $stmt = $connectObj->query($query_tables);
    while ($row = $stmt->fetch(PDO::FETCH_ASSOC)) {
        $lines[] = $row;

        if (($i % 20000) == 0) {
            // 内存使用以 MB 为单位
            echo "$i\t" . round(memory_get_usage() / 1024 / 1024, 2) . ' MB' . PHP_EOL;
        }

        $i++;
    }

    return $lines;

}

$time1 = microtime(true) * 1000;
echo "常规函数读取 10W行数据库数据内存使用量\n";
foreach (readSqlData() as $key => $value) {
    # code...
}
$time2 = microtime(true) * 1000;
echo "耗时:".( round($time2 - $time1,3)). "毫秒\n";

$time1 = microtime(true) * 1000;
echo "\nyield函数读取 10W行数据库数据内存使用量\n";
foreach (readSqlDataYield() as $key => $value) {
    # code...
}
$time2 = microtime(true) * 1000;
echo "耗时:".( round($time2 - $time1,3)). "毫秒\n";

执行结果分析:

综合三种场景,发现内存使用量明显减少,并且执行用时不会有太大差异。

相关推荐
初心未改HD1 分钟前
Go 文件与 I/O 操作完全指南
开发语言·golang
eastyuxiao4 分钟前
流程图 + 配置清单 在团队 / 公司知识管理场景的应用落地
大数据·流程图
szial4 分钟前
uv 实战指南:用一个工具重塑 Python 开发工作流
开发语言·python·uv
Adolf_19935 分钟前
Mac 配置Homebrew + Oh My Zsh + npm全局权限问题
大数据·elasticsearch·搜索引擎
wjs202417 分钟前
HTML 段落
开发语言
数字化顾问20 分钟前
(121页PPT)DG1886IT信息化规划报告(附下载方式)
大数据·人工智能
CSCN新手听安22 分钟前
【Qt】Qt窗口(五)QDialog对话框的使用,点击按钮弹出新的对话框,自定义对话框界面,模态对话框model
开发语言·c++·qt
枫叶丹424 分钟前
【HarmonyOS 6.0】CANN Kit 新增支持获取 AI 模型 Dump 维测数据功能详解
开发语言·人工智能·华为·信息可视化·harmonyos
沐知全栈开发27 分钟前
JavaScript Array(数组)
开发语言
xxjj998a29 分钟前
PHP vs C++:性能与用途全解析
php·laravel