MyBatis Plus 增强批量更新插入删除

import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.baomidou.mybatisplus.core.metadata.TableInfo;

import java.util.List;

public class SqlInjectorPlus extends DefaultSqlInjector {
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
        //继承原来的通用方法
        List<AbstractMethod> methods = super.getMethodList(mapperClass, tableInfo);
        methods.add(new EnhanceBatchInsert());
        return methods;
    }
}


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;
import org.springframework.transaction.annotation.EnableTransactionManagement;

/**
 * @author kong
 */
@Configuration
@MapperScan("com.jianmu.mapper*")
@EnableTransactionManagement(proxyTargetClass = true)
public class MybatisPlusConfig {

    /**
     * 分页插件
     */
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
        return interceptor;
    }

    /**
     * 增强sql注入的Bean
     */
    @Bean
    public SqlInjectorPlus sqlInjectorPlus() {
        return new SqlInjectorPlus();
    }

}

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

import java.util.Collection;

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

    /**
     * 增强批量新增
     */
    Integer enhanceBatchInsert(@Param("list") Collection<T> rows);

    /**
     * 增强批量更新
     */
    Integer enhanceBatchUpdate(@Param("list") Collection<T> rows);


    /**
     * 增强批量删除
     */
    Integer enhanceBatchDelete(@Param("list") Collection<T> rows);
}

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.AllArgsConstructor;
import lombok.NoArgsConstructor;
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;

@AllArgsConstructor
@NoArgsConstructor
public class EnhanceBatchInsert extends AbstractMethod {

    @Setter
    @Accessors(chain = true)
    private Predicate<TableFieldInfo> predicate;

