在Java中如何使用Semaphore实现流量控制
Semaphore是Java中用于控制并发访问资源数量的同步工具，通过维护许可数量限制同时执行的线程数。创建时指定许可数，在关键代码前调用acquire()获取许可，执行完成后在finally块中调用release()释放许可，从而实现流量控制。常用于限制数据库连接、控制API调用频率等场景，需注意避免许可泄露，可选公平模式防止线程饥饿，适用于大多数限流需求。
在Java中，Semaphore 是一种用于控制并发访问资源数量的同步工具，非常适合实现流量控制。它通过维护一定数量的“许可”来限制同时执行某段代码的线程数，从而防止系统因瞬时高并发而崩溃。
什么是Semaphore
Semaphore（信号量）可以理解为一组许可。线程在执行受控操作前必须先获取许可，执行完成后释放许可。如果当前没有可用许可，线程将被阻塞，直到有其他线程释放许可为止。
例如，你希望最多允许10个线程同时访问某个服务接口，就可以初始化一个拥有10个许可的Semaphore。
如何用Semaphore实现流量控制
使用Semaphore进行流量控制的核心思路是：限制并发执行的线程数量，使其不超过系统处理能力。
以下是具体实现步骤：
- 创建Semaphore实例：指定最大并发数（即许可数量）。
- 在关键代码前调用acquire()：尝试获取许可，获取成功则继续执行。
- 执行完任务后调用release()：释放许可，供其他线程使用。
import java.util.concurrent.Semaphore; public class TrafficController { // 允许最多5个线程同时访问 private static final Semaphore semaphore = new Semaphore(5); public static void handleRequest(String requestId) { try { // 获取许可，如果没有可用许可，线程会阻塞 semaphore.acquire(); System.out.println("请求 " + requestId + " 开始处理"); // 模拟业务处理耗时 Thread.sleep(2000); System.out.println("请求 " + requestId + " 处理完成"); } catch (InterruptedException e) { Thread.currentThread().interrupt(); } finally { // 释放许可 semaphore.release(); } } public static void main(String[] args) { // 模拟10个并发请求 for (int i = 1; i <= 10; i++) { new Thread(() -> handleRequest(Thread.currentThread().getName())).start(); } } }
hore;
public class TrafficController {
// 允许最多5个线程同时访问
private static final Semaphore semaphore = new Semaphore(5);
public static void handleRequest(String requestId) {
try {
// 获取许可，如果没有可用许可，线程会阻塞
semaphore.acquire();
System.out.println("请求 " + requestId + " 开始处理");
// 模拟业务处理耗时
Thread.sleep(2000);
System.out.println("请求 " + requestId + " 处理完成");
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
} finally {
// 释放许可
semaphore.release();
}
}
public static void main(String[] args) {
// 模拟10个并发请求
for (int i = 1; i <= 10; i++) {
new Thread(() -> handleRequest(Thread.currentThread().getName())).start();
}
}
}
实际应用场景与注意事项
Semaphore常用于以下场景：
- 限制数据库连接池的连接数
- 控制对第三方API的调用频率
- 保护脆弱的服务不被突发流量压垮
需要注意的是：
- 确保在finally块中调用release()，避免许可泄露。
- 可根据需要选择公平模式（new Semaphore(permits, true)），避免线程饥饿。
- Semaphore只控制并发度，不保证顺序或重入性。
