- 主页 > 生活百科 > >
接口限流还不太明白? 这篇文章可以看懂!( 三 )
/** * @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配置拦截器的代码就不贴了,以上就是完整的代码,至此分布式限流就完成了 。
- 如果觉得作者写的好,有所收获的话,点个关注推荐一下哟!!!
推荐阅读
-
「永恒晨曦」失业率攀升, 或是故意所为,世界人民血汗钱被美国政府发给美国人,
-
-
Angelia妈咪|说明胎儿进入猛涨期,饮食注意“4要2不要”,孕妈出现四个信号
-
星座人缘知识汇|越容易被深爱的三大星座,惹人怜爱!,越患得患失
-
-
-
-
-
-
甘肃省医疗卫生重点专业(兰大二院功能神经外科专业)癫痫术前评估及外科技术学习班暨2020西部(兰州)癫痫外科论坛顺利举办
-
辽宁省pe管生产厂家--8890政务便民服务大数据平台怎么下载?
-
-
-
手机大魔王▲3S无界瀑布屏即将惊艳亮相,打破边界,触碰未来,NEX
-
-
华为|华为新一代“小方表”来了:Watch FIT 2正式官宣
-
-
-
烹饪|大暑过后天天吃,酸爽开胃,与米饭一起吃,上桌就抢光
-
怀特|如果本周比赛爆冷,怀特、波维特金将迎来完全不同的命运