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);
             }
         }
     }