JAVA基础知识之JDBC——JDBC事务处理及批量更新

JDBC事务

JDBC的事务由Connection提供，默认是打开的。

要开启事务，则要关闭自动提交，

conn.setAutoCommit(false);

提交事务使用

conn.commit();

回滚事务使用

conn.rollback();

可以在某个位置设置一个保存点，回滚时只回滚到这个点，

try {
    Savepoint sp = conn.setSavepoint();
...
} catch (SQLException e) {
    conn.rollback(sp);
}

当Connection遇到一个未处理的SQLException异常时，系统会非正常退出，事务还是会自动回滚。但如果程序捕获了该异常，则需要在异常处理块中显示地回滚。

下面演示JDBC事务的基本用法，

package db;


import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Properties;

public class TransactionTest {
    private String driver;
    private String url;
    private String user;
    private String pass;
    public void initParam(String paramFile) throws FileNotFoundException, IOException, ClassNotFoundException {
        //用Properties类加载属性文件
        Properties prop = new Properties();
        prop.load(new FileInputStream(paramFile));
        driver = prop.getProperty("driver");
        url = prop.getProperty("url");
        user = prop.getProperty("user");
        pass = prop.getProperty("pass");
        Class.forName(driver);
    }
    public void insertInTransaction(String[] sqls) throws SQLException{
        try (Connection conn = DriverManager.getConnection(url, user, pass)) {
            //关闭自动提交，即开启事务
            conn.setAutoCommit(false);
            try (Statement stmt = conn.createStatement()) {
                for (String sql : sqls) {
                    stmt.executeUpdate(sql);
                }
                //提交事务
                conn.commit();
            }catch (SQLException e) {
                conn.rollback();
                System.out.println("Exception:");
                System.out.println(e.getMessage());
            }
        } 
    }
    public static void main(String[] args) throws FileNotFoundException, ClassNotFoundException, IOException, SQLException {
        TransactionTest tt = new TransactionTest();
        tt.initParam("mysql.ini");
        String[] sqls = new String[] {
                "insert into jdbc_test values (null, 'aaa','111')",
                "insert into jdbc_test values (null, 'bbb', '222')",
                "insert into jdbc_test values (null, 'ccc', '333')",
                //下面这条sql会失败，因为已经存在主键为2的记录
                "insert into jdbc_test values (2, 'ddd','444')"
        };
        tt.insertInTransaction(sqls);
    }
}

上面的第53行sql会引发SQLException异常， 因此前面三个insert都会被回滚， 程序执行如下，

Exception:
Duplicate entry '2' for key 'PRIMARY'

JDBC批量更新

使用批量更新时，多条sql语句将被作为同一批操作被同时收集，同时提交。

使用Statement收集sql的方法为

stmt.addBatch(sql1);
stmt.addBatch(sql1);
stmt.addBatch(sql1);
...

同时执行所有sql的方法为执行Statement的executeBatch().  (JDK.8中如果执行影响记录太多，则可以用executeLargeBatch()方法)

stmt.executeBatch();

之后可以提交更新，为了达到批量更新的目的，程序应该在批量操作之前关闭自动提交，然后收集sql，当批量操作执行之后，提交事务，并恢复之前的自动提交模式。

关键代码如下，

    public void insertInTransaction(String[] sqls) throws SQLException{
        try ( Connection conn = DriverManager.getConnection(url, user, pass)) {
            //关闭自动提交，即开启事务
            conn.setAutoCommit(false);
            try(Statement stmt = conn.createStatement()) {
                for (String sql : sqls) {
                    stmt.addBatch(sql);
                }
                stmt.executeBatch();
                //提交事务
                conn.commit();
            } catch (SQLException e) {
                conn.rollback();
                System.out.println("Exception:");
                System.out.println(e.getMessage());
            } finally {
                conn.setAutoCommit(true);
            }
        } 
    }