- 主页 > 生活百科 > >
接口限流还不太明白? 这篇文章可以看懂!( 三 )
/** * @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配置拦截器的代码就不贴了,以上就是完整的代码,至此分布式限流就完成了 。
- 如果觉得作者写的好,有所收获的话,点个关注推荐一下哟!!!
推荐阅读
-
胡八一:奇异博士穿上钢铁侠战甲并非概念,漫威终于公布了当时的拍摄场景
-
-
小池百合子再次当选日本东京都知事 领导能力将受考验
-
-
体坛小说|双姜再发威,CBA最强黑马诞生!吉林轰18分大胜北控获6连胜
-
-
[连衣裙]周润发前妻余安安太惊艳了,穿碎花连衣裙配短发,哪敢相信61岁
-
冒汗|酸香爽口、出锅抢着吃,大冬天,家人也能吃得鼻尖冒汗,真爽快!
-
白酒|曝i茅台APP申购成功无法付款 眼睁睁看着到手茅台飞了
-
业态|前三季度文化企业营业收入降幅收窄 文化新业态发展势头强劲
-
4G@4G版iPhone12起售价4198,留给国产手机的时间不多了!
-
#梦幻西游#梦幻西游:经验不够,箱子来凑?结果,玩家却有意外收获!
-
#安兔兔#一加8系列官宣:2K/120Hz屏/骁龙865
-
-
##为什么日本房子都不铺瓷砖?内行人说出实情,真后悔察觉太晚了
-
-
-
小小谈健康既养眼又养生,你想在家中的阳台上种点啥?种点草药吧
-
阿里巴巴|得分率92.3%!阿里云计算能力全球第1,亚马逊、微软、谷歌随后?
-
中国青年网|安倍成为日本连续执政时间最长首相,2799天