当前位置:

首页 > 编程开发 > 批处理教程 >

JDBC的PreparedStatement启动事务使用批处理executeBatch()

JDBC使用MySQL处理大数据的时候,自然而然的想到要使用批处理，

普通的执行过程是：每处理一条数据，就访问一次数据库；

而批处理是：累积到一定数量，再一次性提交到数据库，减少了与数据库的交互次数，所以效率会大大提高

至于事务：事务指逻辑上的一组操作，组成这组操作的各个单元，要不全部成功，要不全部不成功，默认是关闭事务的。

更多事务的资料，请参考这里：http://blog.csdn.net/caomiao2006/article/details/22412755

1. PreparedStatement使用批处理 executeBatch()

1.1. 不使用executeBatch()，而使用executeUpdate()

代码如下：

Class.forName("com.mysql.jdbc.Driver");
Connection conn = DriverManager.getConnection(dbUrl, user, password);
PreparedStatement pstmt = conn.prepareStatement("update content set introtext=? where id=?");

for(int i=0; i<10000; i++){
pstmt.setString(1, "abc"+i);
pstmt.setInt(2, id);
pstmt.executeUpdate();
}

这样，更新10000条数据，就得访问数据库10000次

1.2. 而使用executeBatch()

代码如下：

for(int i=0; i<10000; i++){
pstmt.setString(1, "abc"+i);
pstmt.setInt(2, id);
pstmt.addBatch();//添加到同一个批处理中
}

pstmt.executeBatch();//执行批处理

注意：1. 如果使用了 addBatch() -> executeBatch() 还是很慢，那就得使用到这个参数了

rewriteBatchedStatements=true (启动批处理操作)

在数据库连接URL后面加上这个参数：

String dbUrl = "jdbc:mysql://localhost:3306/User? rewriteBatchedStatements=true";

2. 在代码中，pstmt的位置不能乱放，

//必须放在循环体外

pstmt = conn.prepareStatement("update content set introtext=? where id=?");

for(int i=0; i<10000; i++){

//放这里，批处理会执行不了，因为每次循环重新生成了pstmt，不是同一个了

//pstmt = conn.prepareStatement("update content set introtext=? where id=?");
pstmt.setString(1, "abc"+i);
pstmt.setInt(2, id);
pstmt.addBatch();//添加到同一个批处理中
}

pstmt.executeBatch();//执行批处理

2. 启用事务处理

Class.forName("com.mysql.jdbc.Driver");

Connection conn = DriverManager.getConnection(dbUrl, user, password);

conn.setAutoCommit(false);//将自动提交关闭
PreparedStatement pstmt = conn.prepareStatement("update content set introtext=? where id=?");
pstmt.setString(1, tempintrotext);
pstmt.setInt(2, id);
pstmt.addBatch();
pstmt.executeBatch();
pstmt.close();

conn.commit();//执行完后，手动提交事务
conn.setAutoCommit(true);//再把自动提交打开，避免影响其他需要自动提交的操作
conn.close();

3. 事务和批处理混合使用

Class.forName("com.mysql.jdbc.Driver");

Connection conn = DriverManager.getConnection(dbUrl, user, password);

conn.setAutoCommit(false);//将自动提交关闭
PreparedStatement pstmt = conn.prepareStatement("update content set introtext=? where id=?");

for(int i=0; i<1000000; i++){
pstmt.setString(1, tempintrotext);
pstmt.setInt(2, id);
pstmt.addBatch();

//每500条执行一次，避免内存不够的情况，可参考，Eclipse设置JVM的内存参数

if(i>0 && i%500==0){
pstmt.executeBatch();

//如果不想出错后，完全没保留数据，则可以每执行一次提交一次，但得保证数据不会重复

conn.commit();

}

}
pstmt.executeBatch();//执行最后剩下不够500条的
pstmt.close();

conn.commit();//执行完后，手动提交事务
conn.setAutoCommit(true);//再把自动提交打开，避免影响其他需要自动提交的操作
conn.close();

较完整的代码：

			
public class ExecuteBatchTest {
   private Connection conn;
   private PreparedStatement pstmt;
   private PreparedStatement pstmt2;
   private ResultSet rs;
   private String user = "root";
   private String password = "123456";
   private String dbUrl = "jdbc:mysql://localhost:3306/user?rewriteBatchedStatements=true";
   private int limitNum = 10000;

   public void changeData() {
       try {
           Class.forName("com.mysql.jdbc.Driver");
           conn = DriverManager.getConnection(dbUrl, user, password);
           
           //既不用batch,也不用事务
           testBatch(false,false);
           //只用batch, 不用事务
           testBatch(false,true);
           //只用事务，不用batch
           testBatch(true,false);
           //不仅用事务，还用batch
           testBatch(true,true);
           
           pstmt.close();
           conn.close();
       } catch (ClassNotFoundException e) {
           e.printStackTrace();
       } catch (SQLException e) {
           e.printStackTrace();
       }
   }
   
   public void testBatch(Boolean openTransaction, Boolean useBatch) throws SQLException{
       if(openTransaction)
           conn.setAutoCommit(false);
       
       if(pstmt!=null){
           pstmt.clearParameters();
           pstmt.clearBatch();
       }
       
       pstmt = conn.prepareStatement("insert into person (name) values (?)");
       long start = System.currentTimeMillis();
       for(int a = 0;a<limitNum;a++){
           String name = "tommy"+a;
           pstmt.setString(1, name);
           if(useBatch)
               pstmt.addBatch();
           else
               pstmt.executeUpdate();
       }
       
       if(useBatch)
          pstmt.executeBatch();
        
       if(openTransaction){
           conn.commit();
           conn.setAutoCommit(true);
       }
       long end = System.currentTimeMillis();
       System.out.println("use time:"+(end-start)+" ms");
       
   }
   
   //main method
   publi static void main(String[] args){
       ExecuteBatchTest ebt = new ExecuteBatchTest();
       ebt.changeData();
   }
      
}
			