优化aop语法 使用spring自动注入注解 基于注解拦截的aop注解不可能为空

RuoYi

2021-09-18 76db434373791347a54daa4fca88826d86e68699

 src/main/java/com/ruoyi/framework/aspectj/RateLimiterAspect.java

@@ -4,10 +4,8 @@
import java.util.Collections;

import java.util.List;

import org.aspectj.lang.JoinPoint;

import org.aspectj.lang.Signature;

import org.aspectj.lang.annotation.Aspect;

import org.aspectj.lang.annotation.Before;

import org.aspectj.lang.annotation.Pointcut;

import org.aspectj.lang.reflect.MethodSignature;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

@@ -49,16 +47,9 @@
        this.limitScript = limitScript;

    }



    // 配置织入点

    @Pointcut("@annotation(com.ruoyi.framework.aspectj.lang.annotation.RateLimiter)")

    public void rateLimiterPointCut()

    @Before("@annotation(rateLimiter)")

    public void doBefore(JoinPoint point, RateLimiter rateLimiter) throws Throwable

    {

    }



    @Before("rateLimiterPointCut()")

    public void doBefore(JoinPoint point) throws Throwable

    {

        RateLimiter rateLimiter = getAnnotationRateLimiter(point);

        String key = rateLimiter.key();

        int time = rateLimiter.time();

        int count = rateLimiter.count();

@@ -82,22 +73,6 @@
        {

            throw new RuntimeException("服务器限流异常，请稍后再试");

        }

    }



    /**

     * 是否存在注解，如果存在就获取

     */

    private RateLimiter getAnnotationRateLimiter(JoinPoint joinPoint)

    {

        Signature signature = joinPoint.getSignature();

        MethodSignature methodSignature = (MethodSignature) signature;

        Method method = methodSignature.getMethod();



        if (method != null)

        {

            return method.getAnnotation(RateLimiter.class);

        }

        return null;

    }



    public String getCombineKey(RateLimiter rateLimiter, JoinPoint point)