如何解决DoctrineORM批量处理内存溢出？ocramius/doctrine-batch-utils助你轻松优化！

<pre class="brush:php;toolbar:false;">$users = $entityManager->getRepository(User::class)->findAll(); // 或者一个大查询
foreach ($users as $user) {
    $user->setInvitationCode(generateUniqueCode());
    // ... 其他业务逻辑
}
$entityManager->flush();

Allowed memory size of X bytes exhausted

$query = $entityManager->createQuery('SELECT u FROM App\Entity\User u');
$iterableResult = $query->iterate(); // 使用iterate()减少初始内存占用
$batchSize = 100;
$i = 0;

foreach ($iterableResult as $row) {
    $user = $row[0];
    $user->setInvitationCode(generateUniqueCode());
    // ... 其他业务逻辑

    if (($i % $batchSize) === 0) {
        $entityManager->flush(); // 每100个实体刷新一次
        $entityManager->clear(); // 清除内存中的实体，释放内存
    }
    ++$i;
}
$entityManager->flush(); // 刷新剩余的实体
$entityManager->clear(); // 清除剩余的实体
登录后复制

<pre class="brush:php;toolbar:false;">composer require ocramius/doctrine-batch-utils

IteratorAggregate

ObjectManager#flush()

ObjectManager#clear()

use DoctrineBatchUtilsBatchProcessingSimpleBatchIteratorAggregate;
use AppEntityUser; // 假设你的用户实体

// 1. 定义你的查询，获取需要处理的实体
$query = $entityManager->createQuery('SELECT u FROM App\Entity\User u WHERE u.invitationCode IS NULL');

// 2. 使用 SimpleBatchIteratorAggregate 包装查询结果
// 第一个参数是查询对象，第二个参数是批次大小（例如：每100个实体刷新一次）
$iterable = SimpleBatchIteratorAggregate::fromQuery(
    $query,
    100 // 每100个实体执行一次 flush() 和 clear()
);

// 3. 像往常一样遍历迭代器
foreach ($iterable as $user) {
    // 这里的 $user 始终是“新鲜”的，即处于 managed 状态
    // 因为迭代器会自动重新获取实体，避免了手动 clear 后的实体游离问题
    $user->setInvitationCode(generateUniqueCode());
    // ... 执行你的业务逻辑
}

// 4. 循环结束后，SimpleBatchIteratorAggregate 会自动处理剩余的 flush/clear
// 你无需再手动调用 $entityManager->flush(); $entityManager->clear();
echo "所有用户邀请码已更新完毕，内存管理妥当！";
登录后复制