- 主页 > 生活百科 > >
接口限流还不太明白? 这篇文章可以看懂!( 三 )
/** * @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配置拦截器的代码就不贴了,以上就是完整的代码,至此分布式限流就完成了 。
- 如果觉得作者写的好,有所收获的话,点个关注推荐一下哟!!!
推荐阅读
-
大卫聊科技 中科院强势入局!“卡脖子”项目变专研重点!,华为断供才一天
-
带你全面认识宝骏510 宝骏510均衡器最佳效果图
-
-
肥胖|美式饮食“威胁国家安全”:肥胖成美军一大祸害
-
-
布达拉宫|布达拉宫内的三大未解之谜,神秘流传着千年之久!
-
-
【取消】再见,铁饭碗!2020年全面取消事业编制!这3类人即将变
-
和男友订婚后,他拒绝在全款房本上加我名字:人性经不起考验
-
[铜钱]盘点农村中,比较值钱的5中“老物件”,你老家有吗?
-
招聘|回乡工作的人都怎样了?大城市、小城市的工作之路有什么区别?
-
-
历史驿栈|年少的太子就霸占了后妈,却不小心造就一位千古明君,皇帝刚去世
-
青岛已采样247247份进行检测 连续3天未发现新增阳性病例
-
说穿|穿越她说穿到古代多的是暖男“听说你在找男人看我可不可”
-
-
腾讯科技用思维控制iPhone?马斯克希望年底前用脑机接口帮助瘫痪病人
-
穿搭@《老友记》里慵懒迷人的复古穿搭,又飒又美,看背影就好心动
-
-
十一区@装运前却遭美国开高价截标,法国向中国订购口罩