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);  5             try(Statement stmt = conn.createStatement()) {
                for (String sql : sqls) {
 stmt.addBatch(sql);  8                 }
 stmt.executeBatch(); 10                 //提交事務
 conn.commit(); 12             } catch (SQLException e) {
 conn.rollback(); 14                 System.out.println("Exception:");
                System.out.println(e.getMessage());
            } finally {
                conn.setAutoCommit(true); 18             }
        } 
    }