- 主页 > 生活百科 > >
接口限流还不太明白? 这篇文章可以看懂!( 三 )
/** * @Description 限流的注解,标注在类上或者方法上 。在方法上的注解会覆盖类上的注解,同@Transactional * @Author CJB * @Date 2020/3/20 13:36 */@Inherited@Target({ElementType.TYPE, ElementType.METHOD})@Retention(RetentionPolicy.RUNTIME)public @interface RateLimit {/*** 令牌桶的容量,默认100* @return*/int capacity() default 100;/*** 每秒钟默认产生令牌数量,默认10个* @return*/int rate() default 10;}/** * @Description 限流的拦器 * @Author CJB * @Date 2020/3/19 14:34 */@Componentpublic class RateLimiterIntercept implements HandlerInterceptor {@Overridepublic boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {if (handler instanceof HandlerMethod){HandlerMethod handlerMethod=(HandlerMethod)handler;Method method = handlerMethod.getMethod();/*** 首先获取方法上的注解*/RateLimit rateLimit = AnnotationUtils.findAnnotation(method, RateLimit.class);//方法上没有标注该注解,尝试获取类上的注解if (Objects.isNull(rateLimit)){//获取类上的注解rateLimit = AnnotationUtils.findAnnotation(handlerMethod.getBean().getClass(), RateLimit.class);}//没有标注注解,放行if (Objects.isNull(rateLimit))return true;//尝试获取令牌,如果没有令牌了if (!RedisLimiterUtils.tryAcquire(request.getRequestURI(),rateLimit.capacity(),rateLimit.rate())){//抛出请求超时的异常throw newTimeOutException();}}return true;}} - SpringBoot配置拦截器的代码就不贴了,以上就是完整的代码,至此分布式限流就完成了 。
- 如果觉得作者写的好,有所收获的话,点个关注推荐一下哟!!!
推荐阅读
-
-
金星|【9.29日播报】金星换座 天秤白羊双子接桃花啦!
-
-
对娱乐的兴趣|全是钢铁大“直男”,那个是你的爱豆?,公认“0绯闻”的4位男星
-
下饭视频|熟龄女人别穿太老气,学学时尚博主的优雅通勤风穿搭,减龄又气质
-
硕士毕业生留18篇日记后殉情|硕士毕业生留18篇日记后殉情 靠爱心人士资助上学
-
-
陶式防务评论|破坏丹麦战斗机公平竞标,丹麦媒体挺身而出:曝光美国搞间谍活动
-
白宫|大批美军撤离中东!枪口对准真正心腹大患,白宫:好戏已经开演
-
凤凰娱乐|夏雨10岁女儿近照曝光,身材高挑颜值高,完美继承妈妈好基因
-
-
-
萧敬腾携林有慧游挪威!晒同框合照超有夫妻相?高调秀恩爱!
-
乐居财经|实现净利润18.11亿元,时代中国控股上半年销售额稳中有升
-
-
-
“全国学雷锋先进个人”张义:慧心雕刻人生 倾情回馈社会
-
金亚军事■用436发子弹灭敌214人,回来后为何不再碰枪了?,靠一杆老式步枪
-
-