【229期】Spring Boot 使用令牌桶算法+拦截器+自定义注解+自定义异常实现简单的限流...
点击上方“Java精选”,选择“设为星标”
别问别人为什么,多问自己凭什么!
下方有惊喜,留言必回,有问必答!
每天 08:15 更新文章,每天进步一点点...
令牌桶
在高并发的情况下,限流是后端常用的手段之一,可以对系统限流、接口限流、用户限流等,本文就使用令牌桶算法+拦截器+自定义注解+自定义异常实现限流的demo。
令牌桶思想
大小固定的令牌桶可自行以恒定的速率源源不断地产生令牌。如果令牌不被消耗,或者被消耗的速度小于产生的速度,令牌就会不断地增多,直到把桶填满。
后面再产生的令牌就会从桶中溢出。最后桶中可以保存的最大令牌数永远不会超过桶的大小。然后每个访问的用户都会从中取走一块令牌,取到了令牌才能访问,如果没取到令牌即代表已达到访问上限,将被限流不允许访问
限流demo实现思路
创建令牌桶类
项目启动初始化令牌桶,并设置定时器,定时向桶内放入令牌
自定义限流注解,在需要限流的接口上打上注解
配置令牌桶拦截器,对所有路径进行拦截,对无限流注解的接口直接放行,对有限流注解的做取令牌处理,取到令牌即放行,没取到令牌即抛出自定义异常
自定义异常并使用AOP做全局异常处理
创建令牌类
这里为了防止并发问题在生成令牌和取令牌的方法上加了synchronized
BucketUtil如下
public class BucketUtil {//默认容量10static final int DEFAULT_MAX_COUNT = 10;// 默认增长速率为1static final int DEFAULT_CREATE_RATE = 1;// 使用HashMap存放令牌桶,这里默认为10个令牌桶public static HashMap<String, BucketUtil> buckets = new HashMap(10);//自定义容量,一旦创建不可改变final int maxCount;//自定义增长速率1s几个令牌int createRate;//当前令牌数int size=0;// 默认令牌桶的容量及增长速率public BucketUtil() {maxCount = DEFAULT_MAX_COUNT;createRate = DEFAULT_CREATE_RATE;}// 自定义令牌桶容量及增长速率public BucketUtil(int maxCount, int createRate) {this.maxCount = maxCount;this.createRate = createRate;}public int getSize() {return size;}public boolean isFull() {return size == maxCount;}//根据速率自增生成一个令牌public synchronized void incrTokens() {for (int i = 0; i < createRate; i++){if (isFull())return;size++;}}// 取一个令牌public synchronized boolean getToken() {if (size > 0)size--;elsereturn false;return true;}@Overridepublic boolean equals(Object obj) {if (obj == null)return false;BucketUtil bucket = (BucketUtil) obj;if (bucket.size != size || bucket.createRate != createRate || bucket.maxCount != maxCount)return false;return true;}@Overridepublic int hashCode() {return Objects.hash(maxCount, size, createRate);}}
初始化令牌桶
在启动类上初始化并生成定时器
推荐下自己几个月熬夜整理的面试资料大全:
https://gitee.com/yoodb/ebooks
@EnableScheduling
@SpringBootApplication
public class DemoApplication {public static void main(String[] args) {SpringApplication.run(DemoApplication.class, args);// 为了方便测试这里定义1容量 1增长速率BucketUtil bucketUtil = new BucketUtil(1,1);// 生成名为:bucket的令牌桶BucketUtil.buckets.put("bucket",bucketUtil);}@Scheduled(fixedRate = 1000)// 定时1spublic void timer() {if (BucketUtil.buckets.containsKey("bucket")){//名为:bucket的令牌桶 开始不断生成令牌BucketUtil.buckets.get("bucket").incrTokens();}}
}
自定义注解以及异常
BucketAnnotation注解
@Target({ElementType.METHOD})// METHOD代表是用在方法上
@Retention(RetentionPolicy.RUNTIME)
public @interface BucketAnnotation {
}
APIException 异常
public class APIException extends RuntimeException {private static final long serialVersionUID = 1L;private String msg;public APIException(String msg) {super(msg);this.msg = msg;}
}
配置拦截器
BucketInterceptor拦截器如下
/*** 令牌桶拦截器*/
public class BucketInterceptor implements HandlerInterceptor {// 预处理回调方法,在接口调用之前使用 true代表放行 false代表不放行@Overridepublic boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {if (!(handler instanceof HandlerMethod)) {return true;}HandlerMethod handlerMethod = (HandlerMethod) handler;Method method = handlerMethod.getMethod();BucketAnnotation methodAnnotation = method.getAnnotation(BucketAnnotation.class);if (methodAnnotation!=null){// 在名为:bucket的令牌桶里取令牌 取到即放行 未取到即抛出异常if(BucketUtil.buckets.get("bucket").getToken()){return true;}else{// 抛出自定义异常throw new APIException("不好意思,您被限流了");}}else {return true;}}// 接口调用之后,返回之前 使用@Overridepublic void postHandle(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, ModelAndView modelAndView) throws Exception {}// 整个请求完成后,在视图渲染前使用@Overridepublic void afterCompletion(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, Exception e) throws Exception {}
}
将拦截器注入
@Configuration
public class WebMvcConfg implements WebMvcConfigurer {@Overridepublic void addInterceptors(InterceptorRegistry registry) {// 令牌桶拦截器 添加拦截器并选择拦截路径registry.addInterceptor(bucketInterceptor()).addPathPatterns("/**");}@Beanpublic BucketInterceptor bucketInterceptor() {return new BucketInterceptor();}
}
AOP全局异常处理
@RestControllerAdvice
public class WebExceptionControl {@ExceptionHandler(APIException.class)public E3Result APIExceptionHandler(APIException e) {return E3Result.build(400,e.getMessage());}
}
测试
在我们需要限流的接口上打上自定义注解,如下
@BucketAnnotation
@RequestMapping(value = "/bucket")
public E3Result bucket(){return E3Result.ok("访问成功");
}
关于E3Result只是一个封装好的返回类,这里就不贴出来了,大家有的替换成自己的,没有的可以直接用String型测试
推荐下自己做的 Spring boot 的实战项目:
https://gitee.com/yoodb/jing-xuan
上面为了方便测试,令牌桶的容量设置成了1,所以这是取到令牌成功的
这是没取到令牌被限流的
总结
上面的限流只是一个demo还有很多不足的地方,如:
分布式环境下不适用
令牌桶可以有多个,不同的接口采用不同令牌桶的时候,拦截器无法分开限流
一次请求消耗一个令牌,可以被恶意消耗
改进方法:
令牌桶实现采用redis集群存取
注解添加value参数,可以给对应接口打上对应的令牌桶参数,拦截器需对注解参数校验,实现多个接口多个令牌桶的限流
对用户IP校验限制次数,防止恶意攻击
实际项目限流会更加严谨,上述只是提供了一个思路以及演示demo,不喜勿喷谢谢。
版权声明:本文为CSDN博主「Colins~」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
https://blog.csdn.net/weixin_44102992/article/details/106492691
公众号“Java精选”所发表内容注明来源的,版权归原出处所有(无法查证版权的或者未注明出处的均来自网络,系转载,转载的目的在于传递更多信息,版权属于原作者。如有侵权,请联系,笔者会第一时间删除处理!
------ THE END ------
精品资料,超赞福利!
>Java精选面试题<
3000+ 道面试题在线刷,最新、最全 Java 面试题!
期往精选 点击标题可跳转
【221期】扔掉工具类,Mybatis 一个简单配置搞定数据加密解密!
【222期】一个适用于 Spring Boot 项目的轻量级 HTTP 客户端框架
【223期】面试官问:什么是 YAML?和 Spring Boot 有什么关系?
【224期】Java 字符串拼接五种方法的性能比较分析,从执行100次到90万次?
【225期】为什么数据库连接池不采用 IO 多路复用?
【226期】面试问我,HashMap 的默认初始容量是多少,我该怎么说?
【227期】面试官:MySQL 数据查询太多会 OOM 吗?
【228期】面试官:跨库多表存在大量数据依赖问题,有哪些解决方案?
技术交流群!
最近有很多人问,有没有读者&异性交流群,你懂的!想知道如何加入。加入方式很简单,有兴趣的同学,只需要点击下方卡片,回复“加群”,即可免费加入交流群!
文章有帮助的话,在看,转发吧!
【229期】Spring Boot 使用令牌桶算法+拦截器+自定义注解+自定义异常实现简单的限流...相关推荐
- Spring Boot实战:过滤器、拦截器与切片
Spring Boot实战:过滤器.拦截器与切片 Q:使用过滤器.拦截器与切片实现每个请求耗时的统计,并比较三者的区别与联系 过滤器Filter 过滤器概念 Filter是J2E中来的,可以看做是Se ...
- Spring Boot(5) web开发(3)拦截器、文件上传、异常处理
Spring Boot(5) web开发(3)拦截器.文件上传.异常处理 学习视频: https://www.bilibili.com/video/BV19K4y1L7MT?p=49&spm_ ...
- 【转载】Spring Boot 过滤器、监听器、拦截器的使用
关注"Java后端技术全栈" 回复"面试"获取全套大厂面试资料 在开发中用到过滤器.监听器.拦截器的场景非常多,今天就来聊聊这三者在日常开发中是如何使用的. 概 ...
- 使用令牌桶算法解决调用第三方接口限流问题
我们在调用第三方接口时常常会碰到接口限流问题,为了解决这一问题,大家想出了许多方法.我这里介绍一下我的方法,第三方接口限流一般是基于令牌桶算法的,那么我们可以以彼之道还治彼身,使用令牌桶算法实现我方调 ...
- SpringBoot基于guava集成令牌桶算法
SpringBoot基于guava集成令牌桶算法 一.什么是令牌桶 1.令牌桶 2.功能图 二.Guava 1.简单介绍 2.pom引入 3.限速器 4.浅析重载方法 三.系统应用 1.单个接口应用 ...
- 【232期】面试官:如何保护 Spring Boot 配置文件敏感信息?
点击上方"Java精选",选择"设为星标" 别问别人为什么,多问自己凭什么! 下方有惊喜,留言必回,有问必答! 每天 08:15 更新文章,每天进步一点点... ...
- 【230期】Spring Boot 集成 Elasticsearch7.6 实现高亮分词及简单查询
点击上方"Java精选",选择"设为星标" 别问别人为什么,多问自己凭什么! 下方有惊喜,留言必回,有问必答! 每天 08:15 更新文章,每天进步一点点... ...
- 限流10万QPS、跨域、过滤器、令牌桶算法-网关Gateway内容都在这儿
点击上方 好好学java ,选择 星标 公众号 重磅资讯.干货,第一时间送达 今日推荐:硬刚一周,3W字总结,一年的经验告诉你如何准备校招! 个人原创100W+访问量博客:点击前往,查看更多 作者:雄 ...
- 网关限流(令牌桶算法)
一.需求: 每个ip地址1秒内只能发送1次请求,多出来的请求返回429错误. 二.引入依赖 spring cloud gateway 默认使用redis的RateLimter限流算法来实现.所以我们要 ...
最新文章
- mysql没有frm文件_【1.1】mysql frm文件丢失(ibd文件丢失)
- 用递归的方法求s = k! + n! / m!的值
- 客户端读写hdfs数据
- call/apply/bind 的理解与实例分享
- 计算机硬件的基本组成概述
- 长安大学计算机科学与技术(交通信息工程)课程专业课,长安大学硕士专业交通信息工程及控制介绍...
- PHP 表单文件上传的原理,php多文件上传功能实现原理及代码
- BZOJ 1007 水平可见直线
- WCF分布式开发常见错误(13):此方法调用的事务被异步中断
- 深度神经网络反向传播算法
- Aspose.word保存PDF时进行授权访问设置
- 计算机无法复制大文件格式,秒解U盘复制文件电脑提示文件过大无法复制
- 计算机专业毕业祝福语,大学毕业生祝福语
- Android 获取本地视频列表
- There‘s no Qt version assigned to project xxx.vcxproj for configuration Debug/x64
- 多线程3——线程连接、分离和取消(linux)
- C# 提取Word中插入的多媒体文件(视频、音频)
- 《微SaaS创富周刊》第5期:基于A/B测试的20+创富项目盘点
- 全球著名大学计算机视觉相关实验室
- FC总线基础知识(1)
热门文章
- Python语音基础操作--5.1自适应滤波
- 详细的EVE-NG部署方法+EVE各种镜像与模板,附带SCRT+WINSCP等工具提供(VM部署)
- 由VC2010与VC2017数据结构差异造成的程序错误
- EC20 4G LTE模块在FriendlyElec友善电子的R2S软路由(运行FriendlyWrt)上的应用演示
- python开发的软件可以申请软著吗_python自动化生成软件著作权的源代码
- pdf转换工具怎么用?手把手教会你~
- iPhone免密支付被盗刷上千万,三招教你保护钱包安全!
- 阿里云ECS服务器租用价格表与收费标准参考
- 使用标注工具Labelme和PaddleSeg实现铁路的图像分割
- FPGA设计开发(基础课题):七人表决器设计