Redisson多策略注解限流

限流：使用Redisson的RRateLimiter进行限流
多策略：map+函数式接口优化if判断
自定义注解

1. /**
2. * aop限流注解
3. */
4. @Target({ElementType.METHOD, ElementType.TYPE})
5. @Retention(RetentionPolicy.RUNTIME)
6. @Inherited
7. @Documented
8. public @interface RedisLimit {
11.     String prefix() default "rateLimit:";
13.     //限流唯一标示
14.     String key() default "";
16.     //限流单位时间（单位为s）
17.     int time() default 1;
19.     //单位时间内限制的访问次数
20.     int count();
22.     //限流类型
23.     LimitType type() default LimitType.CUSTOM;
25. }

复制代码

定义限流类型

1. public enum LimitType {
3.     /**
4.      * 自定义key
5.      */
6.     CUSTOM,
8.     /**
9.      * 请求者IP
10.      */
11.     IP,
13.     /**
14.      * 方法级别限流
15.      * key = ClassName+MethodName
16.      */
17.     METHOD,
19.     /**
20.      * 参数级别限流
21.      * key = ClassName+MethodName+Params
22.      */
23.     PARAMS,
25.     /**
26.      * 用户级别限流
27.      * key = ClassName+MethodName+Params+UserId
28.      */
29.     USER,
31.     /**
32.      * 根据request的uri限流
33.      * key = Request_uri
34.      */
35.     REQUEST_URI,
37.     /**
38.      * 对requesturi+userId限流
39.      * key = Request_uri+UserId
40.      */
41.     REQUESTURI_USERID,
44.     /**
45.      * 对userId限流
46.      * key = userId
47.      */
48.     SINGLEUSER,
50.     /**
51.      * 对方法限流
52.      * key = ClassName+MethodName
53.      */
54.     SINGLEMETHOD,
56.     /**
57.      * 对uri+params限流
58.      * key = uri+params
59.      */
60.     REQUEST_URI_PARAMS,
62.     /**
63.      * 对uri+params+userId限流
64.      * key = uri+params+userId
65.      */
66.     REQUEST_URI_PARAMS_USERID;
67.    
68. }

复制代码

生成key的工具类

根据类型生成锁的对象(key)的工具类，使用map+函数式接口优化if，其中BaseContext是一个获取用户唯一标识userId的工具类

