mybatis-plus批量更新updateBatchById問題

更新時(shí)間：2023年07月14日 10:38:52 作者：KevinPan_1992

這篇文章主要介紹了mybatis-plus批量更新updateBatchById問題，具有很好的參考價(jià)值，希望對大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

前言

在使用mybatis-plus過程中，有很多插件都特別優(yōu)秀，不僅使我們代碼更加優(yōu)雅，也提升了效率。

其中有個(gè)批量插入的插件insertBatchSomeColumn使用起來也挺方便的，但是批量更新一直沒有官方插件，網(wǎng)絡(luò)上面也沒有找到靠譜的，于是就參照mybatis-plus這些官方的方法自定義了一個(gè)批量更新的方法。

實(shí)現(xiàn)效果

案例：用戶排序

最終更新語句：

UPDATE sys_user 
SET user_order =
CASE
		id 
		WHEN 1 THEN	1 
		WHEN 2 THEN	2 
		WHEN 3 THEN	3 
		WHEN 4 THEN	4 
END 
WHERE tenant_id = 1 
AND id IN (1,2,3,4)

批量新增插件的配置

定義一個(gè)自己的BaseMapper繼承自mybatis-plus的BaseMapper，聲明批量新增方法，如下：

public interface MyBaseMapper<T> extends BaseMapper<T> {
    /**
     * 批量插入
     *
     * @param entityList 實(shí)體列表
     * @return 影響行數(shù)
     */
    int insertBatchSomeColumn(Collection<T> entityList);
}

把批量新增方法添加到方法列表中：

/**
 * 通用方法注入
 */
public class MySqlInjector extends DefaultSqlInjector {
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass);
        // 添加批量新增方法
        methodList.add(new InsertBatchSomeColumn());
        return methodList;
    }
}

MybatisPlusConfig配置中注入bean：

    @Bean
    public MySqlInjector mySqlInjector() {
        return new MySqlInjector();
    }

業(yè)務(wù)Mapper繼承自自定義的MyBaseMapper，則就可以使用批量新增方法了。

下面進(jìn)入正題

updateBatchById實(shí)現(xiàn)

自定義方法枚舉

參照官方的SqlMethod，創(chuàng)建枚舉MySqlMethod，并定義批量更新方法，如下：

public enum MySqlMethod {
    UPDATE_BATCH_BY_ID("updateBatchById", "通過主鍵批量更新數(shù)據(jù)", "<script>UPDATE %s \n%s \nWHERE %s IN %s\n</script>");
    private final String method;
    private final String desc;
    private final String sql;
    MySqlMethod(String method, String desc, String sql) {
        this.method = method;
        this.desc = desc;
        this.sql = sql;
    }
    public String getMethod() {
        return this.method;
    }
    public String getDesc() {
        return this.desc;
    }
    public String getSql() {
        return this.sql;
    }
}

自定義批量更新方法

定義UpdateBatchById繼承自AbstractMethod，實(shí)現(xiàn)其抽象方法injectMappedStatement，功能就是拼接sql，具體實(shí)現(xiàn)如下：

/**
 * 通過ID批量更新
 */
public class UpdateBatchById extends AbstractMethod {
    private static final long serialVersionUID = 4198102405483580486L;
    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        MySqlMethod sqlMethod = MySqlMethod.UPDATE_BATCH_BY_ID;
        String sql = String.format(sqlMethod.getSql(), tableInfo.getTableName(), this.sqlSet(tableInfo), tableInfo.getKeyColumn(), this.sqlIn(tableInfo.getKeyProperty()));
        SqlSource sqlSource = this.languageDriver.createSqlSource(this.configuration, sql, modelClass);
        return this.addUpdateMappedStatement(mapperClass, modelClass, sqlMethod.getMethod(), sqlSource);
    }
    private String sqlSet(TableInfo tableInfo) {
        List<TableFieldInfo> fieldList = tableInfo.getFieldList();
        StringBuilder sb = new StringBuilder();
        for (TableFieldInfo fieldInfo : fieldList) {
            sb.append("<if test=\"ew.updateFields.contains(&quot;").append(fieldInfo.getColumn()).append("&quot;)\">")
                    .append(fieldInfo.getColumn()).append(" =\n")
                    .append("CASE ").append(tableInfo.getKeyColumn()).append("\n")
                    .append("<foreach collection=\"list\" item=\"et\" >\n")
                    .append("WHEN #{et.").append(tableInfo.getKeyProperty()).append("} THEN #{et.").append(fieldInfo.getProperty()).append("}\n")
                    .append("</foreach>\n").append("END ,\n")
                    .append("</if>\n");
        }
        return "<set>\n" + sb + "</set>";
    }
    private String sqlIn(String keyProperty) {
        StringBuilder sb = new StringBuilder();
        sb.append("<foreach collection=\"list\" item=\"et\" separator=\",\" open=\"(\" close=\")\">\n")
                .append("#{et.").append(keyProperty).append("}")
                .append("</foreach>\n");
        return sb.toString();
    }
}

