mybatis批量插入、mybatis-plus批量插入、mybatis实现insertList、mybatis自定义实现批量插入

avatar
作者
猴君
阅读量:0

文章目录


众所周知,mybatisplus提供的BaseMapper里只有单条插入的方法,没有批量插入的方法,
而在Service层的批量插入并不是真的批量插入,实际上是遍历insert,但也不是一次insert就一次IO,而是到一定数量才会去IO一次,性能不是很差,但也不够好。

怎么才能实现真正的批量插入呢?

这里是mybatisplus官方的演示仓库,可以先去了解一下。

一、mybatis新增批量插入

1.1、引入依赖

本文基于mybatis-plus3.5.0进行讲解,不同的版本写法稍微有点不同。

<!--mybatis-plus--> <dependency>      <groupId>com.baomidou</groupId>      <artifactId>mybatis-plus-boot-starter</artifactId>      <version>3.5.0</version>  </dependency> 

1.2、自定义通用批量插入Mapper

1.新建一个InsertListMapper文件,提供insertList方法,后续需要用到批量插入的dao继承该Mapper即可拥有批量插入接口。

import org.apache.ibatis.annotations.Param; import java.util.List;  /**  * 通用Mapper接口,特殊方法,批量插入,支持批量插入的数据库都可以使用,例如mysql,h2等  *  * @param <T> 不能为空  * @author  */ public interface InsertListMapper<T> {     /**      * 批量插入,支持批量插入的数据库可以使用,例如MySQL,H2等      * <p>      * 注意:      * 1.若实体类中设置了@TableId注解,则需要设置value值  如:@TableId(value = "id")      * 2.若建表语句中设置了default, 需要在插入时手动设置默认值,否则存储的是null。      * 如:delete_flag tinyint default 0 comment '删除标志 0-未删除 1-已删除',这种需要在insert的时候设置实体类改字段值。setDeleteFlag(0)      * <p>      *      * @param recordList      * @return      */     int insertList(@Param("list") List<T> recordList); } 

1.3、把通用方法注册到mybatisplus注入器中

import com.baomidou.mybatisplus.core.injector.AbstractMethod; import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector; import com.baomidou.mybatisplus.core.metadata.TableInfo; import org.springframework.stereotype.Component;  import java.util.List;  /**  * 添加Sql注入方法,支持空字段更新  */ @Component public class CustomerSqlInjector extends DefaultSqlInjector {     @Override     public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo){         List<AbstractMethod> methodList=super.getMethodList(mapperClass, tableInfo);         //注册自定义方法         //注意:InsertList中的name需要与xxxMapper中的方法名一致,即insertList         methodList.add(new InsertList("insertList"));          return methodList;     } } 

关键的一句在于methodList.add(new InsertList("insertList"));,意为注册一个新的方法叫insertList,具体实现在InsertList类。

1.4、实现InsertList类

import com.baomidou.mybatisplus.core.injector.AbstractMethod; import com.baomidou.mybatisplus.core.metadata.TableInfo; import org.apache.ibatis.executor.keygen.KeyGenerator; import org.apache.ibatis.executor.keygen.NoKeyGenerator; import org.apache.ibatis.mapping.MappedStatement; import org.apache.ibatis.mapping.SqlSource; import java.util.Collections;  public class InsertList extends AbstractMethod {      public InsertList(String name) {         super(name);     }      @Override     public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {         KeyGenerator keyGenerator = new NoKeyGenerator();         SqlSource sqlSource = languageDriver.createSqlSource(configuration, getBatchInsertSql(tableInfo,modelClass), Collections.class);         return this.addInsertMappedStatement(mapperClass,modelClass,"insertList",sqlSource,keyGenerator,null,null);      }      private String getBatchInsertSql(TableInfo tableInfo, Class<?> modelClass){         String batchInsertSql="<script> INSERT INTO %s (%s) values %s</script>";          //要插入的字段 即insert into table(要插入的字段) values         StringBuilder insertColumnSql=new StringBuilder();         insertColumnSql.append(tableInfo.getKeyColumn()).append(",");          StringBuilder valueSql=new StringBuilder();         valueSql.append("<foreach collection='list' item='item' open='(' separator='),(' close=')' >\n");         valueSql.append("#{item."+tableInfo.getKeyProperty()+"},");          tableInfo.getFieldList().forEach(x->{             insertColumnSql.append(x.getColumn()).append(",");             valueSql.append("#{item."+x.getProperty()+"},");         });          insertColumnSql.delete(insertColumnSql.length()-1,insertColumnSql.length());         valueSql.delete(valueSql.length()-1,valueSql.length());         valueSql.append("</foreach>");          return  String.format(batchInsertSql,tableInfo.getTableName(),insertColumnSql,valueSql);     }  } 

1.5、需要批量插入的dao层继承批量插入Mapper

import com.baomidou.mybatisplus.core.mapper.BaseMapper; import com.demo.common.mapper.InsertListMapper; import com.demo.entity.User;  /** * @Entity generator.domain.User */ public interface UserMapper extends BaseMapper<User>, InsertListMapper<User> {   } 

以上代码全都放在gitee仓库,详情见:
代码仓库地址

二、可能遇到的问题

2.1、Invalid bound statement

org.apache.ibatis.binding.BindingException: Invalid bound statement (not found): com.demo.mapper.User1Mapper.insertList  	at org.apache.ibatis.binding.MapperMethod$SqlCommand.<init>(MapperMethod.java:235) 	at com.baomidou.mybatisplus.core.override.MybatisMapperMethod.<init>(MybatisMapperMethod.java:50) 	at com.baomidou.mybatisplus.core.override.MybatisMapperProxy.lambda$cachedInvoker$0(MybatisMapperProxy.java:111) 	at java.util.concurrent.ConcurrentHashMap.computeIfAbsent(ConcurrentHashMap.java:1660) 	at com.baomidou.mybatisplus.core.toolkit.CollectionUtils.computeIfAbsent(CollectionUtils.java:115) 	at com.baomidou.mybatisplus.core.override.MybatisMapperProxy.cachedInvoker(MybatisMapperProxy.java:98) 	at com.baomidou.mybatisplus.core.override.MybatisMapperProxy.invoke(MybatisMapperProxy.java:89) 	at com.sun.proxy.$Proxy70.insertList(Unknown Source) 	at com.demo.User1Test.test1(User1Test.java:50) 	at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method) 	at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62) 	at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43) 	at java.lang.reflect.Method.invoke(Method.java:498) 	at org.junit.runners.model.FrameworkMethod$1.runReflectiveCall(FrameworkMethod.java:50) 	at org.junit.internal.runners.model.ReflectiveCallable.run(ReflectiveCallable.java:12) 	at org.junit.runners.model.FrameworkMethod.invokeExplosively(FrameworkMethod.java:47) 	at org.junit.internal.runners.statements.InvokeMethod.evaluate(InvokeMethod.java:17) 

解决方法:https://blog.csdn.net/weixin_49114503/article/details/140645048







参考文章:https://blog.csdn.net/m0_51390969/article/details/134730527
https://blog.csdn.net/m0_63297646/article/details/131843517

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!