是什么
Semaphore是向外分发资源的许可证,可以允许一个或多个任务同时访问资源。
Semaphore通过构造参数来指定许可证的数量;acquire方法阻塞式获取许可证;release方法释放许可证。
可以将其比喻为地铁的安检,每当人流量高峰的时候,安检会先让几个人进去,然后拦住后面的人,待前面几人通过安检门后,会对后面的人用相同的方式放行。
应用场景
- 限流:比如大批量请求有并发限制的接口。
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(); } }
- 轮询:在生产者-消费者模式中,消费者可以通过信号量方式判断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 并发编程的艺术》