【JAVA】Java高级：批处理与性能提升：在JDBC中实现批处理，提高数据库操作性能

本文链接：https://blog.youkuaiyun.com/thinking_chou/article/details/144179229

你在一家餐厅，点了一份大餐。厨师需要准备多个菜品，如果每个菜品都单独点单、单独下单，服务员来回跑动，既浪费时间又增加了出错的机会。相反，如果服务员一次性把所有菜品的订单传给厨房，厨师可以更高效地进行准备，这样不仅节省了时间，还提高了工作效率。在这个例子中，单独下单就像是逐条执行数据库操作，而一次性下单则类似于JDBC中的批处理。

一、JDBC批处理的基本概念

JDBC批处理是指将多个SQL语句组合成一个批次，一次性提交给数据库执行。这种方式可以显著减少与数据库的交互次数，从而提高性能。批处理的主要优点包括：

减少网络往返：通过将多个操作合并为一个批次，减少了与数据库的连接和数据传输次数。
提高执行效率：数据库能够更有效地处理批量操作，优化执行计划。
降低资源消耗：减少了连接和关闭数据库的开销。

二、JDBC批处理的实现步骤

下面将通过一个具体的示例，详细介绍如何在JDBC中实现批处理。

1. 创建数据库表

首先，我们需要一个数据库表来进行批处理操作。假设我们有一个用户表users，用于存储用户信息。

CREATE TABLE users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    username VARCHAR(50) NOT NULL,
    password VARCHAR(50) NOT NULL
);

2. JDBC批处理示例代码

以下是一个使用JDBC进行批处理的示例代码，展示了如何插入多条用户记录。

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;

public class BatchProcessingExample {
    public static void main(String[] args) {
        Connection connection = null;
        Statement statement = null;

        try {
            // 连接到数据库
            connection = DriverManager.getConnection("jdbc:mysql://localhost:3306/mydb", "root", "password");
            // 关闭自动提交
            connection.setAutoCommit(false);
            statement = connection.createStatement();

            // 添加批处理的SQL语句
            String sql1 = "INSERT INTO users (username, password) VALUES ('user1', 'password1')";
            String sql2 = "INSERT INTO users (username, password) VALUES ('user2', 'password2')";
            String sql3 = "INSERT INTO users (username, password) VALUES ('user3', 'password3')";

            // 将SQL语句添加到批处理中
            statement.addBatch(sql1);
            statement.addBatch(sql2);
            statement.addBatch(sql3);

            // 执行批处理
            int[] results = statement.executeBatch();

            // 提交事务
            connection.commit();
            System.out.println("Batch processing completed successfully.");

            // 输出每条SQL语句的执行结果
            for (int result : results) {
                System.out.println("Affected rows: " + result);
            }

        } catch (SQLException e) {
            // 发生异常时回滚事务
            try {
                if (connection != null) {
                    connection.rollback();
                    System.out.println("Transaction rolled back due to an error.");
                }
            } catch (SQLException rollbackEx) {
                rollbackEx.printStackTrace();
            }
            // 处理SQLException
            System.err.println("SQL Error: " + e.getMessage());
            e.printStackTrace();
        } finally {
            // 关闭连接
            try {
                if (statement != null) statement.close();
                if (connection != null) connection.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }
}

三、代码解释

连接数据库：使用DriverManager.getConnection方法连接到数据库。
关闭自动提交：调用connection.setAutoCommit(false)，将连接的自动提交功能关闭，以便手动管理事务。
创建Statement对象：使用connection.createStatement()创建一个Statement对象，用于执行SQL语句。
添加批处理的SQL语句：使用statement.addBatch(sql)方法将多个SQL语句添加到批处理中。
执行批处理：调用statement.executeBatch()方法执行批处理，返回一个整型数组，表示每条SQL语句影响的行数。
提交事务：如果所有操作成功，调用connection.commit()提交事务。
异常处理：如果在执行批处理时发生SQLException，则捕获异常并调用connection.rollback()回滚事务，以确保数据的一致性。
关闭连接：在finally块中关闭Statement和Connection，确保资源被释放。