Mybatis Plus用自定义新增，来实现真正的批量新增方式

最新推荐文章于 2024-05-07 13:29:32 发布
最新推荐文章于 2024-05-07 13:29:32 发布
阅读量390 收藏
点赞数
文章标签： mybatis
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/he_lei/article/details/134786361
版权

编写自定义insert和update的SQL 批量处理方法
insert

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.enums.SqlMethod;
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.metadata.TableFieldInfo;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.baomidou.mybatisplus.core.toolkit.sql.SqlScriptUtils;
import lombok.Setter;
import lombok.experimental.Accessors;
import org.apache.ibatis.executor.keygen.Jdbc3KeyGenerator;
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.List;
import java.util.function.Predicate;


/**
 * @author aaa
 * @date 2023-09-27 18:27
 * @description
 */
public class InsertBatchColumn extends AbstractMethod {

    /**
     * 字段筛选条件
     */
    @Setter
    @Accessors(chain = true)
    private Predicate<TableFieldInfo> predicate;

    public InsertBatchColumn() {
        super("insertBatch");
    }

    public InsertBatchColumn(Predicate<TableFieldInfo> predicate) {
    	// 此处的名称必须与后续的RootMapper的新增方法名称一致
        super("insertBatch");
        this.predicate = predicate;
    }

    @SuppressWarnings("Duplicates")
    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        KeyGenerator keyGenerator = NoKeyGenerator.INSTANCE;
        SqlMethod sqlMethod = SqlMethod.INSERT_ONE;
        List<TableFieldInfo> fieldList = tableInfo.getFieldList();
        String insertSqlColumn = tableInfo.getKeyInsertSqlColumn(true,false) +
                this.filterTableFieldInfo(fieldList, predicate, TableFieldInfo::getInsertSqlColumn, EMPTY);
        String columnScript = LEFT_BRACKET + insertSqlColumn.substring(0, insertSqlColumn.length() - 1) + RIGHT_BRACKET;
        String insertSqlProperty = tableInfo.getKeyInsertSqlProperty(true,ENTITY_DOT, false) +
                this.filterTableFieldInfo(fieldList, predicate, i -> i.getInsertSqlProperty(ENTITY_DOT), EMPTY);
        insertSqlProperty = LEFT_BRACKET + insertSqlProperty.substring(0, insertSqlProperty.length() - 1) + RIGHT_BRACKET;
        String valuesScript = SqlScriptUtils.convertForeach(insertSqlProperty, "list", null, ENTITY, COMMA);
        String keyProperty = null;
        String keyColumn = null;
        // 表包含主键处理逻辑,如果不包含主键当普通字段处理
        if (tableInfo.havePK()) {
            if (tableInfo.getIdType() == IdType.AUTO) {
                /* 自增主键 */
                keyGenerator = Jdbc3KeyGenerator.INSTANCE;
                keyProperty = tableInfo.getKeyProperty();
                keyColumn = tableInfo.getKeyColumn();
            } else {
                if (null != tableInfo.getKeySequence()) {
                    keyGenerator = TableInfoHelper.genKeyGenerator(modelClass.getName(), tableInfo, builderAssistant);
                    keyProperty = tableInfo.getKeyProperty();
                    keyColumn = tableInfo.getKeyColumn();
                }
            }
        }
        String sql = String.format(sqlMethod.getSql(), tableInfo.getTableName(), columnScript, valuesScript);
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
		// 注意第三个参数，需要与后续的RootMapper里面新增方法名称要一致，不然会报无法绑定异常
        return this.addInsertMappedStatement(mapperClass, modelClass, "insertBatch", sqlSource, keyGenerator, keyProperty, keyColumn);

    }
}

update

import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlSource;


/**
 * @author aaa
 * @date 2023-09
 * @description
 */
public class UpdateBatchColumn extends AbstractMethod {

    /**
     * @param methodName 方法名
     * @since 3.5.0
     */
    public UpdateBatchColumn(String methodName) {
        super(methodName);
    }

