如何使用 PDOStatement::rowCount 配合事务回滚防止数据丢失

gitbox 2025-05-19

在进行数据库操作时，数据丢失是开发者最头疼的问题之一。无论是用户资料更新、库存扣减，还是订单状态修改，一旦数据库操作未成功执行却未被检测并处理，都可能导致严重后果。

本文将讲解如何结合使用 PDOStatement::rowCount() 和数据库事务回滚机制，在 PHP 中有效防止这种情况的发生。

1. 什么是 rowCount()？

PDOStatement::rowCount() 是 PDO 中用于返回上一条 DELETE、INSERT 或 UPDATE 语句所影响的行数的函数。它能帮助我们判断操作是否真正对数据库产生了影响。

需要注意的是，并非所有数据库驱动都支持对 SELECT 操作调用 rowCount()，通常我们用它来检测是否成功更新或删除了数据。

2. 为什么要配合事务使用？

在执行多个相关的数据库操作时，如果其中某一步未成功，就应当回滚之前的更改，避免出现数据不一致。例如，如果你希望更新一条用户的邮箱地址，但实际上该记录并未被修改，程序却继续执行后续操作，这就可能导致逻辑混乱甚至数据丢失。

这时，结合使用事务和 rowCount() 就显得尤为关键。

3. 示例：用户信息更新中的事务控制

下面是一个实际例子，展示如何在更新用户信息时使用 rowCount() 和事务进行控制，确保数据安全：

<?php
$dsn = 'mysql:host=localhost;dbname=test_db;charset=utf8mb4';
$user = 'db_user';
$password = 'db_pass';

try {
    $pdo = new PDO($dsn, $user, $password, [
        PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION
    ]);

    // 假设我们要更新用户的邮箱
    $userId = 123;
    $newEmail = '[email protected]';

    $pdo->beginTransaction();

    $stmt = $pdo->prepare("UPDATE users SET email = :email WHERE id = :id");
    $stmt->execute([
        ':email' => $newEmail,
        ':id' => $userId
    ]);

    // 使用 rowCount 判断是否真的更新了数据
    if ($stmt->rowCount() === 0) {
        // 没有任何记录被更新，回滚事务
        $pdo->rollBack();
        echo "更新失败，未找到匹配的用户，事务已回滚。";
    } else {
        // 更新成功，提交事务
        $pdo->commit();
        echo "邮箱更新成功，事务已提交。";
    }

} catch (PDOException $e) {
    if ($pdo->inTransaction()) {
        $pdo->rollBack();
    }
    echo "数据库操作失败：" . $e->getMessage();
}
?>

说明：

使用 beginTransaction() 开始一个事务；
execute() 执行更新；
使用 rowCount() 判断是否有记录被实际更新；
若未更新任何记录，立即回滚事务；
否则，安全提交。

4. 应用场景扩展

此方法不仅适用于更新操作，在库存系统、积分扣减、支付处理等场景中都非常实用。例如：

扣减库存失败时回滚订单状态；
更新用户余额失败时回滚支付记录插入；
删除用户失败时回滚日志写入操作等。

5. 小贴士

永远不要假设 UPDATE 或 DELETE 操作一定会成功；
使用事务时务必处理异常，确保回滚；
对于多个依赖性强的操作，务必包裹在一个事务中；
rowCount() 返回 0 并不一定意味着失败，有时是值未变。但在严谨的数据场景中，正是我们想要防范的“无效写入”。

结语

通过结合 PDOStatement::rowCount() 和事务控制机制，开发者可以更精准地判断数据库操作是否真正生效，从而在必要时进行事务回滚，防止“看起来成功、实际上失败”的数据问题。这是高可靠系统开发中不可或缺的一项技巧。

你是否在项目中也遇到过类似的问题？欢迎留言分享你的经验！