全网最简洁Java实现多线程安全的令牌桶限流算法

在许多应用中，我们需要限制某些操作的频率，例如，限制API调用的速率，防止系统被过度使用，这种需求就需要一个限流算法来满足

令牌桶算法是一种常用的限流算法

它的基本思想是：系统以恒定的速率向桶中添加令牌，当桶满时，新添加的令牌会被丢弃。当一个请求到达时，系统会尝试从桶中取出一个令牌，如果桶中没有令牌，则请求被拒绝，这种算法常用于网络流量整形和流量控制。

实现原理

我们需要实现一个令牌桶限流算法，主要包括以下几个步骤：

查看：👉 实现原理

Java实现

我们要求这个令牌桶限流算法是多线程安全的，因此我们需要使用线程安全的数据结构来存储桶中的令牌数，这里我们使用AtomicInteger来表示桶中的令牌数。

获取请求的时间可以使用 System.currentTimeMillis()，这个时间戳可以精确到毫秒级别。

代码如下：👉 查看源码

测试

👉 查看测试与完整代码

入职啦实战项目交流

入职啦官方公众号