Semaphore
Semaphore(信号量)是用来控制同时访问特定资源的线程数量,它通过协调各个线程,以保证合理的使用公共资源。
一个计数信号量。从概念上讲,信号量维护了一个许可集。如有必要,在许可可用前会阻塞每一个 acquire(),然后再获取该许可。每个 release() 添加一个许可,从而可能释放一个正在阻塞的获取者。但是,不使用实际的许可对象,Semaphore 只对可用许可的号码进行计数,并采取相应的行动。拿到信号量的线程可以进入代码,否则就等待。通过acquire()和release()获取和释放访问许可。
常用方法
Semaphore(int permits, boolean fair) 可以设置该信号量是否采用公平模式,如果以公平方式执行,则线程将会按到达的顺序(FIFO)执行,如果是非公平,则可以后请求的有可能排在队列的头部。
int availablePermits() :返回此信号量中当前可用的许可证数。
int getQueueLength():返回正在等待获取许可证的线程数。
boolean hasQueuedThreads() :是否有线程正在等待获取许可证。
示例
用10个线程运行,通过信号量控制只允许3个并发的执行。
public class SemaphorePractice {
public void semaphorePrac(){
Semaphore semaphore = new Semaphore(3);
ExecutorService executorService = Executors.newFixedThreadPool(10);
for(int i = 0; i < 10; i++){
executorService.execute(new Runnable(){
public void run(){
try {
semaphore.acquire();
Thread.sleep(2000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(Thread.currentThread() + " is working");
semaphore.release();
}
});
}
executorService.shutdown();
}
public static void main(String[] args){
SemaphorePractice semaphorePractice = new SemaphorePractice();
semaphorePractice.semaphorePrac();
}
}
从运行结果中可以看到,因为中间睡眠了2秒,会每两秒打印三个结果,直到10个线程执行完,做到了并发数的控制。
应用场景
Semaphore可以用于做流量控制,控制并发数,特别公用资源有限的应用场景,比如数据库连接。
Guava中有个RateLimit类使用令牌算法也可以很好的做到流量控制,而且他们的方法也很类似,有空可以看下他的实现。