SpringBoot中间件封装限流器的方案详解

RateLimiter rateLimiter = RateLimiter.create(10);

if (rateLimiter.tryAcquire()) {
    // do something
}

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface DoRateLimiter {
    // 每秒访问次数限制
    public double rate() default 0d;
    // 超频后的触发熔断默认响应结果
    public String returnJson() default "";
}

public class Constant {
    /**
     * 全局Hashmap，用于存放【方法】对应的限流类
     * key: 类名 + 方法名
     * value: 对应限流类
     */
    public static Map<String, RateLimiter> stringRateLimiterMap = Collections.synchronizedMap(new HashMap<String, RateLimiter>());
}

@Component
@Aspect
public class RateLimiterHandle {
    // 切点
    @Pointcut("@annotation(com.example.ratelimiteer.annotation.DoRateLimiter)")
    public void aopPointcut() {
    }
    /**
     * @param jp            切点
     * @param doRateLimiter 注解实例
     * @return 方法响应结果
     * @throws Throwable
     */
    @Around(value = "aopPointcut()&& @annotation(doRateLimiter)")
    public Object handle(ProceedingJoinPoint jp, DoRateLimiter doRateLimiter) throws Throwable {
        // 判断是否开启限流
        if (doRateLimiter.rate() == 0d) {
            return jp.proceed();
        }
        // 构建限流哈希Map的Key: 类名 + "." + "方法名"
        String className = jp.getTarget().getClass().getName();
        MethodSignature signature = (MethodSignature) jp.getSignature();
        Method method = jp.getTarget().getClass().getMethod(signature.getMethod().getName(), signature.getMethod().getParameterTypes());
        String methodName = method.getName();
        String key = className + "." + methodName;
        // 第一次走到该接口
        if (!Constant.stringRateLimiterMap.containsKey(key)) {
            Map<String, RateLimiter> rateLimiterMap = Constant.stringRateLimiterMap;
            rateLimiterMap.put(key, RateLimiter.create(doRateLimiter.rate()));
        }
        // 判断是否限流
        RateLimiter rateLimiter = Constant.stringRateLimiterMap.get(key);
        if (rateLimiter.tryAcquire()) {
            // 未超过访问限制，则继续执行对应业务逻辑
            return jp.proceed();
        } else {
            // 超过访问限制，则返回默认熔断响应结果（即注解中的returnJson）
            return JSON.parseObject(doRateLimiter.returnJson());
        }
    }
}