如何在INSERT 操作中獲取影響的行數並執行相應操作

gitbox 2025-05-29

在使用PHP 操作數據庫時， PDO提供了強大的功能來執行SQL 查詢。而在執行INSERT操作時，了解影響的行數對於後續的處理是非常重要的。 PDOStatement::rowCount()方法可以幫助我們獲取執行SQL 後受影響的行數。本文將詳細介紹如何在使用PDO執行INSERT操作後，利用rowCount獲取影響的行數，並根據該結果進行處理。

1. 簡單的INSERT操作

首先，確保你已經連接到數據庫並準備好執行INSERT操作。以下是一個簡單的INSERT操作：

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

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

    // 準備插入數據的 SQL 語句
    $sql = "INSERT INTO users (name, email) VALUES (:name, :email)";
    $stmt = $pdo->prepare($sql);

    // 綁定參數
    $stmt->bindParam(':name', $name);
    $stmt->bindParam(':email', $email);

    // 示例數據
    $name = 'John Doe';
    $email = '[email protected]';

    // 執行 INSERT 操作
    $stmt->execute();

    // 獲取影響的行數
    $affectedRows = $stmt->rowCount();
    
    // 輸出影響的行數
    echo "插入了 $affectedRows 行數據\n";

} catch (PDOException $e) {
    echo 'Connection failed: ' . $e->getMessage();
}
?>

2. 解釋PDOStatement::rowCount()

在上述代碼中， $stmt->execute()執行了INSERT操作， $stmt->rowCount()方法被用來返回影響的行數。在INSERT操作中，通常返回的是影響的行數，即插入的記錄數量。如果一切正常，返回的值應該為1 ，表示成功插入了一行數據。

3. 根據rowCount()的結果進行不同處理

接下來，我們可以根據rowCount()返回的值進行不同的處理。通常情況下，我們可以使用這個返回值來判斷是否成功插入數據。例如：

 <?php
if ($affectedRows > 0) {
    echo "數據插入成功！\n";
} else {
    echo "數據插入失敗！\n";
}
?>

4. 完整代碼

結合上面的示例，我們可以將整個過程放在一起，確保根據插入結果進行相應處理。

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

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

    // 準備插入數據的 SQL 語句
    $sql = "INSERT INTO users (name, email) VALUES (:name, :email)";
    $stmt = $pdo->prepare($sql);

    // 綁定參數
    $stmt->bindParam(':name', $name);
    $stmt->bindParam(':email', $email);

    // 示例數據
    $name = 'John Doe';
    $email = '[email protected]';

    // 執行 INSERT 操作
    $stmt->execute();

    // 獲取影響的行數
    $affectedRows = $stmt->rowCount();

    // 根據影響行數進行不同的處理
    if ($affectedRows > 0) {
        echo "數據插入成功！\n";
    } else {
        echo "數據插入失敗！\n";
    }

} catch (PDOException $e) {
    echo 'Connection failed: ' . $e->getMessage();
}
?>

5. 注意事項

rowCount()並不適用於所有數據庫操作。例如，某些數據庫管理系統（如MySQL）在INSERT操作時返回的值總是1 ，即使沒有數據被插入。這種情況會依賴於數據庫的具體實現。
對於沒有影響行數的操作， rowCount()可能返回0 ，因此在實際開發中，建議在執行SQL 後，檢查返回值以確保數據的準確插入。

結論

PDOStatement::rowCount()方法在執行INSERT操作後非常有用，它能幫助開發者檢查是否成功插入了數據。通過根據rowCount()的值進行不同的處理，我們可以有效地進行錯誤處理和數據驗證。