springboot如何通過自定義注解對方法參數(shù)進行攔截驗證

更新時間：2025年04月27日 16:46:39 作者：藍眸少年CY

這篇文章主要介紹了springboot如何通過自定義注解對方法參數(shù)進行攔截驗證問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

springboot通過自定義注解對方法參數(shù)進行攔截驗證

元注解參數(shù)說明

@Target 定義注解的作用目標，也就是可以定義注解具體作用在類上，方法上，還是變量上
@Retention 定義注解的保留策略

RetentionPolicy.SOURCE	注解僅存在于源碼中在class字節(jié)碼文件中不包含
RetentionPolicy.CLASS	默認的保留策略注解會在class字節(jié)碼文件中存在但運行時無法獲得;
RetentionPolicy.RUNTIME	注解會在class字節(jié)碼文件中存在，在運行時可以通過反射獲取到。

@Document 說明該注解將被包含在javadoc中
@Inherited 說明子類可以繼承父類中的該注解

@Target類型和說明

類型	說明
ElementType.TYPE	接口、類、枚舉、注解
ElementType.FIELD	字段、枚舉的常量
ElementType.METHOD	方法
ElementType.PARAMETER	方法參數(shù)
ElementType.CONSTRUCTOR	構造函數(shù)
ElementType.LOCAL_VARIABLE	局部變量
ElementType.ANNOTATION_TYPE	注解
ElementType.PACKAGE	包

具體實現(xiàn):簡單版

1、引入坐標

<!-- 引入aop切面支持 -->
<dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-aop</artifactId>
</dependency>

2、創(chuàng)建自定義注解

package com.hk.annotation;
 
import java.lang.annotation.*;
 
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Inherited
public @interface ParamsIntercept {
  
}

@Target注解指定ElementType
@Inherited 說明子類可以繼承父類中的該注解
@Retention注解指定RetentionPolicy

注意：

注解支持的元素類型除了上面的String之外還有以下：

基本類型（int、char、byte、double、float、long、boolean）
字符串String
類Class
枚舉enum
注解Annotation
上述類型的數(shù)組類型

當使用其他類型修飾注解元素時，編譯期會報錯

Invalid type 'Integer' for annotation member

基本類型的包裝類型也是不允許在注解中修飾注解元素的；上述代碼中 subjectId 不能定義為 Integer

3、創(chuàng)建切面進行判斷

import com.alibaba.fastjson.JSONObject;
import com.bxm.adsmanager.model.dao.user.User;
import org.apache.commons.lang.ArrayUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.log4j.Logger;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;
  
import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
  
@Aspect
@Component
public class ParamsBeforeAspect {
  
    private static final Logger logger = Logger.getLogger(ParamsBeforeAspect .class);
  
    @Before("@annotation(com.hk.annotation.ParamsIntercept)")
    public void doBefore(JoinPoint point){
        try {
            long startTime = System.currentTimeMillis();//開始時間
            Method method = getMethod(point);
            if(null == method){
                if(logger.isWarnEnabled()){
                    logger.warn("method is null");
                }
                return;
            }            
 
            Object[] args = point.getArgs();//獲取請求參數(shù)
            
            if (ArrayUtils.isNotEmpty(args)) {
                for (Object arg : args) {
                   // 對參數(shù)進行判斷
                }
            }
  
            long endTime = System.currentTimeMillis();//結束時間
            float excTime=(float)(endTime-startTime)/1000;
            logger.info("總耗時：" + excTime+"s");
        }catch (Exception e){
            logger.error("記錄日志異常",e);
        }
    } 
 
