使用Guava的RateLimiter类实现限流功能

常用的限流算法有漏桶算法和令牌桶算法：

        漏桶算法（Leaky Bucket）：请求进入到漏桶中，漏桶以一定的速率漏水。当请求过多时，水直接溢出。

        令牌桶算法（Token Bucket）：以一定速率向桶中放入令牌，如果有请求时，请求会从桶中取出令牌，如果能取到令牌，则可以继续完成请求，否则等待或者拒绝服务。

 

Guava的RateLimiter类基于令牌桶算法实现流量限制。获取令牌的方式有两种：阻塞等待令牌或者取不到立即返回失败。

 

Guava有两种限流模式，一种为稳定模式(SmoothBursty:令牌生成速度恒定)，一种为渐进模式(SmoothWarmingUp:令牌生成速度缓慢提升直到维持在一个稳定值)

 

public class RateLimiterTest {
    private static int threadCount = 10; //线程数
    
	public static void main(String[] args) {
		ExecutorService executorService = Executors.newCachedThreadPool();
		
		//限速器：速率是每秒生成3个许可
		RateLimiter rateLimiter = RateLimiter.create(3);
		
		for(int i=0; i<threadCount; i++){
			executorService.submit(() -> {
				//返回等待时间
				double d = rateLimiter.acquire(); //获取一个许可，可能要等待
				System.out.println(Thread.currentThread().getName() + ": " + d);
				
				//非堵塞地尝试获取许可，如果获取到则返回true，否则返回false
//				boolean b = rateLimiter.tryAcquire(); 
			});
		}
		
		executorService.shutdown();
	}
}