Spring使用Spel表達式獲取參數值

一、依賴

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>

<!-- aop -->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-aop</artifactId>
</dependency>

二、注解

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface Limiter {
    /**
     * 使用spel表達式獲取限流的key
     * @return
     */
    String value();
}

三、AOP切面的應用

@Aspect
@Component
public class LimiterAspect {
    private ExpressionParser parser = new SpelExpressionParser();
    private LocalVariableTableParameterNameDiscoverer discoverer = new LocalVariableTableParameterNameDiscoverer();

    @Pointcut("@annotation(limiter)")
    public void pointcut(Limiter limiter) {
    }

    @Around("pointcut(limiter)")
    public Object around(ProceedingJoinPoint pjp, Limiter limiter) throws Throwable {
        Method method = this.getMethod(pjp);
        String methodName = method.toString();

        //獲取方法的參數值
        Object[] args = pjp.getArgs();
        EvaluationContext context = this.bindParam(method, args);

        //根據spel表達式獲取值
        Expression expression = parser.parseExpression(limiter.value());
        Object key = expression.getValue(context);
        //打印
        System.out.println(key);
        
        return pjp.proceed();
    }

    /**
     * 獲取當前執行的方法
     *
     * @param pjp
     * @return
     * @throws NoSuchMethodException
     */
    private Method getMethod(ProceedingJoinPoint pjp) throws NoSuchMethodException {
        MethodSignature methodSignature = (MethodSignature) pjp.getSignature();
        Method method = methodSignature.getMethod();
        Method targetMethod = pjp.getTarget().getClass().getMethod(method.getName(), method.getParameterTypes());
        return targetMethod;
    }

    /**
     * 將方法的參數名和參數值綁定
     *
     * @param method 方法，根據方法獲取參數名
     * @param args   方法的參數值
     * @return
     */
    private EvaluationContext bindParam(Method method, Object[] args) {
        //獲取方法的參數名
        String[] params = discoverer.getParameterNames(method);

        //將參數名與參數值對應起來
        EvaluationContext context = new StandardEvaluationContext();
        for (int len = 0; len < params.length; len++) {
            context.setVariable(params[len], args[len]);
        }
        return context;
    }

四、Controller

@RestController
public class TestController {

    //獲取名為id的參數
    @Limiter("#id")
    @GetMapping("test")
    public String test(Long id){
        return "hello";
    }
}

五、獲取對象（補充）

1、注解

@Limter(value = "#testObj")
public JSONObject test01(TestObj  testObj){
      // ......      
}

多個切點同時獲取

/**
 * 設置操作日志切入點 記錄操作日志 在注解的位置切入代碼
 */
//@Pointcut("@annotation(com.test.Limter)")
@Pointcut("@annotation(limter)")
public void operLogPointCut(Limter limter) {
}

@Pointcut("execution(public * com.test.aaa..*.*(..))")
public void operLogPointMethod() {
}

線程變量的使用，當前切面類中使用線程變量存儲變量

@Aspect
@Component
public class LogAspect {

    @Autowired
    private LogsService logsService;
    // 線程變量
    private ThreadLocal<String> threadLocal = new ThreadLocal<>();  9 
    /**
     * 設置操作日志切入點 記錄操作日志 在注解的位置切入代碼
     */

方法體中存入數據

public void savaData(){
    threadLocal.set("asdf");
}

在另一個方法體中獲取當前線程數據

public void savaData(){
    String value = threadLocal.get();
}

切點多條件限制 &&

@AfterReturning(value = "operLogPointCut(limter) && operLogPointMethod()", returning = "returnValue")
    public void saveOperLog(JoinPoint joinPoint, Limter limter, Object returnValue) {
        Object[] args = joinPoint.getArgs();
        // 從切面織入點處通過反射機制獲取織入點處的方法
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        // 獲取切入點所在的方法
        Method method = signature.getMethod();
        EvaluationContext context = this.bindParam(method, args);  9         Expression expression = parser.parseExpression(limter.value()); 10 
        TestObj testObj= expression.getValue(context, TestObj.class); 12         // ......
        new Thread(() -> logsService.saveLogs(Obj...)).start();

        // 存入數據庫后移除當前線程變量
        threadLocal.remove();
    }

 

 參考：

https://blog.csdn.net/weixin_45052750/article/details/105742934

 其他參考：

https://www.codeleading.com/article/30952452765/