- 主页 > 生活百科 > >
接口限流还不太明白? 这篇文章可以看懂!( 三 )
/** * @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配置拦截器的代码就不贴了,以上就是完整的代码,至此分布式限流就完成了 。
- 如果觉得作者写的好,有所收获的话,点个关注推荐一下哟!!!
推荐阅读
-
-
-
九江新闻网|瑞昌市黄金乡文化站开展老年人健步走活动
-
幽谷乔木■虚拟体育大放光彩?,英超:曼联皇马有意万博狼队前锋希门尼斯
-
-
「朱一龙」三十几的朱一龙,一身卡其色工装连体裤,活力十足
-
国际丨特朗普乘坐“空军一号”差点撞上无人机 操作者意图尚不明
-
-
通信|红星资本局|他可能是中国最有钱的大学校长:死磕量子通信24年,一朝上市身家超24亿
-
品质|十分亮眼!洋河股份披露三季报,白酒业绩营利双增,更好迎接下一个十年
-
-
青年|《梦想世界3D》如何有效提升修炼等级通过功绩提升更加便捷
-
-
炸小酥肉时,别直接挂糊,大厨教我1招,酥香嫩滑,放凉也不硬
-
-
京师幼儿园举办"庆元旦 忆民俗"土家风情大联欢活动
-
戴利亚|14岁少女生下女婴,10岁男友疑“喜当爹”!警方正在调查
-
-
-