JDBC批处理

255次阅读

共计 1556 个字符，预计需要花费 4 分钟才能阅读完成。

使用 JDBC 操作数据库的时候，经常会执行一些批量操作。

例如，一次性给会员增加可用优惠券若干，我们可以执行以下 SQL 代码：

INSERT INTO coupons (user_id, type, expires) VALUES (123, 'DISCOUNT', '2030-12-31');
INSERT INTO coupons (user_id, type, expires) VALUES (234, 'DISCOUNT', '2030-12-31');
INSERT INTO coupons (user_id, type, expires) VALUES (345, 'DISCOUNT', '2030-12-31');
INSERT INTO coupons (user_id, type, expires) VALUES (456, 'DISCOUNT', '2030-12-31');
...

实际上执行 JDBC 时，因为只有占位符参数不同，所以 SQL 实际上是一样的：

for (var params : paramsList) {PreparedStatement ps = conn.preparedStatement("INSERT INTO coupons (user_id, type, expires) VALUES (?,?,?)");
    ps.setLong(params.get(0));
    ps.setString(params.get(1));
    ps.setString(params.get(2));
    ps.executeUpdate();}

类似的还有，给每个员工薪水增加 10%～30%：

UPDATE employees SET salary = salary * ? WHERE id = ?

通过一个循环来执行每个 PreparedStatement 虽然可行，但是性能很低。SQL 数据库对 SQL 语句相同，但只有参数不同的若干语句可以作为 batch 执行，即批量执行，这种操作有特别优化，速度远远快于循环执行每个 SQL。

在 JDBC 代码中，我们可以利用 SQL 数据库的这一特性，把同一个 SQL 但参数不同的若干次操作合并为一个 batch 执行。我们以批量插入为例，示例代码如下：

try (PreparedStatement ps = conn.prepareStatement("INSERT INTO students (name, gender, grade, score) VALUES (?, ?, ?, ?)")) {// 对同一个 PreparedStatement 反复设置参数并调用 addBatch():
    for (Student s : students) {ps.setString(1, s.name);
        ps.setBoolean(2, s.gender);
        ps.setInt(3, s.grade);
        ps.setInt(4, s.score);
        ps.addBatch(); // 添加到 batch
    }
    // 执行 batch:
    int[] ns = ps.executeBatch();
    for (int n : ns) {System.out.println(n + "inserted."); // batch 中每个 SQL 执行的结果数量
    }
}