快捷導(dǎo)航

MybatisPlus更新為null的字段及自定義sql注入

更新時間：2024年05月07日 10:56:37 作者：擺爛小趙

mybatis-plus在執(zhí)行更新操作,當更新字段為空字符串或者null的則不會執(zhí)行更新,本文主要介紹了MybatisPlus更新為null的字段及自定義sql注入,感興趣的可以了解一下

1、全局配置（不推薦）

可以在 application.yml 配置文件中注入配置 GlobalConfiguration 屬性 update-strategy，

將 update-strategy 策略調(diào)整為 IGNORED，即忽略判斷策略。即可調(diào)整全局的驗證策略。

# yml 配置：
mybatis-plus:
  global-config:
    db-config:
      update-strategy: IGNORED

2、添加updateStrategy 注解（不推薦）

在實體類中將需要更新為null的字段上添加updateStrategy 注解，并將更新策略設(shè)置為IGNORED。

@TableField(value = "name", updateStrategy=FieldStrategy.IGNORED)
private String name;

3、使用UpdateWrapper更新（當屬性很多寫起來很麻煩）

通過baseMapper的UpdateWrapper將實體類的指定字段set為null。

baseMapper.update(sysUser, Wrappers.<SysUser>lambdaUpdate()
                        .set(SysUser::getOrgId, null)
                        .set(SysUser::getOrgName, null)
                        .eq(SysUser::getId, sysUser.getId())
                );

4、Mybatis Plus 自定義SqlInjector sql注入器（推薦）

（1）、自定義sql注入器MySqlInjector

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 MySqlInjector extends DefaultSqlInjector {
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass, tableInfo);
        // 根據(jù)id更新所有數(shù)據(jù)
        methodList.add(new UpdateWithNullMethod());
        return methodList;
    }

}

（2）、方法對應(yīng)的實現(xiàn)類UpdateWithNullMethod

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.FieldStrategy;
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.toolkit.StringPool;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.core.toolkit.sql.SqlScriptUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlSource;

import java.lang.reflect.Field;
import java.util.List;

import static java.util.stream.Collectors.joining;

/**
 * @author ：zzt
 */
@Slf4j
public class UpdateWithNullMethod extends AbstractMethod {
    protected UpdateWithNullMethod() {
        super("updateWithNull");
    }

    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        String sql = "<script>\nupdate %s %s where %s=#{%s}\n</script>";

        final List<TableFieldInfo> fieldList = tableInfo.getFieldList();
        for (TableFieldInfo tableFieldInfo : fieldList) {
            final Class<? extends TableFieldInfo> aClass = tableFieldInfo.getClass();
            try {
                final Field fieldFill = aClass.getDeclaredField("fieldFill");
                fieldFill.setAccessible(true);
                fieldFill.set(tableFieldInfo, FieldFill.UPDATE);
            } catch (NoSuchFieldException | IllegalAccessException e) {
                log.error("獲取fieldFill失敗", e);
            }
        }
        String setSql = this.sqlSet(tableInfo);
        String sqlResult = String.format(sql, tableInfo.getTableName(), setSql, tableInfo.getKeyColumn(), tableInfo.getKeyProperty());
        log.debug("sqlResult----->{}", sqlResult);
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sqlResult, modelClass);
        // 第三個參數(shù)必須和rootMapper的自定義方法名一致
        return this.addUpdateMappedStatement(mapperClass, modelClass, "updateWithNull", sqlSource);
    }

    protected String sqlSet(TableInfo table) {
        String sqlScript = table.getFieldList()
                .stream().map(i -> this.getSqlSet(i, StringPool.EMPTY)).collect(joining(StringPool.NEWLINE));
        sqlScript = SqlScriptUtils.convertTrim(sqlScript, "SET", null, null, ",");
        return sqlScript;
    }

    public String getSqlSet(TableFieldInfo i, String prefix) {
        String newPrefix = prefix == null ? StringPool.EMPTY : prefix;
        String column = i.getColumn();
        String update = i.getUpdate();
        FieldFill fieldFill = i.getFieldFill();
        String el = i.getEl();

        // 默認：column=
        String sqlSet = column + StringPool.EQUALS;
        if (StringUtils.isNotEmpty(update)) {
            sqlSet += String.format(update, column);
        } else {
            sqlSet += SqlScriptUtils.safeParam(newPrefix +el);
        }

        sqlSet += StringPool.COMMA;
        if (fieldFill == FieldFill.UPDATE || fieldFill == FieldFill.INSERT_UPDATE) {
            // 不進行if包裹
            return sqlSet;
        }
        return convertIf(sqlSet, convertIfProperty(newPrefix, column), i.getUpdateStrategy(),
                i.getPropertyType().isPrimitive(), StringUtils.isCharSequence(i.getPropertyType()));
    }

    private String convertIfProperty(String prefix, String property) {
        return StringUtils.isNotBlank(prefix)
                ? prefix.substring(0, prefix.length() - 1) + "['" + property + "']" : property;
    }


    private String convertIf(final String sqlScript, final String property, final FieldStrategy fieldStrategy,
                             boolean isPrimitive, boolean isCharSequence) {
        if (fieldStrategy == FieldStrategy.NEVER) {
            return null;
        }
        if (isPrimitive || fieldStrategy == FieldStrategy.IGNORED) {
            return sqlScript;
        }
        if (fieldStrategy == FieldStrategy.NOT_EMPTY && isCharSequence) {
            return SqlScriptUtils.convertIf(sqlScript, String.format("%s != null and %s !=''", property, property), false);
        }
        return SqlScriptUtils.convertIf(sqlScript, String.format("%s != null", property), false);
    }
}

（2）、自定義基礎(chǔ)Mapper繼承BaseMapper

import com.baomidou.mybatisplus.core.mapper.BaseMapper;

/**
 * @description: 自定義mapper借口，mybatisPlus
 * @author: zzt
 */
public interface MyBaseMapper<T> extends BaseMapper<T> {

    /**
     * 全字段更新，更新為null的字段
     * @param obj 待更新對象
     * @return 是否成功
     */
    boolean updateWithNull(T obj);
}

最后直接調(diào)用即可。

到此這篇關(guān)于MybatisPlus更新為null的字段及自定義sql注入的文章就介紹到這了,更多相關(guān)MybatisPlus更新為null內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: