Mybatis使用攔截器自定義審計處理

    void test_save_1(@Param("relatedBookCategoryEntity") RelatedBookCategoryEntity relatedBookCategoryEntity11, BookEntity bookEntity11, String categoryName11,Integer age11);
　　　　

void test_save_2(RelatedBookCategoryEntity relatedBookCategoryEntity22, BookEntity bookEntity22, String categoryName22,Integer age22);
　　　　

　　 void test_save_3(String categoryName33);
　　　　

　　

    void test_save_4( String categoryName22,Integer age22);

　　　　　

 

　　　　

　　void test_save_5(RelatedBookCategoryEntity relatedBookCategoryEntity22);
　　　　

　　　　　　

 

 

 

 　　

 

以上是測試的接口，方法簽名下方圖片是攔截器內部形參的值結構。

 

以下是Java代碼，實現mybatis的Interceptor接口

package cn.dmahz.config;

import cn.dmahz.dao.mapper.RelatedBookCategoryMapper;
import cn.dmahz.entity.Base;
import cn.dmahz.entity.BookEntity;
import cn.dmahz.entity.RelatedBookCategoryEntity;
import cn.dmahz.utils.MyReflectUtils;
import cn.dmahz.utils.SecurityContextHolderUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
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.data.annotation.*;
import org.springframework.stereotype.Component;

import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.lang.reflect.Parameter;
import java.sql.Statement;
import java.util.*;

/**
 * @author Dream
 */
@Component
@Intercepts({@Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class }),@Signature(type= StatementHandler.class,method = "parameterize",args = {Statement.class}) })
public class MybatisEntityPluginInterceptor implements Interceptor {

    @Override
    public Object intercept(Invocation invocation) throws Throwable {

        Object[] args = invocation.getArgs();
        if(args[0] instanceof MappedStatement){
            // 映射的各種信息，SQL信息、接口方法對應的參數、接口方法的全名稱等等
            MappedStatement mappedStatement = (MappedStatement) args[0];

            // 獲取執行語句的類型
            SqlCommandType sqlCommandType = mappedStatement.getSqlCommandType();


            if(args[1] instanceof HashMap<?, ?>){
                HashMap<?,?> hashMap = (HashMap<?, ?>) args[1];
                // 這裏分別處理每個參數的實體注入
                // mappedStatement.getId()   -> 獲取方法的全路徑 ;例如: cn.xxx.xxx.xxx.ClassName.methodName
                Method methodInMapper = MyReflectUtils.getMethodInMapper(mappedStatement.getId());
                String[] paramKeys = parseParamKeysByMethod(methodInMapper);
                for(String key:paramKeys){
                    Object o = hashMap.get(key);
                    if(o instanceof List){
                        List<?> list = (List<?>) o;
                        for(Object entity:list){
                            Field[] allAuditFields = MyReflectUtils.getAllAuditFields(entity.getClass());
                            setAuditField(sqlCommandType,entity,allAuditFields);
                        }
                    }else if(o instanceof Base){
                        // 這裏處理不是集合的情況，通過繼承 cn.dmahz.entity.Base，可證明爲Java Bean
                        Field[] allAuditFields = MyReflectUtils.getAllAuditFields(o.getClass());
                        setAuditField(sqlCommandType,o,allAuditFields);
                    }
                }
            } else if(args[1] instanceof Base){
                Object o =args[1];
                // 這裏處理不是集合的情況，通過繼承 cn.dmahz.entity.Base，可證明爲Java Bean
                Field[] allAuditFields = MyReflectUtils.getAllAuditFields(o.getClass());
                setAuditField(sqlCommandType,o,allAuditFields);
            }
        }

        // 讓攔截器繼續處理剩餘的操作
        return invocation.proceed();
    }

    /**
     * 解析方法的形參的key值，key值用於在 ParamMap中查找值，進行填充審計字段
     * @param method
     */
    private String[] parseParamKeysByMethod(Method method){
        ArrayList<String> keyList = new ArrayList<>();
        Parameter[] parameters = method.getParameters();
        for (Parameter parameter:parameters) {
            Param parameterAnnotation = parameter.getAnnotation(Param.class);
            if(parameterAnnotation != null){
                keyList.add(parameterAnnotation.value());
            }else {
                // 形參名稱
                String name = parameter.getName();
                // 類型的簡寫名稱
//                String simpleName = parameter.getType().getSimpleName();
                if(StringUtils.isNotBlank(name)){
                    keyList.add(name);
                }
            }
        }
        return keyList.toArray(new String[0]);
    }

    /**
     * 包裝一下重複的代碼，方便調用
     * @param sqlCommandType
     * @param o
     * @param fields
     * @throws IllegalAccessException
     */
    private void setAuditField(SqlCommandType sqlCommandType,Object o,Field[] fields) throws IllegalAccessException{
        for(Field field:fields){
            setAuditField(sqlCommandType,o,field);
        }
    }

    /**
     * 設置審計字段,包括創建人，主鍵ID，創建時間，更新人，更新時間。
     * @param sqlCommandType
     * @param o
     * @param field
     * @throws IllegalAccessException
     */
    private void setAuditField(SqlCommandType sqlCommandType,Object o,Field field) throws IllegalAccessException {
        if(sqlCommandType == SqlCommandType.INSERT){
            if(field.isAnnotationPresent(CreatedBy.class)){
                String currentUserId;
                try {
                    currentUserId = SecurityContextHolderUtils.getCurrentUserId();
                } catch (NullPointerException e) {
                    //這裏僅作測試，忽略空指針異常
                    currentUserId = "非Web環境,當前用戶ID測試值(創建值)";
                }
                field.set(o,currentUserId);
            }else if(field.isAnnotationPresent(CreatedDate.class)){
                field.set(o,System.currentTimeMillis());
            }else if(field.isAnnotationPresent(Id.class)){
                String uuId = UUID.randomUUID().toString();
                field.set(o,uuId.replace("-",""));
            }
        }else if(sqlCommandType == SqlCommandType.UPDATE){
            if(field.isAnnotationPresent(LastModifiedBy.class)){
                String currentUserId;
                try {
                    currentUserId = SecurityContextHolderUtils.getCurrentUserId();
                } catch (NullPointerException e) {
                    //這裏僅作測試，忽略空指針異常
                    currentUserId = "非Web環境,當前用戶ID測試值(更新值)";
                }
                field.set(o,currentUserId);
            }else if(field.isAnnotationPresent(LastModifiedDate.class)){
                field.set(o,System.currentTimeMillis());
            }
        }
    }

    public static void main(String[] args) throws InterruptedException, NoSuchMethodException {

        Method test_save_1 = RelatedBookCategoryMapper.class.getDeclaredMethod("test_save_1", RelatedBookCategoryEntity.class, BookEntity.class, String.class);
//        parseParamKeysByMethod(test_save_1);
    }
}