CountDownLatch原理
CountDownLatch是通过一个计数器来实现的,计数器的初始化值为线程的数量。每当一个线程完成了自己的任务后,计数器的值就相应得减1。当计数器到达0时,表示所有的线程都已完成任务,然后在闭锁上等待的线程就可以恢复执行任务。CountDownLatch可以起到阻塞线程,并保证线程在满足某种特定的情况后继续执行。
CountDownLatch 的两种典型用法
- 某一线程在开始运行前等待n个线程执行完毕。将 CountDownLatch 的计数器初始化为n :new CountDownLatch(n),每当一个任务线程执行完毕,就将计数器减1 countdownlatch.countDown(),当计数器的值变为0时,在CountDownLatch上 await()的线程就会被唤醒。一个典型应用场景就是启动一个服务时,主线程需要等待多个组件加载完毕,之后再继续执行。即可以通过初始化,定义线程个数。
- 实现多个线程开始执行任务的最大并行性。注意是并行性,不是并发,强调的是多个线程在某一时刻同时开始执行。类似于赛跑,将多个线程放到起点,等待发令枪响,然后同时开跑。做法是初始化一个共享的 CountDownLatch 对象,将其计数器初始化为 1 :new CountDownLatch(1),多个线程在开始执行任务前首先 coundownlatch.await(),当主线程调用 countDown() 时,计数器变为0,多个线程同时被唤醒。
源码分析
/**
* 构造器中的计数值(count)实际上就是闭锁需要等待的线程数量。这个值只能被设置一次,
*而且CountDownLatch没有提供任何机制去重新设置这个计数值。
*
* @param count 在线程可以通过 {@link #await} 之前必须调用 {@link #countDown} 的次数
* @throws IllegalArgumentException 如果给定参数小于0则抛出异常
*/
public CountDownLatch(int count) {
if (count < 0) throw new IllegalArgumentException("count < 0");
this.sync = new Sync(count);
}
复制代码
/**
* 使当前线程等待,直到计数器为零,除非当前线程被中断
* 1.当前计数为零,则此方法立即返回
* 2.如果当前计数大于零,则当前线程将因线程调度目的而被禁用并处于休眠状态,
* 直到发生如下可能:
* 由于调用了 {@link #countDown} 方法,计数达到零
* 线程等待被中断。
* @throws 如果当前线程在等待时被中断
*/
public void await() throws InterruptedException {
sync.acquireSharedInterruptibly(1);