- 主页 > 生活百科 > >
接口限流还不太明白? 这篇文章可以看懂!( 三 )
/** * @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配置拦截器的代码就不贴了,以上就是完整的代码,至此分布式限流就完成了 。
- 如果觉得作者写的好,有所收获的话,点个关注推荐一下哟!!!
推荐阅读
-
鹿畔阳光|赵露思和李兰迪长得多像?同一支宣传片,看到侧颜:脸盲了
-
环球科技在此Z1回应了你心中的疑惑,沈义人快看:vivo用iQOO
-
-
青年|中国体坛公认的美女,曾表白易建联被拒绝,为何在19岁选择退役?
-
前方高能|《明月曾照江东寒》发布先导预告,你最期待谁
-
呼吸的鱼|你是不是在祈祷能早一点遇上我,搞笑GIF:妹子
-
-
跟着摄影走天下个性很重要,好身材才是关键,小姐姐的时尚美搭
-
骨折|老年人运动,为什么不要急于求成?掌握这3个常识很重要
-
-
双逸家居|双逸家居 · 卧室布局 |,你的卧室是这样吗?有变化吗?|
-
|一百年前,李鸿章劝告后人说:要想中国没后患,必须灭其国
-
-
车味auto|带“电”后还硬不硬?牧马人插电混动版前瞻
-
丰田c-hr■全系NEDC续航400km,到店实拍丰田C-HR EV
-
上海宝山|一杯奶茶钱让你享受清凉!宝山单次收费泳池大盘点 快来看看
-
-
数码东东 V5被迫降价389!,红米搞事,友商遭殃!realme
-
-
录取通知书不得投递至快件箱|录取通知书不得投递至快件箱