    @SuppressWarnings("Duplicates")
    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        String sql = "<script>\n<foreach collection=\"list\" item=\"item\" separator=\";\">\nupdate %s %s where %s=#{%s} %s\n</foreach>\n</script>";
        String additional = tableInfo.isWithVersion() ? tableInfo.getVersionFieldInfo().getVersionOli("item", "item.") : "" + tableInfo.getLogicDeleteSql(true, true);
        String setSql = sqlSet(tableInfo.isWithLogicDelete(), false, tableInfo, false, "item", "item.");
        String sqlResult = String.format(sql, tableInfo.getTableName(), setSql, tableInfo.getKeyColumn(), "item." + tableInfo.getKeyProperty(), additional);
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sqlResult, modelClass);
        // 第三个参数必须和RootMapper的自定义方法名一致
        return this.addUpdateMappedStatement(mapperClass, modelClass, "updateBatch", sqlSource);


    }


}

2,创建一个 SQL 注入器 InsertBatchSqlInjector

import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.example.ftserver.plugin.InsertBatchColumn;
import com.example.ftserver.plugin.UpdateBatchColumn;

import java.util.List;

/**
 * @author aaa
 * @description
 */
public class CustomizedSqlInjector extends DefaultSqlInjector {


    public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
        List<AbstractMethod> methods = super.getMethodList(mapperClass,tableInfo);
        // 自定义的insert SQL注入器
        methods.add(new InsertBatchColumn());
        // 自定义的update SQL注入器，参数需要与RootMapper的批量update名称一致
        methods.add(new UpdateBatchColumn("updateBatch")); 
        return methods;
    }


}

3,将SQL 注入器放入配置类

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * @author aaa
 * @description
 */
@Configuration
@MapperScan(basePackages = "com.example.ftserver.mapper") // 不用配置.*
public class MybatisPlusConfig {

    @Bean
    public CustomizedSqlInjector sqlInjector(){
        return new CustomizedSqlInjector();
    }
    /**
     * 添加分页插件
     */
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));//如果配置多个插件,切记分页最后添加
        return interceptor;
    }
}

4,新建一个在 config 包下创建 RootMapper 接口，让其继承自 Mybatis Plus 提供的 BaseMapper, 并定义批量插入方法

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import org.apache.ibatis.annotations.Param;

import java.util.Collection;

/**
 * @author aaa
 * @date 2023-09-28 16:11
 * @description
 */
public interface RootMapper<T> extends BaseMapper<T> {
	// 新增方法
    int insertBatch(@Param("list")Collection<T> batchList);
    // 修改方法
    int updateBatch(@Param("list")Collection<T> batchList);
}

5,新建 自己的Mapper，注意继承刚刚自定义的 RootMapper, 而不是Mybatis plus的 BaseMapper :

import com.test.ft.common.entity.PersonEntity;
import com.example.ftserver.plugin.RootMapper;
import org.apache.ibatis.annotations.Mapper;

/**
 * <p>
 *  Mapper 接口
 * </p>
 *
 * @author aaa
 * @since 2023-10-19 16:32:39
 */
public interface PersonMapper extends RootMapper<PersonEntity> {

}

批量修改会报错，Mybatis映射文件中的sql语句默认是不支持以" ; " 结尾，也就是不支持多条sql语句的执行。需要在连接mysql的url上加 &allowMultiQueries=true 运行批量操作才可以执行。

如果数据库设置了默认值，用批量添加汇会存null，修改如下：

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.enums.SqlMethod;
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.metadata.TableFieldInfo;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.baomidou.mybatisplus.core.toolkit.sql.SqlScriptUtils;
import lombok.Setter;
import lombok.experimental.Accessors;
import org.apache.ibatis.executor.keygen.Jdbc3KeyGenerator;
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.List;
import java.util.function.Predicate;


/**
 * @author aaa
 * @date 2023-09-27 18:27
 * @description
 */
public class InsertBatchColumn extends AbstractMethod {

    /**
     * 字段筛选条件
     */
    @Setter
    @Accessors(chain = true)
    private Predicate<TableFieldInfo> predicate;

    public InsertBatchColumn() {
        super("insertBatch");
    }

    public InsertBatchColumn(Predicate<TableFieldInfo> predicate) {
        // 此处的名称必须与后续的RootMapper的新增方法名称一致
        super("insertBatch");
        this.predicate = predicate;
    }

