Mybatis批量插入或更新的正确姿势

明明如月学长

发布于 2021-08-27 16:28:51

1.8K0

发布于 2021-08-27 16:28:51

文章被收录于专栏：明明如月的技术专栏

一、背景

最近业务中用到批量插入或更新，查了一下资料。

另外一篇博客是对本文的补充，也可以参考一下：https://cloud.tencent.com/developer/article/1868954

二、搜索

2.1 第一种写法

其中stackoverflow中这个回答给了我很大启发。

https://stackoverflow.com/questions/23486547/mybatis-batch-insert-update-for-oracle

就是在映射的方法中拼接多条插入或者更新语句

1 批量更新

    UPDATE parties SET attending_user_count = #{model.attending_count}
    WHERE  fb_party_id = #{model.eid}

2 批量插入

    INSERT INTO accountability_users 
        (
            accountability_user_id, accountability_id, to_username,
            record_status, created_by, created_at, updated_by, updated_at
        ) 
    VALUES
     
        (           
            #{model.accountabilityUserId}, #{model.accountabilityId}, #{model.toUsername}, 
            'A', #{model.createdBy}, #{model.createdAt}, #{model.updatedBy}, #{model.updatedAt}     
        )

参考

https://www.cnblogs.com/yangy608/p/3912580.html

所述的MySQL和Oracle的批量插入区别可以看出可能有兼容性问题（使用Oracle 的同学重点参考下）

而且jdbc链接Url要加上 allowMultiQueries=true参数。

2.2 第二种写法

还可以采用下面的写法：

https://github.com/mybatis/mybatis-3/blob/master/src/test/java/org/apache/ibatis/submitted/batch_keys/BatchKeysTest.java

mybatis官方示例所示的方式来批量插入或者更新应该是最正确的用法。

 @Test
  public void testInsertWithMapper() {
    try (SqlSession sqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH)) {
      Mapper userMapper = sqlSession.getMapper(Mapper.class);
      User user1 = new User(null, "Pocoyo");
      userMapper.insert(user1);
      User user2 = new User(null, "Valentina");
      userMapper.insert(user2);
      sqlSession.flushStatements();
      assertEquals(new Integer(50), user1.getId());
      assertEquals(new Integer(50), user2.getId());
      sqlSession.commit();
    }

    try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
      List users = sqlSession.selectList("select");
      Assert.assertTrue(users.size() == 2);
    }
  }

另外openSession方法第二个参数是设置是否自动提交，可以设置为flase。

这里提到拿到批量更新结果的方式支持拿到批量结果。

另外这里提供批量执行的结果的打印函数

public void printBatchResults(List results) {
 if (results.size() > 0) {
  StringBuilder sb = new StringBuilder();
  sb.append("Batch summary:\n");
  for (int i = 0; i < results.size(); i++) {
   BatchResult result = results.get(i);
   sb.append("Result ").append(i).append(":\t");
   sb.append(result.getSql().replaceAll("\n", "").replaceAll("\\s+", " ")).append("\t");
   sb.append("Update counts: ").append(Arrays.toString(result.getUpdateCounts())).append("\n");
  }
  logDebug("082", sb.toString());
 }
}