RateLimiter 的底层实现是啥?
发布日期:2021-06-30 13:02:12
浏览次数:2
分类:技术文章
本文共 2297 字,大约阅读时间需要 7 分钟。
作者:温安适
来源:前言
本文不是一个RateLimiter的详细分析,仅仅是概要分析。
令牌桶算法
一说到RateLimiter,必然要是说的令牌桶,它的大致逻辑如下
按图实现
令牌桶的图,网上到处可见,按图实现也非常简单,无非是定时添加令牌桶,并提供一个获取令牌的函数,博主实现了一遍代码如下:
import java.util.concurrent.*;public class MyRateLimiter { //令牌桶 BlockingQueueTOKEN_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根本没有集合充当桶,核心是记录了下一令牌产生的时间与现存令牌数,并动态更新它们。
概要逻辑图如下:
按照这个图看核心代码就比较容易了,摘录核心代码如下:
@CanIgnoreReturnValuepublic double acquire(int permits) { long microsToWait = reserve(permits); stopwatch.sleepMicrosUninterruptibly(microsToWait); return 1.0 * microsToWait / SECONDS.toMicros(1L);}//Reserve 一路向下能查到如下代码 reserveEarliestAvailablefinal 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.
2.
3.
4.
5.
觉得不错,别忘了随手点赞+转发哦!
转载地址:https://javastack.blog.csdn.net/article/details/117717398 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!
发表评论
最新留言
逛到本站,mark一下
[***.202.152.39]2024年04月18日 12时43分49秒
关于作者
喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
解决uiautomatorviewer中添加xpath的方法
2019-05-01
Windows Server R2 安装python时报策略不允许的解决方案
2019-05-01
pip无法安装:换成国内镜像
2019-05-01
python安装mysqlclient[MySQLdb]
2019-05-01
性能测试的必要性评估以及评估方法
2019-05-01
性能测试需求分析
2019-05-01
性能测试需求评审
2019-05-01
性能测试实施流程
2019-05-01
Jmeter在多线程当中对某个http请求进行循环读取配置文件
2019-05-01
Python读取配置文件中文乱码问题
2019-05-01
使用Spark读写外部存储介质(Mysql、Hbase、Redis)
2019-05-01
Spark学习——利用Mleap部署spark pipeline模型
2019-05-01
手写LogisticRegression
2019-05-01
SQL经典题目总结
2019-05-01
剑指offer编程题,一周刷完
2019-05-01
推荐系统初学
2019-05-01
Map的遍历和排序
2019-05-01
约瑟夫环问题
2019-05-01
springboot+log4j2体验
2019-05-01