RateLimiter 的底层实现是啥?
实现 底层
2023-09-27 14:22:38 时间
作者:温安适
来源:https://my.oschina.net/floor/blog/4965200
前言
本文不是一个RateLimiter的详细分析,仅仅是概要分析。
令牌桶算法
一说到RateLimiter,必然要是说的令牌桶,它的大致逻辑如下
按图实现
令牌桶的图,网上到处可见,按图实现也非常简单,无非是定时添加令牌桶,并提供一个获取令牌的函数,博主实现了一遍代码如下:
import java.util.concurrent.*;
public class MyRateLimiter {
//令牌桶
BlockingQueue<Integer>TOKEN_BUCKET=new LinkedBlockingDeque<>(5);
public static void main(String[] args) {
MyRateLimiter myRateLimiter=new MyRateLimiter();
myRateLimiter.addTokenFixedRate();
for(int i=0;i<10;i++){
myRateLimiter.acqurie();
System.out.println("第几次执行i:" + i + ",执行时间为:" + System.currentTimeMillis());
}
}
/**
* 定时添加令牌
*/
public void addTokenFixedRate(){
ScheduledExecutorService scheduledExecutorService= Executors.newSingleThreadScheduledExecutor();
scheduledExecutorService.scheduleAtFixedRate(()->{
boolean suc=TOKEN_BUCKET.offer(1);
if(!suc){
System.out.println("令牌桶满了丢弃");
}
},0,200,TimeUnit.MILLISECONDS);
}
public void acqurie(){
while (TOKEN_BUCKET.poll()==null){};
}
}
测试结果如下,基本满足要求
RateLimiter概要实现
我一开始是按照自己实现的逻辑,去查看Guava的RateLimiter的源码的,结果发现RateLimiter根本没有集合充当桶,核心是记录了下一令牌产生的时间与现存令牌数,并动态更新它们。
概要逻辑图如下:
按照这个图看核心代码就比较容易了,摘录核心代码如下:
@CanIgnoreReturnValue
public double acquire(int permits) {
long microsToWait = reserve(permits);
stopwatch.sleepMicrosUninterruptibly(microsToWait);
return 1.0 * microsToWait / SECONDS.toMicros(1L);
}
//Reserve 一路向下能查到如下代码 reserveEarliestAvailable
final long reserveEarliestAvailable(int requiredPermits, long nowMicros) {
resync(nowMicros);
long returnValue = nextFreeTicketMicros;
// 现存令牌可以提供的令牌数
double storedPermitsToSpend = min(requiredPermits, this.storedPermits);
//需要刷新的令牌数
double freshPermits = requiredPermits - storedPermitsToSpend;
//等待时间=需要刷新的令牌数*固定间隔+存储许可等待时间
long waitMicros =
storedPermitsToWaitTime(this.storedPermits, storedPermitsToSpend)
+ (long) (freshPermits * stableIntervalMicros);
//下次令牌生产时间=本次令牌生产时间+等待时间
this.nextFreeTicketMicros = LongMath.saturatedAdd(nextFreeTicketMicros, waitMicros);
this.storedPermits -= storedPermitsToSpend;
return returnValue;
}
总结:RateLimiter根本没有集合充当桶,核心是记录了下一令牌产生的时间与现存令牌数,并动态更新它们。
近期热文推荐:
1.1,000+ 道 Java面试题及答案整理(2021最新版)
2.终于靠开源项目弄到 IntelliJ IDEA 激活码了,真香!
3.阿里 Mock 工具正式开源,干掉市面上所有 Mock 工具!
4.Spring Cloud 2020.0.0 正式发布,全新颠覆性版本!
觉得不错,别忘了随手点赞+转发哦!
相关文章
- 从源码分析快速实现对新开源软件的检测
- 深度解析HashMap底层实现架构
- Apache Flink实现的数据流体系结构
- vue入门:(底层渲染实现render函数、实例生命周期)
- 替代Docker Compose实现容器双向联通的三种方法
- 序列化和反序列化的底层实现原理是什么?
- 数据库中的索引以及索引的底层实现原理(平衡二叉树)
- 基于Springboot+freemarker+Mysql实现的宿舍维修上报管理系统
- Android滤镜效果实现及原理分析
- JQuery文本框水印插件的简单实现
- 容器技术反例:哪些不适合利用Docker实现?
- Android 屏幕截图(底层实现方式)
- [五]类加载机制双亲委派机制 底层代码实现原理 源码分析 java类加载双亲委派机制是如何实现的
- JAVA Socket 底层是怎样基于TCP/IP 实现的???
- Laravel和Lumen的Redis底层实现
- docker基本元素和底层实现
- redis-15 zset 底层跳表 skiplist 实现
- JS实现下雪效果
- Spring5入门到实战------9、AOP基本概念、底层原理、JDK动态代理实现
- php+js+ajax实现简单的回帖功能(适合新手)
- javascript实现倒计时-------Day28
- mxGraph画图区域使用鼠标滚轮实现放大/缩小
- map底层为什么要用红黑树实现
- WCF技术剖析之二十二: 深入剖析WCF底层异常处理框架实现原理[上篇]
- Delphi 悬浮窗口、浮动窗口的实现
- clip-path 教程:使用 CSS 中的 clip-path 轻松实现多边形
- 怎么从零编写一个 v3 版本的 chrome 浏览器插件实现 CSDN 博客网站的暗黑和明亮主题切换?
- 在鸿蒙开发板上使用MQTT协议、OneNET实现第一个物联网项目
- android webview 底层实现的逻辑
- MySQL:索引的底层实现 | 聚集索引和非聚集索引 | 自适应哈希索引
- 《Redis设计与实现》阅读:Redis底层研究之简单动态字符串SDS