java代码实现服务限流-分布式限流

       在当今的系统建设中，“XX平台”类、“XX开放XX”类系统 越来越成为各家公司看重的主阵地，这就涉及到分布式系统的服务降级、增加缓存、限流等功能。

      这次聊一聊限流，

    

一、漏桶算法

漏桶算法思路很简单，水（请求）先进入到漏桶里，漏桶以一定的速度出水，当水流入速度过大会直接溢出，可以看出漏桶算法能强行限制数据的传输速率

对于很多应用场景来说，除了要求能够限制数据的平均传输速率外，还要求允许某种程度的突发传输。这时候漏桶算法可能就不合适了，令牌桶算法更为适合。

 

二、令牌桶算法

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

 

注意： 1、两者主要区别在于“漏桶算法”能够强行限制数据的传输速率，而“令牌桶算法”在能够限制数据的平均传输速率外，还允许某种程度的突发传输。在“令牌桶算法”中，只要令牌桶中存在令牌，那么就允许突发地传输数据直到达到用户配置的门限，所以它适合于具有突发特性的流量。

          2、上面的两种限流算法都只适合单机模式下，不适合分布式系统的限流

三、使用Guava的RateLimiter进行限流控制

RateLimiter其实也是使用了令牌桶算法实现的。

//创建 RateLimiter  对象，表示每秒只能访问两个线程，即每秒也向桶中放两个令牌
private static final RateLimiter rateLimiter = RateLimiter.create(2);

  // 请求RateLimiter, 超过permits会被阻塞
    rateLimiter.acquire();

 这个acquire()方法也就是 RateLimiter的启动方法，只有调用acquire（）方法才能起到限流的作用。

Demo1、

public class RateLimiterTest {

   //表示每秒只能访问两个线程，即每秒也向桶中放两个令牌
    private static final RateLimiter rateLimiter = RateLimiter.create(2);

    public static void main(String args[]) {
        ExecutorService executor = Executors.newFixedThreadPool(10);
        for (int i = 0; i < 10; i++) {
            executor.execute(new Runnable() {
                @Override
                public void run() {
                    // 请求RateLimiter, 超过permits会被阻塞
                    rateLimiter.acquire();
                    //每秒检查桶中有多少令牌，下面的意思是检查每秒桶中是否有5个令牌，有的话(返回ture)，RateLimiter的限流每秒2个才起作用。
                    boolean flag = false;
                    // boolean flag = rateLimiter.tryAcquire(5,TimeUnit.SECONDS);
                    System.out.println(flag+":thread"+Thread.currentThread()+",当前时间:"+new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()));
                }
            });
        }
        executor.shutdown();
        while (!executor.isTerminated()) {
        }
        System.out.println("Finished all threads");
    }

}

运行测试类，可以看出基本上是每秒执行两个线程。

///每秒检查桶中有多少令牌，意思是检查每秒桶中是否有5个令牌，有的话(返回ture)，RateLimiter
//的限流每秒2个才起作用。
boolean flag = rateLimiter.tryAcquire(5,TimeUnit.SECONDS);

public class RateLimiterTest {

   //表示每秒只能访问两个线程，即每秒也向桶中放两个令牌
    private static final RateLimiter rateLimiter = RateLimiter.create(2);

    public static void main(String args[]) {
        ExecutorService executor = Executors.newFixedThreadPool(10);
        for (int i = 0; i < 10; i++) {
            executor.execute(new Runnable() {
                @Override
                public void run() {
                    //每秒检查桶中有多少令牌，下面的意思是检查每秒桶中是否有5个令牌，有的话(返回ture)，RateLimiter的限流每秒2个