通过acquire方法看懂RateLimiter限流机制

最新推荐文章于 2023-10-29 20:23:24 发布

FriendsHelp

最新推荐文章于 2023-10-29 20:23:24 发布

阅读量1.2k

点赞数

分类专栏：限流文章标签： java 开发语言后端

本文链接：https://blog.csdn.net/weixin_45323147/article/details/122130237

版权

本文详细解析了RateLimiter的acquire方法，通过关键的resync和reserveEarliestAvailable方法来阐述限流机制。当令牌不足时，acquire会计算并可能使线程等待。超发情况下，系统会调整下次发放令牌的时间，并在超发时刻释放。文章还提及了单例mutex的实现，并介绍了如何确定下次获取令牌的时间。

摘要由CSDN通过智能技术生成

通过acquire方法看懂RateLimiter限流机制

关键方法

1）resync，动态计算剩余令牌数量和下次发放时间；

2）reserveEarliestAvailable，预定令牌，允许超发，超发后重新计算剩余令牌数量清零，并且会延长nextFreeTicketMicros下一次获取令牌的时间；

如果超发，下一次发放令牌请求，会sleep到超发时刻；

acquire获取令牌的时候，计算下次获取令牌时间，如果没有足够令牌会sleep再返回；

public double acquire(int permits) {
  // 预定令牌
  // mutex控制并发，以达到限流目的
  long microsToWait = reserve(permits);
  // 不可中断的sleep，while循环sleep纳秒级，详情请看源码
  // 发生条件在reserve里
  stopwatch.sleepMicrosUninterruptibly(microsToWait);
  // 返回sleep时间
  return 1.0 * microsToWait / SECONDS.toMicros(1L);
}

预定方法，mutex做同步控制

final long reserve(int permits) {
  // 判断>0
  checkPermits(permits);
  // 同步控制
  synchronized (mutex()) {
    return reserveAndGetWaitLength(permits, stopwatch.readMicros());
  }
}

单例mutex&#