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

时间:2022-09-17 23:40:38

  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代表以公平的方式获取信号量,即先进先出的原则。