1. @Component
2. public class ProceedingJoinPointUtil {
3.     @Autowired
4.     private HttpServletRequest request;
6.     private Map<LimitType, Function<ProceedingJoinPoint,String>> functionMap = new HashMap<>(9);
8.     @PostConstruct
9.     void initMap(){
10.         //初始化策略
11.         functionMap.put(LimitType.METHOD, this::getMethodTypeKey);
12.         functionMap.put(LimitType.PARAMS, this::getParamsTypeKey);
13.         functionMap.put(LimitType.USER, this::getUserTypeKey);
14.         functionMap.put(LimitType.REQUEST_URI,proceedingJoinPoint ->
15.                 request.getRequestURI());
16.         functionMap.put(LimitType.REQUESTURI_USERID, proceedingJoinPoint ->
17.                 request.getRequestURI()+BaseContext.getUserId());
18.         functionMap.put(LimitType.REQUEST_URI_PARAMS,proceedingJoinPoint ->
19.                 request.getRequestURI()+getParams(proceedingJoinPoint));
20.         functionMap.put(LimitType.REQUEST_URI_PARAMS_USERID,proceedingJoinPoint ->
21.                 request.getRequestURI()+getParams(proceedingJoinPoint)+BaseContext.getUserId());
22.         functionMap.put(LimitType.SINGLEUSER,(proceedingJoinPoint)->
23.                 String.valueOf(BaseContext.getUserId()));
24.         functionMap.put(LimitType.SINGLEMETHOD,(proceedingJoinPoint -> {
25.             StringBuilder sb = new StringBuilder();
26.             appendMthodName(proceedingJoinPoint,sb);
27.             return sb.toString();
28.         }));
29.     }
31.     public Object getKey(ProceedingJoinPoint joinPoint, RedisLimit redisLimit) {
32.         //根据限制类型生成key
33.         Object generateKey = "";
34.         //自定义
35.         if(redisLimit.type() != LimitType.CUSTOM){
36.             generateKey = generateKey(redisLimit.type(), joinPoint);
37.         }else {
38.             //非自定义
39.             generateKey = redisLimit.key();
40.         }
41.         return generateKey;
42.     }
44.     /**
45.      * 根据LimitType生成key
46.      * @param type
47.      * @param joinPoint
48.      * @return
49.      */
50.     private Object generateKey(LimitType type , ProceedingJoinPoint joinPoint) {
51.         Function function = functionMap.get(type);
52.         Object result = function.apply(joinPoint);
53.         return result;
54.     }
56.     /**
57.      * 方法级别
58.      * key = ClassName+MethodName
59.      * @param joinPoint
60.      * @return
61.      */
62.     private String getMethodTypeKey(ProceedingJoinPoint joinPoint){
63.         StringBuilder sb = new StringBuilder();
64.         appendMthodName(joinPoint, sb);
65.         return sb.toString();
66.     }
70.     /**
71.      * 参数级别
72.      * key = ClassName+MethodName+Params
73.      * @param joinPoint
74.      * @return
75.      */
76.     private String getParamsTypeKey(ProceedingJoinPoint joinPoint){
77.         StringBuilder sb = new StringBuilder();
78.         appendMthodName(joinPoint, sb);
79.         appendParams(joinPoint, sb);
80.         return sb.toString();
81.     }
85.     /**
86.      * 用户级别
87.      * key = ClassName+MethodName+Params+UserId
88.      */
89.     private String getUserTypeKey(ProceedingJoinPoint joinPoint){
90.         StringBuilder sb = new StringBuilder();
91.         appendMthodName(joinPoint, sb);
92.         appendParams(joinPoint, sb);
93.         //获取userId
94.         appendUserId(sb);
95.         return sb.toString();
96.     }
99.     /**
100.      * StringBuilder添加类名和方法名
101.      * @param joinPoint
102.      * @param sb
103.      */
104.     private void appendMthodName(ProceedingJoinPoint joinPoint, StringBuilder sb) {
105.         Signature signature = joinPoint.getSignature();
106.         MethodSignature methodSignature = (MethodSignature) signature;
107.         Method method = methodSignature.getMethod();
108.         sb.append(joinPoint.getTarget().getClass().getName())//类名
109.                 .append(method.getName());//方法名
110.     }
112.     /**
113.      * StringBuilder添加方法参数值
114.      * @param joinPoint
115.      * @param sb
116.      */
117.     private void appendParams(ProceedingJoinPoint joinPoint, StringBuilder sb) {
118.         for (Object o : joinPoint.getArgs()) {
119.             sb.append(o.toString());
120.         }
121.     }
123.     private String getParams(ProceedingJoinPoint joinPoint) {
124.         StringBuilder sb = new StringBuilder();
125.         for (Object o : joinPoint.getArgs()) {
126.             if(o instanceof MultipartFile){
127.                 try {
128.                     ImageTypeCheck.getImgHeightAndWidth(((MultipartFile) o).getInputStream());
129.                 } catch (IOException e) {
130.                     throw new BusinessException("MultipartFile输入流获取失败,source:ProceedingJoinPointUtils.149",USER_PRINCIPAL_EMAIL);
131.                 }
132.             }else {
133.                 sb.append(o.toString());
134.             }
135.         }
136.         return sb.toString();
137.     }
139.     /**
140.      * StringBuilder添加UserId
141.      * @param sb
142.      */
143.     private void appendUserId(StringBuilder sb) {
144.         sb.append(BaseContext.getUserId());
145.     }
146. }

复制代码

定义aop具体逻辑

1. @Aspect
2. @Component
3. @Slf4j
4. public class RedisLimitAspect {
5.     @Autowired
6.     private RedissonClient redissonClient;
8.     @Autowired
9.     private ProceedingJoinPointUtil proceedingJoinPointUtil;
11.     @Pointcut("@annotation(com.cat.www.aop.limit.anno.RedisLimit)")
12.     private void pointCut() {
13.     }
15.     @Around("pointCut() && @annotation(redisLimit)")
16.     private Object around(ProceedingJoinPoint joinPoint, RedisLimit redisLimit) {
17.         Object generateKey = proceedingJoinPointUtil.getKey(joinPoint, redisLimit);
18.         //redis key
19.         String key = redisLimit.prefix() +generateKey.toString();
20.         //声明一个限流器
21.         RRateLimiter rateLimiter = redissonClient.getRateLimiter(key);
23.         //设置速率，time秒中产生count个令牌
24.         rateLimiter.trySetRate(RateType.OVERALL, redisLimit.count(), redisLimit.time(), RateIntervalUnit.SECONDS);
26.         // 试图获取一个令牌，获取到返回true
27.         boolean tryAcquire = rateLimiter.tryAcquire();
28.         if (!tryAcquire) {
29.             return new ResultData<>().FAILED().setResultIns("访问过于频繁");
30.         }
31.         Object obj = null;
32.         try {
33.             obj = joinPoint.proceed();
34.         } catch (Throwable e) {
35.             throw new RuntimeException();
36.         }
38.         return obj;
39.     }
40. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！