【实用工具】Mybatis Plus扩展通用Mapper，实现批量插入和批量更新

秋装什么

已于 2022-12-27 00:20:54 修改

阅读量1.5k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： mybatis 文章标签： mybatis 数据库

于 2022-12-15 16:43:44 首次发布

本文链接：https://blog.csdn.net/qq_42985872/article/details/128331992

mybatis 专栏收录该内容

8 篇文章

订阅专栏

本文介绍如何使用MybatisPlus扩展通用Mapper，实现批量插入和批量更新功能。通过自定义Mapper接口，结合自定义批量插入和更新的方法，以及SQL注入器，实现高效的数据批量处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Mybatis Plus扩展通用Mapper，实现批量插入和批量更新

自定义Mapper

public interface RootMapper<T> extends BaseMapper<T> {

    int insertBatch(@Param("list") List<T> list);

    int updateBatch(@Param("list") List<T> list);
}

自定义批量插入方法

@Slf4j
public class InsertBatchMethod extends AbstractMethod {

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

    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        final String sql = "<script>insert into %s %s values %s</script>";
        final String fieldSql = prepareFieldSql(tableInfo);
        final String valueSql = prepareValuesSql(tableInfo);
        final String sqlResult = String.format(sql, tableInfo.getTableName(), fieldSql, valueSql);
        log.debug("sqlResult----->{}", sqlResult);
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sqlResult, modelClass);
        // 第三个参数必须和RootMapper的自定义方法名一致
        return this.addInsertMappedStatement(mapperClass, modelClass, "insertBatch", sqlSource, new NoKeyGenerator(), null, null);
    }

    private String prepareFieldSql(TableInfo tableInfo) {
        StringBuilder fieldSql = new StringBuilder();
        fieldSql.append(tableInfo.getKeyColumn()).append(",");
        fieldSql.append(tableInfo.getFieldList().stream().map(TableFieldInfo::getColumn).collect(Collectors.joining(",")));
        fieldSql.insert(0, "(");
        fieldSql.append(")");
        return fieldSql.toString();
    }

    private String prepareValuesSql(TableInfo tableInfo) {
        final StringBuilder valueSql = new StringBuilder();
        valueSql.append("<foreach collection=\"list\" item=\"item\" index=\"index\" open=\"(\" separator=\"),(\" close=\")\">");
        valueSql.append("#{item.").append(tableInfo.getKeyProperty()).append("},");
        valueSql.append(tableInfo.getFieldList().stream().map(item -> {
            return "#{item." + item.getProperty() + "}";
        }).collect(Collectors.joining(",")));
        valueSql.append("</foreach>");
        return valueSql.toString();
    }
}

自定义批量更新方法

@Slf4j
public class UpdateBatchMethod extends AbstractMethod {

    public UpdateBatchMethod() {
        super("updateBatch");
    }

    //update user set name = "a", age = 17 where id = 1;
    @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);
        log.debug("sqlResult----->{}", sqlResult);
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sqlResult, modelClass);
        // 第三个参数必须和RootMapper的自定义方法名一致
        return this.addUpdateMappedStatement(mapperClass, modelClass, "updateBatch", sqlSource);
    }
}

自定义方法注入器

public class CustomizedSqlInjector extends DefaultSqlInjector {

    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass, tableInfo);
        methodList.add(new InsertBatchMethod());
        return methodList;
    }
}

配置MybatisPlus，使SQL注入器生效

@Configuration
public class MybatisConfig {

    @Bean
    public CustomizedSqlInjector customizedSqlInjector() {
        return new CustomizedSqlInjector();
    }
}

备注：以下是当时所用的 Mybatis Plus 引用依赖版本

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