    private Method getMethod(JoinPoint point) {
        MethodSignature methodSignature = (MethodSignature) point.getSignature();
        Class<?> targetClass = point.getTarget().getClass();
        try {
            return targetClass.getMethod(methodSignature.getName(), methodSignature.getParameterTypes());
        } catch (NoSuchMethodException e) {
            return null;
        }
    }

4、controller使用

@Controller
@RequestMapping(value = "/")
public class UserManagerController {
    //1.傳入的是一個具體的值
	@ParamsIntercept
    @RequestMapping(value = "/getUser/{subjectId}")
    public R<String> getUserDetail(@PathVariable Integer subjectId) {
        try {
            //處理自己的業(yè)務
        } catch (Exception e) {
            e.printStackTrace();
            return R.error(e.getMessage());
        }
        return R.error("操作失敗");
    }
   
}

具體實現(xiàn):封裝版

1、引入坐標

<!-- 引入aop切面支持 -->
<dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-aop</artifactId>
</dependency>

2、創(chuàng)建自定義注解

package com.hk.annotation;
 
import java.lang.annotation.*;
 
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Inherited
public @interface ParamsIntercept {
  String subjectId();
}

3、創(chuàng)建切面

package com.hk.aspect;
 
import com.warmer.base.enums.ReturnStatus;
import com.warmer.base.util.R;
import com.warmer.base.util.SpringUtils;
import com.warmer.base.util.StringUtil;
import com.warmer.web.annotation.AnnotationResolver;
import com.warmer.web.annotation.DomainOwner;
import com.warmer.web.entity.KgDomain;
import com.warmer.web.security.TokenService;
import com.warmer.web.service.KnowledgeGraphService;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
 
import javax.servlet.http.HttpServletRequest;
import java.util.List;
 
@Aspect
@Component
public class DomainValidAspect {
 
    @Pointcut("@annotation(com.hk.annotation.ParamsIntercept)")
    public void annotationPointCut() {
 
    }
 
    @Before("annotationPointCut()")
    public void doBefore(JoinPoint joinPoint) {
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        // 獲取方法注解
        ParamsIntercept paramsIntercept = signature.getMethod().getAnnotation(ParamsIntercept.class);
        // 獲取參數(shù)
        String subjectIdCode = paramsIntercept.subjectId();
        ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = servletRequestAttributes.getRequest();
        Integer memberId = (Integer) request.getAttribute("memberId");
        AnnotationResolver annotationResolver = AnnotationResolver.newInstance();
        Integer resolver = (Integer) annotationResolver.resolver(joinPoint, subjectIdCode);
        log.info("獲取請求參數(shù)：subjectId：{}， memberId：{}", resolver, memberId);
        // 具體業(yè)務代碼
        .......
    }
}

4、封裝獲取參數(shù)工具

package com.hk.aspect;

import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.reflect.MethodSignature;

import java.lang.reflect.Method;
import java.util.Map;

/**
 * @description:
 * @author: HK
 * @since: 2024/9/25 15:16
 */
public class AnnotationResolver {
    private static AnnotationResolver resolver ;
    public static AnnotationResolver newInstance(){
        if (resolver == null) {
            return resolver = new AnnotationResolver();
        }else{
            return resolver;
        }
    }
    
    public Object resolver(JoinPoint joinPoint, String str) {

        if (str == null) return null ;

        Object value = null;
        if (str.matches("#\\{\\D*\\}")) {
            String newStr = str.replaceAll("#\\{", "").replaceAll("\\}", "");
            if (newStr.contains(".")) { // 復雜類型
                try {
                    value = complexResolver(joinPoint, newStr);
                } catch (Exception e) {
                    e.printStackTrace();
                }
            } else {
                value = simpleResolver(joinPoint, newStr);
            }
        } else { //非變量
            value = str;
        }
        return value;
    }


    private Object complexResolver(JoinPoint joinPoint, String str) throws Exception {
        MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
        String[] names = methodSignature.getParameterNames();
        Object[] args = joinPoint.getArgs();
        String[] strs = str.split("\\.");

        for (int i = 0; i < names.length; i++) {
            if (strs[0].equals(names[i])) {
                Object obj = args[i];
                //這里處理出入?yún)?shù)為Map的邏輯
                if(obj instanceof Map){
                    Map item=(Map) obj;
                    return item.get(strs[1]);
                }
                Method dmethod = obj.getClass().getDeclaredMethod(getMethodName(strs[1]), null);
                Object value = dmethod.invoke(args[i]);
                return getValue(value, 1, strs);
            }
        }
        return null;
    }

