MyBatis 自定义参数拦截器

目录

目录

一、使用场景

二、自定义MyBatis 拦截器

三、注意注意 映射文件

四、结果

---

---

一、使用场景

•     每次修改页面时候，都要像给最后修改人这个参数赋值 ，当类很多时就造成了冗余代码（每个类的service中都要重新给最后修改人赋值） 

①首先我们要把最后修改人和最后修改时间抽出来单独作为一个类，其他类需要该字段直接继承它就好。

package com.beiyou.model;

import java.time.LocalDateTime;

public abstract class BaseModel {
    /** 最近更新人 */
    private String lastUpdateBy ;
    /** 最近更新时间 */
    private LocalDateTime lastUpdateTime ;


    public String getLastUpdateBy() {
        return lastUpdateBy;
    }

    public void setLastUpdateBy(String lastUpdateBy) {
        this.lastUpdateBy = lastUpdateBy;
    }

    public LocalDateTime getLastUpdateTime() {
        return lastUpdateTime;
    }

    public void setLastUpdateTime(LocalDateTime lastUpdateTime) {
        this.lastUpdateTime = lastUpdateTime;
    }
}

 

二、自定义MyBatis 拦截器

package com.beiyou.config;


import cn.hutool.core.util.ReflectUtil;
import cn.smart.core.threadlocal.LocalUserUtil;
import com.beiyou.model.BaseModel;
import com.sun.xml.internal.bind.v2.runtime.unmarshaller.Intercepter;
import com.sun.xml.internal.bind.v2.runtime.unmarshaller.UnmarshallingContext;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.executor.parameter.ParameterHandler;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Signature;
import org.springframework.stereotype.Component;
import org.xml.sax.SAXException;

import java.lang.reflect.Field;
import java.sql.PreparedStatement;

@Slf4j
@Component
@Intercepts(

        /**  我们要点进去ParameterHandler，就可以找到下面的类型，方法，参数等
         *   type 表示拦截器类型，这里我们使用参数拦截器
         *   method 表示参数该拦截器类型里的方法名
         *   args 表示参数列表，这里我们使用了PreparedStatement
         *
        */

        {@Signature(type = ParameterHandler.class,
                method = "setParameters", args = {PreparedStatement.class})})

public class ParameterIntercepter implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        ParameterHandler target = (ParameterHandler) invocation.getTarget();
        // 获取参数对象，这里面有所有的参数
        Object parameterObject = target.getParameterObject();
        // 1.第一种方法：判断类型 性能高
        if (parameterObject instanceof BaseModel) {
            BaseModel baseModel = (BaseModel) parameterObject;
            // 设置最后更新人
            baseModel.setLastUpdateBy(LocalUserUtil.getLocalUser().getNickName());
            String lastUpdateBy = baseModel.getLastUpdateBy();
            log.info(lastUpdateBy);
        }
        // 2.第二中方法：反射 getField:根据参数名 获取参数对象： 性能低
//        Field lastUpdateBy = ReflectUtil.getField(parameterObject.getClass(), "lastUpdateBy");
//        if(lastUpdateBy !=null){
//            ReflectUtil.setFieldValue(parameterObject,lastUpdateBy,LocalUserUtil.getLocalUser().getNickName());
//        }
        /**
         * 返回目标执行方法
         */
        return invocation.proceed();
    }
}

实现功能 ：以后操作数据时，就可以不用在service里每个都设置最后更新人。

三、注意注意 映射文件

当我们修改时 java首先进入mybatis拦截器，直接帮我们设置好了参数 （最后更新人），zhe时候还没有执行sql 所以这个时候（最后更新人）是空值，所以当我们使用动态sql 时 这个值是空的，直接给过滤了，我们只需要将动态标签删掉即可。

四、结果