到了這一步已經(jīng)能夠基本實(shí)現(xiàn)功能了，但是無法控制需要更新的字段，繼續(xù)看下面。

自定義更新wrapper

自定義UpdateBatchWrapper繼承自AbstractLambdaWrapper，此類主要為updateFields屬性設(shè)置值，拼接sql的時(shí)候只對設(shè)置的屬性更新，其他屬性不變。

public class UpdateBatchWrapper<T> extends AbstractLambdaWrapper<T, UpdateBatchWrapper<T>> {
    private static final long serialVersionUID = 114684162001472707L;
    /**
     * 需要更新的字段
     */
    private List<String> updateFields = null;
    @Override
    protected UpdateBatchWrapper<T> instance() {
        this.updateFields = new ArrayList<>();
        return this;
    }
	/**
     * 關(guān)鍵代碼,為屬性設(shè)置值
     */
    @SafeVarargs
    public final UpdateBatchWrapper<T> setUpdateFields(SFunction<T, ?>... columns) {
        this.updateFields = Arrays.asList(columnsToString(columns).split(","));
        return this;
    }
    public List<String> getUpdateFields() {
        return updateFields;
    }
}

參照批量新增把方法添加到方法列表

MyBaseMapper增加配置：

 /**
     * 通過ID批量更新數(shù)據(jù)
     *
     * @param entityList 實(shí)體列表
     * @return 影響行數(shù)
     */
    int updateBatchById(@Param("list") Collection<T> entityList, @Param("ew") Wrapper<T> updateWrapper);

MySqlInjector增加配置：

// 添加批量更新方法
methodList.add(new UpdateBatchById());

測試updateBatchById

創(chuàng)建一個(gè)接口saveUserOrder實(shí)現(xiàn)用戶排序，進(jìn)而檢查批量更新方法。

controller層

    @PostMapping("saveUserOrder")
    @ApiOperation("用戶排序")
    public Result saveUserOrder(@RequestBody List<OrderUserSO> soList) {
        sysService.saveUserOrder(soList);
        return Result.success();
    }

service層

@Override
    public void saveUserOrder(List<OrderUserSO> soList) {
        // 業(yè)務(wù)實(shí)體轉(zhuǎn)換為數(shù)據(jù)庫實(shí)體
        List<SysUser> userList = JSONUtil.toList(JSONUtil.toJsonStr(soList), SysUser.class);
        // 批量更新-設(shè)置更新字段為userOrder
        sysUserMapper.updateBatchById(userList,
                new UpdateBatchWrapper<SysUser>()
                        .setUpdateFields(SysUser::getUserOrder));
    }

OrderUserSO實(shí)體

@Data
@ApiModel("用戶排序業(yè)務(wù)實(shí)體")
public class OrderUserSO implements Serializable {
    private static final long serialVersionUID = 509541044282315352L;
    @ApiModelProperty(value = "用戶ID", required = true)
    @NotNull
    private Integer id;
    @ApiModelProperty(value = "用戶順序", required = true)
    @NotNull
    private Integer userOrder;
}

見證奇跡的時(shí)刻到了…去文章開頭見證奇跡吧。