    /**
     * <script>
     * INSERT INTO s_user
     * (<if test="list.get(0).id!=null">id,</if>username,nickname)
     * VALUES
     * <foreach collection="list" item="et" separator=",">
     * (<if test="list.get(0).id!=null">#{et.id},</if>#{et.username},#{et.nickname})
     * </foreach>
     * </script>
     */
    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        KeyGenerator keyGenerator = new NoKeyGenerator();
        SqlMethod sqlMethod = SqlMethod.INSERT_ONE;
        List<TableFieldInfo> fieldList = tableInfo.getFieldList();
        //(<if test="list.get(0).id!=null">id,</if>username,nickname)
        String insertSqlColumn = "<if test=\"list.get(0).id!=null\">id,</if>" + this.filterTableFieldInfo(fieldList, predicate, TableFieldInfo::getInsertSqlColumn, EMPTY);
        String columnScript = LEFT_BRACKET + insertSqlColumn.substring(0, insertSqlColumn.length() - 1) + RIGHT_BRACKET;
        //(<if test="list.get(0).id!=null">#{et.id},</if>#{et.username},#{et.nickname})
        String insertSqlProperty = "<if test=\"list.get(0).id!=null\">#{et.id},</if>" + this.filterTableFieldInfo(fieldList, predicate, i -> i.getInsertSqlProperty(ENTITY_DOT), EMPTY);
        insertSqlProperty = LEFT_BRACKET + insertSqlProperty.substring(0, insertSqlProperty.length() - 1) + RIGHT_BRACKET;
        //item=ENTITY
        String valuesScript = SqlScriptUtils.convertForeach(insertSqlProperty, "list", null, ENTITY, COMMA);
        String keyProperty = null;
        String keyColumn = null;
        // 表包含主键处理逻辑,如果不包含主键当普通字段处理
        if (tableInfo.havePK()) {
            if (tableInfo.getIdType() == IdType.AUTO) {
                /* 自增主键 */
                keyGenerator = new Jdbc3KeyGenerator();
                keyProperty = tableInfo.getKeyProperty();
                keyColumn = tableInfo.getKeyColumn();
            } else {
                if (null != tableInfo.getKeySequence()) {
                    keyGenerator = TableInfoHelper.genKeyGenerator(getMethod(sqlMethod), 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);
        return this.addInsertMappedStatement(mapperClass, modelClass, getMethod(sqlMethod), sqlSource, keyGenerator, keyProperty, keyColumn);

    }

    @Override
    public String getMethod(SqlMethod sqlMethod) {
        return "enhanceBatchInsert";
    }
}

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.jianmu.config.mybatis.EnhancePlusMapper;
import com.jianmu.model.system.User;

/**
 * <p>
 * 用户Mapper 接口
 * </p>
 *
 * @author kong
 * @since 2022-02-17
 */
public interface UserMapper extends EnhancePlusMapper<User> {

}

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.jianmu.bean.base.BasePageParam;
import com.jianmu.bean.base.PageVO;
import com.jianmu.mapper.system.UserMapper;
import com.jianmu.model.system.User;
import com.jianmu.service.i.system.UserService;
import org.springframework.stereotype.Service;

import java.util.Collection;

/**
 * <p>
 * 用户服务实现类
 * </p>
 *
 * @author kong
 * @since 2022-02-17
 */
@Service
public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService {
    @Override
    public PageVO<User> page(BasePageParam param) {
        LambdaQueryWrapper<User> uw = Wrappers.lambdaQuery();
        uw.like(User::getNickname, param.getSearch());
        Page<User> page = page(new Page<>(param.getCurrent(), param.getSize()), uw);
        return new PageVO<User>().setPages(page.getPages())
                .setCurrent(page.getCurrent())
                .setData(page.getRecords())
                .setSize(page.getSize())
                .setTotal(page.getTotal());
    }


    @Override
    public boolean saveBatch(Collection<User> users) {
        return this.baseMapper.enhanceBatchInsert(users) > 0;
    }
}

    @PostMapping("/batch_save")
    @ResponseResult
    @ApiOperation(value = "用户批量保存", notes = "save")
    public Boolean batchSave(@RequestBody @Validated @Size(min = 1) List<User> users) {
        return userService.saveBatch(users);
    }

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

import java.util.List;

@NoArgsConstructor
public class EnhanceBatchUpdate extends AbstractMethod {

    /**
     * update s_access_log
     * <trim prefix="set" suffixOverrides=",">
     * <if test="list.get(0).description!=null">
     * <trim prefix="description=case" suffix="end,">
     * <foreach collection="list" item="et">
     * <if test="et!=null"> when id=#{et.id} then #{et.description}</if>
     * </foreach>
     * </trim>
     * </if>
     * </trim>
     * where id in
     * <foreach collection="list" item="et" separator="," open="(" close=")">#{et.id}</foreach>
     */
    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        StringBuilder sb = new StringBuilder();
        List<TableFieldInfo> fields = tableInfo.getFieldList();

        fields.forEach(i -> sb.append("")
                .append("<if test=\"list.get(0).").append(i.getProperty()).append("!=null\">")
                .append("<trim prefix=\"")
                .append(i.getColumn())
                .append("=case\" suffix=\"end,\">")
                .append("<foreach collection=\"list\" item=\"et\">")
                .append("<if test=\"et!=null\">")
                .append(" when id=#{et.id} then #{et.").append(i.getProperty()).append("}")
                .append("</if>")
                .append("</foreach>")
                .append("</trim>")
                .append("</if>"));

        final String sql = "<script>update " + tableInfo.getTableName()
                + "<trim prefix=\"set\" suffixOverrides=\",\">"
                + sb
                + "</trim>"
                + "where id in"
                + "<foreach collection=\"list\" item=\"et\" separator=\",\" open=\"(\" close=\")\">"
                + "#{et.id}"
                + "</foreach>"
                + "</script>";


        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
        final String id = "enhanceBatchUpdate";
        return this.addUpdateMappedStatement(mapperClass, modelClass, id, sqlSource);
    }


}

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

@NoArgsConstructor
public class EnhanceBatchDelete extends AbstractMethod {

    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        String sql = "<script>delete from " + tableInfo.getTableName() + " where id in(<foreach collection=\"list\" item=\"i\" separator=\",\">#{i}</foreach>)</script>";
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
        String id = "enhanceBatchDelete";
        return this.addDeleteMappedStatement(mapperClass, id, sqlSource);
    }

}