JAVA中限制接口流量、并发的方法

　　JAVA中限制接口流量可以通过Guava的RateLimiter类或者JDK自带的Semaphore类来实现，两者有点类似，但是也有区别，要根据实际情况使用。简单来说，

RateLimiter类是控制以一定的速率访问接口。

Semaphore类是控制允许同时并发访问接口的数量。

 

一、RateLimiter类

　　RateLimiter翻译过来是速率限制器，使用的是一种叫令牌桶的算法，当线程拿到桶中的令牌时，才可以执行。通过设置每秒生成的令牌数来控制速率。使用例子如下：

public class TestRateLimiter implements Runnable {

    public static final SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

    public static final RateLimiter limiter = RateLimiter.create(1); // 允许每秒最多1个任务

    public static void main(String[] arg) {
        for (int i = 0; i < 10; i++) {
            limiter.acquire(); // 请求令牌,超过许可会被阻塞
            Thread t = new Thread(new TestRateLimiter());
            t.start();
        }
    }

    public void run() {
        System.out.println(sdf.format(new Date()) + " Task End..");
    }
}

 

二、Semaphore类

　　Semaphore翻译过来是信号量，通过设置信号量总数，当线程拿到信号量，才可以执行，当实行完毕再释放信号量。从而控制接口的并发数量。使用例子如下：

public class TestSemaphore implements Runnable {

    public static final SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
    
    public static final Semaphore semaphore = new Semaphore(5, true); // 允许并发的任务量限制为5个
    
    public static void main(String[] arg) {
        for (int i = 0; i < 10; i++) {
             Thread t = new Thread(new TestSemaphore());
             t.start();
        }
    }

    public void run() {
        try {
            semaphore.acquire(); // 获取信号量,不足会阻塞
            System.out.println(sdf.format(new Date()) + " Task Start..");
            Thread.sleep(5000);
            System.out.println(sdf.format(new Date()) + " Task End..");
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            semaphore.release(); // 释放信号量
        }
    }
}

　　初始化中的第二个参数true代表以公平的方式获取信号量，即先进先出的原则。