如何通過PDOStatement::rowCount 判斷批量操作的執行結果

gitbox 2025-05-28

在PHP 中，PDO（PHP Data Objects）提供了與數據庫交互的一個強大接口，能夠幫助我們進行各種數據庫操作。批量操作是指在一次數據庫事務中執行多個插入、更新或刪除操作。而PDOStatement::rowCount是一個用於獲取受影響的行數的方法，通常用於判斷SQL 查詢是否成功執行。

本文將介紹如何使用PDOStatement::rowCount函數來判斷批量操作是否成功執行，並給出相關代碼示例。

1. PDOStatement::rowCount的作用

PDOStatement::rowCount方法返回最後一條SQL 語句影響的行數。這可以用於判斷：

插入操作中多少行被成功插入
更新操作中有多少行受到了修改
刪除操作中有多少行被刪除

不過，需要注意的是，不是所有數據庫驅動都能準確地返回受影響的行數，尤其是對於某些SELECT 查詢。在進行批量操作時， rowCount的返回值可以幫助我們判斷操作是否成功。

2. 使用rowCount判斷批量操作是否成功

在批量操作中，我們通常會使用事務（transaction）來確保一組操作的原子性。在執行多條SQL 語句時，可以通過rowCount判斷每條語句是否成功執行。

示例代碼：

以下是一個使用PDO執行批量插入操作，並通過rowCount判斷是否成功的示例：

 <?php
// 數據庫連接設置
$dsn = 'mysql:host=localhost;dbname=testdb';
$username = 'root';
$password = 'password';

try {
    // 創建 PDO 實例
    $pdo = new PDO($dsn, $username, $password);
    $pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);

    // 開始事務
    $pdo->beginTransaction();

    // 批量插入數據
    $sql = "INSERT INTO users (name, email) VALUES (:name, :email)";
    $stmt = $pdo->prepare($sql);

    // 假設要插入的批量數據
    $users = [
        ['name' => 'John Doe', 'email' => '[email protected]'],
        ['name' => 'Jane Doe', 'email' => '[email protected]'],
        ['name' => 'Alice', 'email' => '[email protected]']
    ];

    $success = true;
    foreach ($users as $user) {
        // 綁定參數並執行插入操作
        $stmt->bindParam(':name', $user['name']);
        $stmt->bindParam(':email', $user['email']);
        
        if (!$stmt->execute()) {
            // 如果執行失敗，標記為失敗
            $success = false;
            break;
        }

        // 檢查是否插入了數據
        if ($stmt->rowCount() == 0) {
            $success = false;
            break;
        }
    }

    // 提交事務，如果插入成功
    if ($success) {
        $pdo->commit();
        echo "批量插入成功！";
    } else {
        $pdo->rollBack();
        echo "批量插入失敗！";
    }
} catch (PDOException $e) {
    // 事務回滾並顯示錯誤信息
    $pdo->rollBack();
    echo "數據庫錯誤：" . $e->getMessage();
}
?>

代碼說明：

數據庫連接：首先，我們使用PDO連接到數據庫。
事務管理：通過$pdo->beginTransaction()開始事務，確保批量插入操作的原子性。
執行插入：在foreach循環中，我們逐個插入用戶數據。每次插入前，使用bindParam綁定參數，並執行SQL 語句。
檢查行數：使用rowCount()方法檢查每次插入操作是否影響了數據庫中的行。如果rowCount()返回0 ，說明沒有插入任何數據，操作失敗。
事務提交或回滾：如果所有插入操作都成功，調用$pdo->commit()提交事務，否則調用$pdo->rollBack()回滾事務。

3. 使用rowCount判斷其他批量操作

除了插入操作， rowCount也可以用於批量更新或刪除操作。在更新或刪除操作時，我們可以根據rowCount的返回值判斷是否成功更新或刪除了指定的行。

示例：批量更新操作

<?php
// 數據庫連接設置
$dsn = 'mysql:host=localhost;dbname=testdb';
$username = 'root';
$password = 'password';

try {
    // 創建 PDO 實例
    $pdo = new PDO($dsn, $username, $password);
    $pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);

    // 開始事務
    $pdo->beginTransaction();

    // 批量更新操作
    $sql = "UPDATE users SET email = :email WHERE name = :name";
    $stmt = $pdo->prepare($sql);

    // 假設要更新的數據
    $users = [
        ['name' => 'John Doe', 'email' => '[email protected]'],
        ['name' => 'Jane Doe', 'email' => '[email protected]']
    ];

    $success = true;
    foreach ($users as $user) {
        // 綁定參數並執行更新操作
        $stmt->bindParam(':name', $user['name']);
        $stmt->bindParam(':email', $user['email']);
        
        if (!$stmt->execute()) {
            // 如果執行失敗，標記為失敗
            $success = false;
            break;
        }

        // 檢查是否更新了數據
        if ($stmt->rowCount() == 0) {
            $success = false;
            break;
        }
    }

    // 提交事務，如果更新成功
    if ($success) {
        $pdo->commit();
        echo "批量更新成功！";
    } else {
        $pdo->rollBack();
        echo "批量更新失敗！";
    }
} catch (PDOException $e) {
    // 事務回滾並顯示錯誤信息
    $pdo->rollBack();
    echo "數據庫錯誤：" . $e->getMessage();
}
?>

4. 總結

通過使用PDOStatement::rowCount函數，我們可以輕鬆判斷批量操作是否成功執行。無論是批量插入、更新還是刪除操作，我們都可以根據rowCount返回的影響行數來判斷操作是否成功執行。這種方法非常適用於需要確保操作成功與否的情況，尤其是在使用事務時，可以確保數據的一致性和完整性。