Semaphore 简介

是什么

Semaphore是向外分发资源的许可证，可以允许一个或多个任务同时访问资源。

Semaphore通过构造参数来指定许可证的数量；acquire方法阻塞式获取许可证；release方法释放许可证。

可以将其比喻为地铁的安检，每当人流量高峰的时候，安检会先让几个人进去，然后拦住后面的人，待前面几人通过安检门后，会对后面的人用相同的方式放行。

应用场景

1. 限流：比如大批量请求有并发限制的接口。

public class SemaphoreDemo {

    private static final int THREAD_COUNT = 30;

    private static ExecutorService service = Executors.newFixedThreadPool(THREAD_COUNT);

    private static Semaphore semaphore = new Semaphore(2);

    public static void main(String[] args) {
        for(int i = 0;i < THREAD_COUNT;i++){
            service.execute(new Runnable() {
                public void run() {
                    try {
                        semaphore.acquire();
                        dowork();
                        TimeUnit.SECONDS.sleep(3);
                        semaphore.release();
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                }
            });
        }
        service.shutdown();
    }
}

2. 轮询：在生产者-消费者模式中，消费者可以通过信号量方式判断queue是否为空。

消费者

@Override
	public void run() {
		this.isStop = false;
		while (!isStop) {
			try {
				Semaphore semaphore = doWork();
				if (semaphore != null) {
					int acquireCount = semaphore.availablePermits();
					acquireCount = acquireCount == 0 ? 1 : acquireCount;
					semaphore.tryAcquire(acquireCount, 1, TimeUnit.SECONDS);
				}
			} catch (Throwable e) {
			}
		}
	}

生产者只需要在添加队列时调用release方法。 消费端的代码相当于判断queue是否为空，为空等待1s后执行dowork，不为空直接执行dowork。

参考：

《Java 编程思想》
《Java 并发编程的艺术》