Mybatis Plus批量插入数据到MySQL中

作为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的方式

1. 新建Mapper继承BaseMapper<T>
   public interface UserMapper extends BaseMapper<UserDO> {
   
   
   }
2. 新建一个Service接口，继承IService<T>
   public interface UserService extends IService<UserDO> {
   void batchAddUser();
   }
3. 新建一个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的读写性能。