?作為CRUD程序員，大部分Java開(kāi)發(fā)者應(yīng)該都在用Mybatis Plus來(lái)操作數(shù)據(jù)庫(kù)。但是BaseMapper默認(rèn)僅提供了int insert(T entity)這個(gè)單條插入的方法。那么我們想批量插入數(shù)據(jù)該怎么辦呢？

在以前用Mybatis的時(shí)候，我們會(huì)在Mapper.xml里面去寫foreach循環(huán):

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>

如果批量的實(shí)體對(duì)象較多的話，我們就會(huì)寫很多這樣的Mapper，如果表字段較多，這工作量的也不小的，當(dāng)然，如果你用了代碼生成工具，那就當(dāng)我沒(méi)說(shuō)。Mybatis Plus作為Mybatis的增強(qiáng)版，也為我們考慮到了這個(gè)問(wèn)題。使用Mybatis Plus批量插入數(shù)據(jù)有兩種方式，第一種是Service層繼承IService,第二種便是insertBatchSomeColumn。先來(lái)看看IService的方式

• 新建Mapper繼承BaseMapper<T>public interface UserMapper extends BaseMapper<UserDO> {}
• 新建一個(gè)Service接口，繼承IService<T>public interface UserService extends IService<UserDO> { void batchAddUser();}
• 新建一個(gè)Service實(shí)現(xiàn)類,繼承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); }}

這種方式，會(huì)生成多條INSERT語(yǔ)句一條條執(zhí)行，然后按設(shè)置的batchSize來(lái)提交事務(wù)。

我們?cè)賮?lái)看看insertBatchSomeColumn是怎么做的呢。1.新建一個(gè)InsertBatchInjector類來(lái)擴(kuò)展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.新建一個(gè)Mapper接口，繼承BaseMapper<T>并添加insertBatchSomeColumn方法，

1
2
3
4

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

3.在要批量寫的地方直接調(diào)用該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，一次性提交到數(shù)據(jù)庫(kù)執(zhí)行

這里，SQL語(yǔ)句的長(zhǎng)度受MySQL服務(wù)端的max_allowed_packet參數(shù)限制，默認(rèn)為1M，如果拼接的字符串過(guò)長(zhǎng)則會(huì)報(bào)錯(cuò)?？梢赃M(jìn)來(lái)減少批量提交的記錄，或者是調(diào)整max_allowed_packet參數(shù)的大小。除了受到max_allowed_packet的影響，innodb_buffer_pool_size的大小也會(huì)影響批量的記錄數(shù)大小，innodb_buffer_pool_size的默認(rèn)值為128M，我們可以根據(jù)情況適當(dāng)調(diào)整，合理地配置innodb_buffer_pool_size參數(shù)，可以提高M(jìn)ySQL的讀寫性能。