- 主页 > 生活百科 > >
接口限流还不太明白? 这篇文章可以看懂!( 三 )
/** * @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配置拦截器的代码就不贴了,以上就是完整的代码,至此分布式限流就完成了 。
- 如果觉得作者写的好,有所收获的话,点个关注推荐一下哟!!!
推荐阅读
-
四月不减肥 六月徒悲伤!对于骨架大,肉结实的人应该咋瘦呢在线等,急!?
-
-
-
-
中新网|70城房价整体上涨加快!买房要趁“金九银十”出手?
-
-
天天聊游戏|网友:很正常!,“吃鸡”光子江郎才尽?升级枪出现“自我模仿”
-
-
载风月|?已经开始了?11家苹果店被“关停”!库克慌了:中国市场很重要
-
57岁香港富豪带新女友亮相,维持“3女共侍1夫”生活超10年
-
张玉环谈前妻:以后只是朋友关系|张玉环谈前妻:以后只是朋友关系 好好生活!!
-
-
穆迪等多家国际权威机构上调中国经济增长预期 为何?
-
歪妈知道|第15天被儿媳赶走,有些事真不能忍,婆婆带俩外孙来“伺候”月子
-
-
-
#儿科医生刘维民#是靠宝妈在孕期这些食物的摄取,准妈妈务必知晓,宝宝的高颜值
-
-
[惩戒]管住“熊孩子”,教育惩戒规则还得迈过这些“坎”
-
脸有些方的脸留什么发型好看呢,方脸用什么发型好看-