RateLimiter 的底层实现是啥？-

沙海

3180
文章

2
评论

2021年6月9日03:56:33Java评论37字数 2599阅读8分39秒阅读模式

摘要

RateLimiter 的底层实现是啥？点击关注 ? 小哈学Java

RateLimiter 的底层实现是啥？

点击关注 ? 小哈学Java 文章源自JAVA秀-https://www.javaxiu.com/30466.html

点击上方蓝色“小哈学Java”，选择“设为星标”回复“资源”获取独家整理的学习资料！

作者：温安适来源：https://my.oschina.net/floor/blog/4965200文章源自JAVA秀-https://www.javaxiu.com/30466.html

前言

本文不是一个RateLimiter的详细分析，仅仅是概要分析。文章源自JAVA秀-https://www.javaxiu.com/30466.html

令牌桶算法

一说到RateLimiter，必然要是说的令牌桶，它的大致逻辑如下：文章源自JAVA秀-https://www.javaxiu.com/30466.html

RateLimiter 的底层实现是啥？文章源自JAVA秀-https://www.javaxiu.com/30466.html

按图实现

令牌桶的图，网上到处可见，按图实现也非常简单，无非是定时添加令牌桶，并提供一个获取令牌的函数，博主实现了一遍代码如下：文章源自JAVA秀-https://www.javaxiu.com/30466.html

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){};    }}

测试结果如下,基本满足要求文章源自JAVA秀-https://www.javaxiu.com/30466.html

RateLimiter 的底层实现是啥？文章源自JAVA秀-https://www.javaxiu.com/30466.html

RateLimiter概要实现

我一开始是按照自己实现的逻辑，去查看Guava的RateLimiter的源码的，结果发现RateLimiter根本没有集合充当桶，核心是记录了下一令牌产生的时间与现存令牌数，并动态更新它们。文章源自JAVA秀-https://www.javaxiu.com/30466.html

概要逻辑图如下：

RateLimiter 的底层实现是啥？文章源自JAVA秀-https://www.javaxiu.com/30466.html

按照这个图看核心代码就比较容易了，摘录核心代码如下：文章源自JAVA秀-https://www.javaxiu.com/30466.html

@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根本没有集合充当桶，核心是记录了下一令牌产生的时间与现存令牌数，并动态更新它们。文章源自JAVA秀-https://www.javaxiu.com/30466.html

1. 熟练掌握 MyBatis-Plus，这一篇就够了！2. 运维发飙了：你们服务内存占用太高，还只增不减！告警了，快来接锅3. 为什么不建议使用 Java 自带的序列化？4. 数据库连接池到底应该设多大？这次直接从100ms优化到3ms！最近面试BAT，整理一份面试资料《Java面试BATJ通关手册》，覆盖了Java核心技术、JVM、Java并发、SSM、微服务、数据库、数据结构等等。获取方式：点“在看”，关注公众号并回复 Java 领取，更多内容陆续奉上。

文章有帮助的话，在看，转发吧。文章源自JAVA秀-https://www.javaxiu.com/30466.html

谢谢支持哟 (*^__^*)文章源自JAVA秀-https://www.javaxiu.com/30466.html

继续阅读