如何結合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();
}
?>

代碼分析

事務的開啟與提交<br> 使用$pdo->beginTransaction()開啟事務， $pdo->commit()用於提交事務如果出現任何問題，我們將使用$pdo->rollBack()來回滾事務。
利用rowCount進行判斷<br> 在每個SQL操作後，我們使用$stmt->rowCount()來檢查操作是否成功例如，如果UPDATE語句沒有更新任何行（即餘額不足或條件不匹配），則可以通過rowCount()返回0來判斷，並觸發回滾操作。
異常處理<br> 通過try-catch語句，我們捕獲到任何異常並回滾事務如果rowCount()返回0或其他錯誤情況出現，都會觸發throw new Exception()拋出異常，進而進行回滾。
數據庫操作的可靠性<br> 使用事務處理可以保證數據庫操作的原子性即使發生異常，事務的回滾保證了數據庫的一致性，避免了部分操作生效，導致數據不一致的情況。

使用rowCount判斷是否成功執行SQL語句

rowCount()函數非常有用，特別是在需要確保數據修改確實生效的情況下。如果SQL語句沒有對任何行做出修改， rowCount()將返回0，這通常意味著操作沒有達到預期效果。在事務處理中，我們可以根據rowCount()的值來決定是否回滾事務。

例如，在銀行轉賬操作中，我們會檢查更新賬戶餘額和插入交易記錄是否成功。如果任何一步失敗，事務就會回滾，從而保持數據的一致性和完整性。

總結

通過結合使用PDOStatement::rowCount和事務管理，我們可以在PHP應用程序中高效、可靠地執行數據庫操作。 rowCount()為我們提供了判斷SQL操作是否成功的一個重要手段，而事務則確保了操作的原子性，一旦出現問題，可以及時回滾，避免數據錯誤。

這種方法可以廣泛應用於需要確保數據一致性的場景，比如金融系統、庫存管理系統等。

PDOStatement::rowCount