令牌桶 java_高并发系统限流操作之令牌桶实现可变TPS控制

年前有个需求，批量请求供应商API,要有限流操作，并支持TPS与并发数可配置，那时候简单的查了查资料，任务结束就过去了，最近又有个限流的小需求，所以又翻出了以前的代码。

本次简单记录一下令牌桶的实现：

令牌桶算法的原理是系统会以一个恒定的速度往桶里放入令牌，而如果请求需要被处理，则需要先从桶里获取一个令牌，当桶里没有令牌可取时，则拒绝服务。

令牌桶算法.png

实现思路：

用LinkedBlockingQueue作为装令牌的桶，Executors.newSingleThreadScheduledExecutor()作为定时器定时将令牌放入桶中，使用构建者模式的代码风格。忘了以前在哪抄的了，就这样吧。

贴上核心代码：

import java.util.concurrent.Executors;

import java.util.concurrent.LinkedBlockingQueue;

import java.util.concurrent.ScheduledExecutorService;

import java.util.concurrent.TimeUnit;

/**

* TokenBucket

*/

public class TokenBucket {

/**