批量插入
一、继承IService(伪批量)
二、insertBatchSomeColumn
Mybatis-plus很强,为我们诞生了极简CURD操作,但对于数据批量操作,显然默认提供的insert方法是不够看的了,于是它和它来了!!! Mybatis-plus提供的两种插入方式
继承IService(伪批量)
insertBatchSomeColumn
在Mapper继承BaseMapper
import com.baomidou.mybatisplus.core.mapper.BaseMapper; import com.example.demo.entity.UserStudy; import org.apache.ibatis.annotations.Mapper; @Mapper public interface UserStudyMapper extends BaseMapper{ }
在Service中继承IService
import com.baomidou.mybatisplus.extension.service.IService; import com.example.demo.entity.UserStudy; /******************************************************************************** ** @author : ZYJ ** @date :2023/04/20 ** @description :厂长老婆催的睡觉了-批量插入Service *********************************************************************************/ public interface UserStudyService extends IService{ }
在Service实现类继承ServiceImpl
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.example.demo.entity.UserStudy; import com.example.demo.mapper.UserStudyMapper; import org.springframework.stereotype.Service; import javax.annotation.Resource; /******************************************************************************** ** @author : ZYJ ** @date :2023/04/04 ** @description :厂长加班写代码-批量插入 *********************************************************************************/ @Service public class UserStudyServiceImpl extends ServiceImplimplements UserStudyService { @Resource private UserStudyMapper userStudyMapper; }
测试代码,调用IService的saveBatch方法
/* *批量插入 */ @Override public void greatMany() { ListuserStudyList = new ArrayList<>(); UserStudy userStudy1 = new UserStudy(); userStudy1.setName("张三"); UserStudy userStudy2 = new UserStudy(); userStudy2.setName("李四"); userStudyList.add(userStudy1); userStudyList.add(userStudy2); //调用IService的saveBatch方法 this.saveBatch(userStudyList); }
Mybatis-plus的SQL日志打印在配置文件application.yml配置
mybatis-plus: configuration: log-impl: org.apache.ibatis.logging.stdout.StdOutImpl #开启sql日志 # log-impl: org.apache.ibatis.logging.nologging.NoLoggingImpl #关闭sql日志
测试结果,代码执行打印了两条SQL,所以可以看得出saveBatch底层也是遍历循环完成
saveBatch方法分析
底层也是通过for来完成,默认是一个事务一次提交1000条数据,点击进入saveBatch可以看到, 也可以自定义每次提交多少条,自定义如下
//调用IService的saveBatch方法 this.saveBatch(userStudyList,2000);
自定义SQL注入器
import com.baomidou.mybatisplus.core.injector.AbstractMethod; import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector; import com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn; import java.util.List; /******************************************************************************** ** @author : ZYJ ** @date :2023/03/09 ** @description :厂长加班写代码-批量插入SQL注入器 *********************************************************************************/ public class InsertBatchSqlInjector extends DefaultSqlInjector { @Override public List getMethodList(Class> mapperClass) { List methodList = super.getMethodList(mapperClass); methodList.add(new InsertBatchSomeColumn()); //添加InsertBatchSomeColumn方法 return methodList; } }
把SQL注入器交给Spring
import org.springframework.context.annotation.Bean; import org.springframework.stereotype.Component; @Component public class MybatisPlusConfig { /******************************************************************************** ** @author : ZYJ ** @date :2023/04/14 ** @description :注入配置 *********************************************************************************/ @Bean public InsertBatchSqlInjector easySqlInjector () { return new InsertBatchSqlInjector(); } }
到此定义完毕,在Mapper中生成insertBatchSomeColumn(必须是这个方法名)方法,你就可以撒手不管了,直接调用就行,或者直接在ServiceImpl通过Mapper调用insertBatchSomeColumn,然后ALT+回车生成此方法。
import com.baomidou.mybatisplus.core.mapper.BaseMapper; import com.example.demo.entity.UserStudy; import org.apache.ibatis.annotations.Mapper; import org.apache.ibatis.annotations.Param; import java.util.List; @Mapper public interface UserStudyMapper extends BaseMapper{ void insertBatchSomeColumn(@Param("list") List userStudyList); }
测试代码,调用insertBatchSomeColumn方法
@Resource private UserStudyMapper userStudyMapper; /* *批量插入 */ @Override public void greatMany() { ListuserStudyList = new ArrayList<>(); UserStudy userStudy1 = new UserStudy(); userStudy1.setName("张三"); UserStudy userStudy2 = new UserStudy(); userStudy2.setName("李四"); userStudyList.add(userStudy1); userStudyList.add(userStudy2); //调用insertBatchSomeColumn方法 userStudyMapper.insertBatchSomeColumn(userStudyList); //调用IService的saveBatch方法 //this.saveBatch(userStudyList,2000); }
测试结果,代码执行打印一条SQL,所以可以看出是一条SQL便新增完成
注意:SQL有语句长度限制,在MySQL中被参数max_allowed_packet限制,默认为1M,如果拼接长度超过此限制就会报错,两种解决方式,一个是调整MySQL的max_allowed_packet 限制,另一个则是通过代码控制每次的提交数量。
通过代码控制每次提交数量,模拟造五条数据,每次提交两条数据
/* *批量插入 */ @Override public void greatMany() { ListuserStudyList = new ArrayList<>(); UserStudy userStudy1 = new UserStudy(); userStudy1.setName("张三"); UserStudy userStudy2 = new UserStudy(); userStudy2.setName("李四"); UserStudy userStudy3 = new UserStudy(); userStudy3.setName("王五"); UserStudy userStudy4 = new UserStudy(); userStudy4.setName("赵六"); UserStudy userStudy5 = new UserStudy(); userStudy5.setName("小红"); userStudyList.add(userStudy1); userStudyList.add(userStudy2); userStudyList.add(userStudy3); userStudyList.add(userStudy4); userStudyList.add(userStudy5); //创建入库的list List userStudyCount = new ArrayList<>(); for (int i = 0; i < userStudyList.size(); i++) { //调用insertBatchSomeColumn方法 userStudyCount.add(userStudyList.get(i)); //控制每次提交数量 if(userStudyCount.size()==2){ userStudyMapper.insertBatchSomeColumn(userStudyCount); //将入库的list清空重新新增 userStudyCount.clear(); } } //将list中size不够2的数据在此处新增 userStudyMapper.insertBatchSomeColumn(userStudyCount); //调用IService的saveBatch方法 //this.saveBatch(userStudyList,2000); }
结果分析,五条数据应该请求三次新增,打印三条SQL,完美结束
总结:默认的insert的方法对寻常业务来说是非常之高效,但对于批量数据的产生确实灾难性的,就是慢,很慢,巨慢,IService的saveBatch方法优于默认的insert方法,但是我选通过SQL注入器的方法insertBatchSomeColumn。