    @SuppressWarnings("Duplicates")
    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        KeyGenerator keyGenerator = NoKeyGenerator.INSTANCE;
        SqlMethod sqlMethod = SqlMethod.INSERT_ONE;
        List<TableFieldInfo> fieldList = tableInfo.getFieldList();
        String insertSqlColumn = tableInfo.getKeyInsertSqlColumn(true, false) +
                this.filterTableFieldInfo(fieldList, predicate, TableFieldInfo::getInsertSqlColumn, EMPTY);
        String columnScript = LEFT_BRACKET + insertSqlColumn.substring(0, insertSqlColumn.length() - 1) + RIGHT_BRACKET;
/*        String insertSqlProperty = tableInfo.getKeyInsertSqlProperty(true, ENTITY_DOT, false) +
                this.filterTableFieldInfo(fieldList, predicate, i -> i.getInsertSqlProperty(ENTITY_DOT), EMPTY);*/
        String insertSqlProperty = tableInfo.getKeyInsertSqlProperty(true, ENTITY_DOT, false) +
                this.filterTableFieldInfo(fieldList, predicate, i -> getInsertSqlProperty(i, ENTITY_DOT), EMPTY);
        insertSqlProperty = LEFT_BRACKET + insertSqlProperty.substring(0, insertSqlProperty.length() - 1) + RIGHT_BRACKET;
        String valuesScript = SqlScriptUtils.convertForeach(insertSqlProperty, "list", null, ENTITY, COMMA);
        String keyProperty = null;
        String keyColumn = null;
        // 表包含主键处理逻辑,如果不包含主键当普通字段处理
        if (tableInfo.havePK()) {
            if (tableInfo.getIdType() == IdType.AUTO) {
                /* 自增主键 */
                keyGenerator = Jdbc3KeyGenerator.INSTANCE;
                keyProperty = tableInfo.getKeyProperty();
                keyColumn = tableInfo.getKeyColumn();
            } else {
                if (null != tableInfo.getKeySequence()) {
                    keyGenerator = TableInfoHelper.genKeyGenerator(modelClass.getName(), tableInfo, builderAssistant);
                    keyProperty = tableInfo.getKeyProperty();
                    keyColumn = tableInfo.getKeyColumn();
                }
            }
        }
        String sql = String.format(sqlMethod.getSql(), tableInfo.getTableName(), columnScript, valuesScript);
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
        // 注意第三个参数，需要与后续的RootMapper里面新增方法名称要一致，不然会报无法绑定异常
        return this.addInsertMappedStatement(mapperClass, modelClass, "insertBatch", sqlSource, keyGenerator, keyProperty, keyColumn);

    }

    private String getInsertSqlProperty(TableFieldInfo tableFieldInfo, final String prefix) {
        String newPrefix = prefix == null ? "" : prefix;
        String elPart = SqlScriptUtils.safeParam(newPrefix + tableFieldInfo.getEl());
        //属性为空时使用默认值
        String result = SqlScriptUtils.convertIf(elPart,
                String.format("%s != null", newPrefix + tableFieldInfo.getEl()), false)
                + SqlScriptUtils.convertIf("default",
                String.format("%s == null", newPrefix + tableFieldInfo.getEl()), false);
        return result + ",";
    }

}

总结
以上就是MyBatis plus对MySQL实现批量新增，批量修改的内容，才疏学浅，如有错误，希望多多的指正。需要注意的是：**该方法只对MySQL测试过，其他的数据库支持度不一致，不一定适用

