批量插入数据速度问题？

原文：https://www.cnblogs.com/tommy-huang/p/4540407.html

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

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

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

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

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

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()
          代码如下：

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