    private Object getValue(Object obj, int index, String[] strs) {
        try {
            if (obj != null && index < strs.length - 1) {
                Method method = obj.getClass().getDeclaredMethod(getMethodName(strs[index + 1]), null);
                obj = method.invoke(obj);
                getValue(obj, index + 1, strs);
            }
            return obj;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    private String getMethodName(String name) {
        return "get" + name.replaceFirst(name.substring(0, 1), name.substring(0, 1).toUpperCase());
    }
    private Object simpleResolver(JoinPoint joinPoint, String str) {
        MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
        String[] names = methodSignature.getParameterNames();
        Object[] args = joinPoint.getArgs();

        for (int i = 0; i < names.length; i++) {
            if (str.equals(names[i])) {
                return args[i];
            }
        }
        return null;
    }

}

5、controller使用

@Controller
@RequestMapping(value = "/")
public class UserManagerController {
    //1.傳入的是一個具體的值
	@ParamsIntercept(subjectId= "#{userCode}")
    @RequestMapping(value = "/getUser/{subjectId}")
    public R<String> getUserDetail(@PathVariable Integer subjectId) {
        try {
            //處理自己的業(yè)務
        } catch (Exception e) {
            e.printStackTrace();
            return R.error(e.getMessage());
        }
        return R.error("操作失敗");
    }
    //2.傳入的是一個對象
    @ParamsIntercept(subjectId= "#{userItem.subjectId}")
    
	//3.傳入的可能是一個map
 	@ParamsIntercept(subjectId= "#{params.subjectId}")
   
}

總結

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

Java?SimpleDateFormat線程不安全問題
這篇文章詳細介紹了如可解決impleDateFormat線程不安全的問題，對多線程問題感興趣的同學可以參考閱讀本文
2023-03-03
Java冒泡排序及優(yōu)化介紹
大家好，本篇文章主要講的是Java冒泡排序及優(yōu)化介紹，感興趣的同學趕快來看一看吧，對你有幫助的話記得收藏一下，方便下次瀏覽
2021-12-12
Intellij IDEA 配置Subversion插件實現(xiàn)步驟詳解
這篇文章主要介紹了Intellij IDEA 配置Subversion插件實現(xiàn)步驟詳解的相關資料,需要的朋友可以參考下
2017-05-05
Java主線程捕獲子線程異常的實現(xiàn)
本文主要介紹了Java主線程捕獲子線程異常的實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧
2024-04-04
springboot~nexus項目打包要注意的地方示例代碼詳解
這篇文章主要介紹了springboot~nexus項目打包要注意的地方,本文通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2020-07-07
eclipse maven maven-archetype-webapp 創(chuàng)建失敗問題解決
這篇文章主要介紹了eclipse maven maven-archetype-webapp 創(chuàng)建失敗問題解決的相關資料,需要的朋友可以參考下
2016-12-12
源碼解析帶你了解LinkedHashMap
大多數(shù)情況下，只要不涉及線程安全問題，Map基本都可以使用HashMap，不過HashMap有一個問題，就是迭代HashMap的順序并不是HashMap放置的順序，也就是無序。HashMap的這一缺點往往會帶來困擾，所以LinkedHashMap就閃亮登場了，這篇文章通過源碼解析帶你了解LinkedHashMap
2021-09-09
Mybatis中SQL的執(zhí)行過程詳解
MyBatis框架通過映射文件或注解將Java代碼中的方法與數(shù)據(jù)庫操作進行映射,執(zhí)行過程包括SQL解析、參數(shù)綁定、SQL預編譯、執(zhí)行、結果映射、事務處理、緩存處理和日志記錄
2024-12-12
關于springboot的跨域配置問題的解決方案
這篇文章主要介紹了關于springboot的跨域配置問題,處理filter,spring?security等過濾器跨域問題,本文通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2023-07-07
Java服務剛啟動時接口超時排查全過程
這篇文章主要為大家介紹了Java服務剛啟動時，一小波接口超時排查全過程，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪
2023-07-07