Java多线程——Semaphore

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类使用令牌算法也可以很好的做到流量控制，而且他们的方法也很类似，有空可以看下他的实现。

参考资料

并发工具类（三）控制并发线程数的Semaphore
JAVA多线程–信号量(Semaphore)