?作為CRUD程序員，大部分Java開發者應該都在用Mybatis Plus來操作數據庫。但是BaseMapper默認僅提供了int insert(T entity)這個單條插入的方法。那么我們想批量插入數據該怎么辦呢？

在以前用Mybatis的時候，我們會在Mapper.xml里面去寫foreach循環:

1
2
3
4
5
6
7
8

insert into user (user_name,create_time)
  values

   <foreach item="item" index="index" collection="list" separator=",">

        (#{list.userName},now())

    </foreach>

如果批量的實體對象較多的話，我們就會寫很多這樣的Mapper，如果表字段較多，這工作量的也不小的，當然，如果你用了代碼生成工具，那就當我沒說。Mybatis Plus作為Mybatis的增強版，也為我們考慮到了這個問題。使用Mybatis Plus批量插入數據有兩種方式，第一種是Service層繼承IService,第二種便是insertBatchSomeColumn。先來看看IService的方式

• 新建Mapper繼承BaseMapper<T>public interface UserMapper extends BaseMapper<UserDO> {}
• 新建一個Service接口，繼承IService<T>public interface UserService extends IService<UserDO> { void batchAddUser();}
• 新建一個Service實現類,繼承ServiceImpl<M,T>public class UserServiceImpl extends ServiceImpl<UserMapper,UserDO> implements UserService { @Autowired private UserMapper userMapper; @Override public void batchAddUser() { List<UserDO> userDOList=new ArrayList<>(); UserDO user1=new UserDO(); user1.setAge(99); user1.setName("99_1"); userDOList.add(user1); UserDO user2=new UserDO(); user2.setAge(999); user2.setName("999_1"); userDOList.add(user2); this.saveBatch(userDOList,2); }}

這種方式，會生成多條INSERT語句一條條執行，然后按設置的batchSize來提交事務。

我們再來看看insertBatchSomeColumn是怎么做的呢。1.新建一個InsertBatchInjector類來擴展DefaultSqlInjector

1
2
3
4
5
6
7
8
9

@Component
public class InsertBatchInjector extends DefaultSqlInjector  {
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass,tableInfo);
        methodList.add(new InsertBatchSomeColumn());
        return methodList;
    }
}

2.新建一個Mapper接口，繼承BaseMapper<T>并添加insertBatchSomeColumn方法，

1
2
3
4

@Mapper
public interface UserMapper extends BaseMapper<UserDO> {
    int insertBatchSomeColumn(List<UserDO> userDOList);
}

3.在要批量寫的地方直接調用該Mapper的insertBatchSomeColumn即可

1
2
3
4
5
6
7
8
9
10
11
12

public void batchAddUser() {
        List<UserDO> userDOList=new ArrayList<>();
        UserDO user1=new UserDO();
        user1.setAge(99);
        user1.setName("99_1");
        userDOList.add(user1);
        UserDO user2=new UserDO();
        user2.setAge(999);
        user2.setName("999_1");
        userDOList.add(user2);
        userMapper.insertBatchSomeColumn(userDOList);
}

這種方式則是拼接成一條SQL，一次性提交到數據庫執行

這里，SQL語句的長度受MySQL服務端的max_allowed_packet參數限制，默認為1M，如果拼接的字符串過長則會報錯。可以進來減少批量提交的記錄，或者是調整max_allowed_packet參數的大小。除了受到max_allowed_packet的影響，innodb_buffer_pool_size的大小也會影響批量的記錄數大小，innodb_buffer_pool_size的默認值為128M，我們可以根據情況適當調整，合理地配置innodb_buffer_pool_size參數，可以提高MySQL的讀寫性能。