- 主页 > 生活百科 > >
接口限流还不太明白? 这篇文章可以看懂!( 三 )
/** * @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配置拦截器的代码就不贴了,以上就是完整的代码,至此分布式限流就完成了 。
- 如果觉得作者写的好,有所收获的话,点个关注推荐一下哟!!!
推荐阅读
-
-
有没有用过火星人x2集成灶的吗厨房想做成半开放的,用这款集成灶能行吗
-
-
-
-
杨幂收图收到手机要放不下了|杨幂收图收到手机要放不下了是怎么回事?终于真相了,原来是这样!
-
东球弟|郎平为何不顾规定坚持16人参加领奖呢?,领奖台只能站14个人
-
睡觉@睡觉时身体突然抖一下,是大脑在试探死亡?科学解释来了
-
『明星大小事』让我们看看,撒哈拉橙色牧马人汽车收藏认证站
-
疫情|?法媒:中国经济已回到疫前水平 服务业向好释放积极信号
-
-
旧街凉风|降至冰点,只要成本价,比途岳大气,斯柯达翻身就靠它
-
欧洲杯历史|C罗梅开二度加冕欧洲杯历史射手王 世界冠军对决法国胜
-
-
汽车舆生活@把车钥匙别在腰上的都是什么人?网友笑了:太“LOW”
-
-
-
「三星」特朗普签署行政令:警察只有在极端紧急情况下才可以使用锁喉动作
-
#美国#瞒不住了!比尔盖茨揭开疫情真相,美总统弥天大谎终将完结
-
公司|[公司]东富龙2020年上半年利润1.72亿元 同比大幅增长350.16%