文字摘抄于:https://yq.aliyun.com/articles/632589
Semaphore(信号量)是用来控制同时访问特定资源的线程数量,它通过协调各个线程,以保证合理的使用公共资源。换句话说,锁(Lock锁或synchronized锁)在任何时刻只允许一个任务访问被加锁的资源,而计数信号量允许n个任务同时访问这个资源,还可以将信号量看作是向外分发使用资源的“许可证”,尽管内部没有这种许可证对象。
“许可证”的数量是有限的,所以当执有“许可证”的线程数量与“许可证”数量相同时,就会阻止其他线程对共享资源的使用,如果某一个或多个线程使用完共享资源后,就会归还“许可证”,此时Semaphore(信号量)就会将这些归还的“许可证”再次分发给阻塞中的线程。通过这种方式就实现了控制线程并发数。
应用:
Semaphore可以用于做流量控制,特别是公用资源有限的应用场景,比如数据库连接。假如有一个需求,要读取几万个文件的数据,因为都是IO密集型任务,我们可以启动几十个线程 并发地读取,但是如果读到内存后,还需要存储到数据库中,而数据库的连接数只有10个,这 时我们必须控制只有10个线程同时获取数据库连接保存数据,否则会报错无法获取数据库连 接。这个时候,就可以使用Semaphore来做流量控制。
代码
public static void main(String[] args) throws InterruptedException {
// 同一时间,允许最大线程数量获取
final MySemaphore semaphore = new MySemaphore(1);
for(int i = 0; i<2; i++ )
new Thread(() -> {
try {
// 获取信号量
semaphore.acquire();
//boolean b = semaphore.tryAcquire(); 尝试去拿,这个方法有入参
// semaphore.drainPermits(); // 获取所有的许可证
TimeUnit.SECONDS.sleep(2);
} catch (InterruptedException e) {
e.printStackTrace();
}finally {
// 释放
semaphore.release();
}
}).start();
while(true){
System.out.println("AP: " + semaphore.availablePermits()); // 当前可用的信号量
System.out.println("QL: " + semaphore.getQueueLength()); // 阻塞的信号量
semaphore.hasQueuedThreads(); // 是否有等待线程
Collection<Thread> queuedThreads = semaphore.getQueuedThreads(); // 在等待的线程队列
TimeUnit.SECONDS.sleep(1);
}
}
static class MySemaphore extends Semaphore{
public MySemaphore(int permits) {
super(permits);
}
public MySemaphore(int permits, boolean fair) {
super(permits, fair);
}
@Override
protected Collection<Thread> getQueuedThreads() {
return super.getQueuedThreads();
}
}