cxyhl
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
mybatis-plus配置分页插件和内选插件(批量更新、批量插入)
DN金猿的博客
07-27 322
channel.basicReject(deliveryTag, true); basic.reject方法拒绝deliveryTag对应的消息，第二个参数是否requeue，true则重新入队列，否则丢弃或者进入死信队列。 该方法reject后，该消费者还是会消费到该条被reject的消息。 channel.basicNack(deliveryTag, false, true); basic.nack方法为不确认deliveryTag对应的消息，第二个参数是否应...
Mybatis Plus 自定义批量插入或批量更新（根据唯一索引）
01-08
Mybatis Plus 自定义批量插入或批量更新（根据唯一索引）
Mybatis-Plus 自定义mapper批量新增修改函数
954L
08-21 740
mapper接口实现自定义的RootMapper，即可调用批量新增修改函数。
MybatisPlus自定义insertBatchSomeColumn实现真正批量插入
最新发布
weixin_42034623的博客
05-07 2256
MySQL版@OverrideOracle版//这里改成我们自己的实现MyInsertBatchSomeColumn methodList . add(new OracleInsertBatchSomeColumn(i -> i . getFieldFill()!} }
MyBatis-Plus 3 实现批量新增和批量修改
zhouzhiwengang的专栏
06-21 5183
是因为spring boot配置文件中，关于数据库的连接配置，并没有允许进行批量更新的操作。这种方式就是通过SQL拼接，单条单条的进行更新，如果行数多的情况下给不建议使用。在数据库连接配置中必须添加 &allowMultiQueries=true，通过case when语句进行批量更新，只要一条SQL语句.mapper xml: 批量插入xml。mapper 接口:批量方法插入。mapper 接口:批量方法插入。mapper xml: 分页查询。mapper 接口:分页查询。- 页面显示条数 size。
优化mybatisPlus批量新增，新增mapper层批量新增方法，附带代码生成vm模板。
XuDream的博客
10-23 2546
经发现，mybatisPlus只有Service中存在saveBatch()方法，而Mapper中不存在批量新增的方法，深入了解，发现mybatisPlus的Service中saveBatch()方法也是循环一条一条插入数据库。上述mybatis-plus中源码中只是循环插入实现的批量新增，优化方式有以下几个思路。看到这里，这里提取Mapper.xml的方法，使代码生成的VM的自动生成（VM是mybatisPlus代码生成的模板）这里使用600条数据在模拟mybatisPlus新增和优化之后的新增。
mybatis-plus_batch_insert：mybatis_plus添加批量插入
02-14
总结，MyBatis-Plus 的批量插入功能是通过自定义 SQL 和设置执行器类型实现的。在实际应用中，应合理调整批量插入的大小和事务管理，以平衡性能和稳定性。在项目 `mybatis-plus_batch_insert-master` 中，可能包含了...
Mybatis plus增强工具包-其他
06-13
该工具包为MyBatis提供了一些高效，有用，即用的功能，使用它可以有效地节省您的开发时间。 Mybatis-plus特征： 与MyBatis完全兼容 启动时自动配置 开箱即用的用于操作数据库的界面 强大而灵活的条件包装器 生成主键...
mybatis-plus-demo.zip
09-15
MyBatis-Plus 是 MyBatis 的一个扩展，它提供了许多方便实用的功能，如自动 CRUD 操作、条件构造器、批量插入、主键自增、填充公共字段等，极大地简化了开发工作。在 SpringBoot 框架下集成 MyBatis-Plus，可以让...
Mybatis-Plus最新教程
11-14
Mybatis-Plus的核心理念是使用反射机制来扫描实体类，并将其作为数据库信息。 二、基本使用 1. 规定：在使用Mybatis-Plus之前，需要在实体类中添加相应的注解，例如@TableName、@TableId、@TableField等。 2. 添加...
【实用工具】Mybatis Plus扩展通用Mapper，实现批量插入和批量更新
秋装什么的博客
12-15 1296
【实用工具】Mybatis Plus扩展通用Mapper，实现批量插入和批量更新
MyBatis实现mysql、oracle批量插入
qq_39768738的博客
12-12 1166
一、mybatis insert标签属性　　　　　　　　　　　　　　　　　　　　　　　　　　 1、parameterType：入参的全限定类名或类型别名。 2、keyColumn：设置数据表自动生成的主键名。对特定数据库（如PostgreSQL），若自动生成的主键不是第一个字段则必须设置。 3、keyProperty：默认值unset，用于设置getGeneratedKeys方法或selec...
spring boot集成mybatis-plus——Mybatis Plus 批量 Insert_新增数据（图文讲解）
小白龙白龙马的博客
02-04 6885
Mybatis Plus 批量 Insert_新增数据(图文讲解) 更新时间 2023-01-10 16:02:58 前言 大家好，我是小哈。 本小节中，我们将学习如何通过 Mybatis Plus 实现 MySQL 批量插入数据。 什么是批量插入？优势在哪里？ 先抛出一个问题：假设老板给你下了个任务，向数据库中添加 100 万条数据，并且不能耗时太久！ 通常来说，我们向 MySQL 中新增...
Mybatis-plus批量操作
热门推荐
小哥哥的博客
08-19 3万+
使用Mybatis-plus可以很方便的实现批量新增和批量修改，不仅比自己写foreach遍历方便很多，而且性能也更加优秀。但是Mybatis-plus官方提供的批量修改和批量新增都是根据id来修改的，有时候我们需求其他字段，所以就需要我们自己修改一下
mybatisplus批量操作
qq_34787180的博客
06-16 7720
昨天太忙算是补发昨天的记录，每天一篇
mybatis数据批量更新
xieqj_0511的博客
07-23 1万+
数据批量更新
2022-05-05 mybatis-plus 批量插入修改操作
m0_54853503的博客
07-27 1459
第一步一般引入mybaits-plus都会有相应的配置类,MybatisPlusConfig名字无所谓,作用是一样的,一般都会用自带的分页插件,可以在此基础上,继续添加,给出我的配置。其中有个小小的问题,即IService中自带的saveBatch和saveOrUpdateBatch等方法,仔细看会发现,他们的批量执行,竟然不是真正的批量执行!第五步使用,将原有的继承BaseMapper的方法,改写为继承RootMapper,后续批量操作,直接使用新增的两个方法进行处理即可。...
mybatisPlus的sqlSessionTemplate批量修改（与mybatis道理一致）
weixin_45502734的博客
03-16 3611
一、简单步骤解析 ①引入SqlSessionTemplate，mapper接口 ②根据需求写一个更新数据的批量处理方法（关掉自动提交，批次commit） ③将需要修改的数据对象list，传入批处理方法中处理（实际相当于传一个update的where条件内容） ④调用接口进行数据更新 ⑤测试+日志+数据展示 二、代码模块示例 ①引入SqlSessionTemplate，mapper接口 @Autowired private SqlSessionTemplate sqlSe
mybatisPlus 联合主键 批量修改
xl_21的博客
03-26 3450
mybatisPlus 联合主键 批量修改 maven依赖 <dependency> <groupId>com.github.jeffreyning</groupId> <artifactId>mybatisplus-plus</artifactId> <version>1.5.1-RELEASE</version> </depen
MyBatis Plus基于自定义BaseMapper来添加批量插入
06-01
是的，MyBatis Plus提供了一个非常方便的批量插入方法，即使用自定义BaseMapper来添加批量插入。 以下是具体实现步骤： 1. 首先，创建一个自定义的BaseMapper类，继承MyBatis Plus提供的BaseMapper类，添加批量插入的方法。 ```java import com.baomidou.mybatisplus.core.mapper.BaseMapper; import org.apache.ibatis.annotations.Param; import java.util.List; public interface MyBaseMapper<T> extends BaseMapper<T> { /** * 批量插入 * * @param list 数据列表 * @return 插入的行数 */ int insertBatch(@Param("list") List<T> list); } ``` 2. 在Mapper接口中继承自定义的BaseMapper接口。 ```java @Repository public interface UserMapper extends MyBaseMapper<User> { } ``` 3. 在Mapper.xml文件中添加insertBatch方法的SQL语句。 ```xml <insert id="insertBatch" useGeneratedKeys="true" keyProperty="id"> INSERT INTO user (name, age, email) VALUES <foreach collection="list" item="item" separator=","> (#{item.name}, #{item.age}, #{item.email}) </foreach> </insert> ``` 4. 最后，在Service层调用批量插入的方法即可。 ```java @Service public class UserService { @Autowired UserMapper userMapper; public int insertBatch(List<User> userList) { return userMapper.insertBatch(userList); } } ``` 以上就是使用自定义BaseMapper实现MyBatis Plus的批量插入的具体实现步骤。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值