作者：溫安適
來源：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 = https://www.cnblogs.com/javastack/p/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 正式發布，全新顛覆性版本！
5.《Java開發手冊（嵩山版）》最新發布，速速下載！
覺得不錯，別忘了隨手點贊+轉發哦！
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/285555.html
標籤：Java