雁过留声 发表于 2024-7-29 14:50:32

mybatis批量插入、mybatis-plus批量插入、mybatis实现insertList、mybatis

众所周知,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>");

      returnString.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

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: mybatis批量插入、mybatis-plus批量插入、mybatis实现insertList、mybatis