如何正确使用 PDOStatement::rowCount 获取 INSERT INTO 语句的影响行数

在使用 PHP 操作数据库时，PDO 是一种推荐的方式，它提供了统一、灵活且安全的接口。其中，PDOStatement::rowCount() 方法常用于获取 SQL 操作影响的行数，但它在不同的语句类型（如 SELECT、INSERT、UPDATE、DELETE）中的行为略有不同。本文重点讲解如何，避免常见的误区。

一、rowCount() 对 INSERT 的适用性

通常我们希望在执行插入操作后，知道有多少行被成功插入，比如：

$pdo = new PDO('mysql:host=localhost;dbname=test', 'user', 'password');
$sql = "INSERT INTO users (name, email) VALUES (:name, :email)";
$stmt = $pdo->prepare($sql);

$stmt->bindValue(':name', 'Alice');
$stmt->bindValue(':email', '[email protected]');
$stmt->execute();

echo $stmt->rowCount(); // 预期输出：1

在大多数情况下，特别是使用 MySQL 时，如果执行的是单条 INSERT 语句，rowCount() 会正确返回影响的行数（通常为 1）。但如果你在意 跨数据库兼容性，就需要小心。

二、批量插入时的 rowCount 行为

对于如下批量插入语句：

$sql = "INSERT INTO users (name, email) VALUES 
        ('Bob', '[email protected]'), 
        ('Charlie', '[email protected]')";
$stmt = $pdo->prepare($sql);
$stmt->execute();

echo $stmt->rowCount(); // 在 MySQL 下通常返回 2

虽然 rowCount() 很可能会返回 2，但这并不总是被所有数据库驱动支持，在 PostgreSQL、SQLite 等数据库中，该行为可能不可用或返回 0。

官方文档提示：

对于某些数据库，如 PostgreSQL，rowCount() 对 INSERT 类型的语句返回值不可靠。

所以，如果你想在跨平台环境中稳妥使用 rowCount() 获取插入行数，请确保：

• 使用支持该功能的数据库驱动（如 MySQL）。

• 使用标准的 INSERT 语句，不依赖 ON DUPLICATE KEY 或 RETURNING 子句（除非数据库明确支持）。

三、如何更稳妥地获取插入影响的行数？

以下是一些实用建议：

1. 确保使用的是 PDO + MySQL

确认当前环境使用的是 MySQL 或兼容的驱动，且 rowCount() 的返回值有意义。

if ($pdo->getAttribute(PDO::ATTR_DRIVER_NAME) === 'mysql') {
    echo "插入成功，影响行数：" . $stmt->rowCount();
} else {
    echo "非 MySQL 数据库，rowCount() 结果可能不准确";
}

2. 对于单行插入，用 rowCount() 是安全的

大多数驱动都支持对单条 INSERT 返回受影响行数为 1，这是推荐做法。

3. 对于多行插入，推荐用事务配合循环

$pdo->beginTransaction();
$total = 0;

$stmt = $pdo->prepare("INSERT INTO users (name, email) VALUES (:name, :email)");
$data = [
    ['name' => 'David', 'email' => '[email protected]'],
    ['name' => 'Eve', 'email' => '[email protected]']
];

foreach ($data as $row) {
    $stmt->execute($row);
    $total += $stmt->rowCount();
}

$pdo->commit();

echo "共插入 $total 行";

通过单条插入配合循环，可以确保 rowCount() 在各驱动下行为一致。

四、常见误区汇总

五、替代方案：使用 LAST_INSERT_ID()

如果你只关心插入的主键 ID（尤其是自增主键），可以用：

$pdo->lastInsertId();

注意：只适用于插入单行且表存在自增主键。

六、总结

PDOStatement::rowCount() 是一个强大但在不同数据库间行为略有不同的工具。在使用它来获取 INSERT INTO 语句的影响行数时，建议：

• 优先在 MySQL 环境下使用。

• 对批量插入使用事务配合循环，更加稳妥。

• 在多数据库场景中，应考虑使用数据库特有的 RETURNING 或其他特性，避免依赖 rowCount()。

如需查看更多 PDO 的实践示例，可访问 https://gitbox.net/docs/pdo。