如何结合 PDOStatement::rowCount 处理事务中的回滚操作

gitbox 2025-05-28

在使用PDO进行数据库操作时，事务处理是一项非常重要的机制，它可以确保一系列操作要么全部成功，要么全部失败。事务通常包括beginTransaction()、commit()和rollBack()这几个核心方法。当我们执行多个SQL操作时，有时需要根据某些条件判断是否要回滚事务，这时，PDOStatement::rowCount函数就可以派上用场。

PDOStatement::rowCount可以返回上一个SQL语句影响的行数。利用它，我们可以有效地判断数据库操作是否成功，从而决定是否回滚事务。

使用PDO进行事务处理

下面是一个示例，演示了如何利用rowCount判断数据库操作结果，并在必要时回滚事务。

<?php
// 数据库连接
$dsn = 'mysql:host=localhost;dbname=testdb';
$username = 'root';
$password = '';
$options = [
    PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
    PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC,
];

// 创建PDO对象
try {
    $pdo = new PDO($dsn, $username, $password, $options);
} catch (PDOException $e) {
    die('Connection failed: ' . $e->getMessage());
}

// 开始事务
$pdo->beginTransaction();

try {
    // 执行第一个SQL操作
    $stmt1 = $pdo->prepare('UPDATE users SET balance = balance - 100 WHERE id = :id');
    $stmt1->execute(['id' => 1]);

    // 检查影响的行数
    if ($stmt1->rowCount() == 0) {
        // 如果没有行被更新，则回滚事务
        throw new Exception('更新失败，用户余额不足');
    }

    // 执行第二个SQL操作
    $stmt2 = $pdo->prepare('INSERT INTO transactions (user_id, amount) VALUES (:user_id, :amount)');
    $stmt2->execute(['user_id' => 1, 'amount' => -100]);

    // 检查第二个操作的影响行数
    if ($stmt2->rowCount() == 0) {
        // 如果插入失败，回滚事务
        throw new Exception('交易记录插入失败');
    }

    // 如果一切正常，提交事务
    $pdo->commit();
    echo "事务提交成功！";
} catch (Exception $e) {
    // 出现异常时回滚事务
    $pdo->rollBack();
    echo "事务回滚，错误信息: " . $e->getMessage();
}
?>

代码分析

事务的开启与提交
使用$pdo->beginTransaction()开启事务，$pdo->commit()用于提交事务。如果出现任何问题，我们将使用$pdo->rollBack()来回滚事务。
利用rowCount进行判断
在每个SQL操作后，我们使用$stmt->rowCount()来检查操作是否成功。例如，如果UPDATE语句没有更新任何行（即余额不足或条件不匹配），则可以通过rowCount()返回0来判断，并触发回滚操作。
异常处理
通过try-catch语句，我们捕获到任何异常并回滚事务。如果rowCount()返回0或其他错误情况出现，都会触发throw new Exception()抛出异常，进而进行回滚。
数据库操作的可靠性
使用事务处理可以保证数据库操作的原子性。即使发生异常，事务的回滚保证了数据库的一致性，避免了部分操作生效，导致数据不一致的情况。

使用 rowCount 判断是否成功执行SQL语句

rowCount()函数非常有用，特别是在需要确保数据修改确实生效的情况下。如果SQL语句没有对任何行做出修改，rowCount()将返回0，这通常意味着操作没有达到预期效果。在事务处理中，我们可以根据rowCount()的值来决定是否回滚事务。

例如，在银行转账操作中，我们会检查更新账户余额和插入交易记录是否成功。如果任何一步失败，事务就会回滚，从而保持数据的一致性和完整性。

总结

通过结合使用PDOStatement::rowCount和事务管理，我们可以在PHP应用程序中高效、可靠地执行数据库操作。rowCount()为我们提供了判断SQL操作是否成功的一个重要手段，而事务则确保了操作的原子性，一旦出现问题，可以及时回滚，避免数据错误。

这种方法可以广泛应用于需要确保数据一致性的场景，比如金融系统、库存管理系统等。

PDOStatement::rowCount