一、什么是 CountDownLatch
CountDownLatch 定义了一个计数器,和一个阻塞队列, 当计数器的值递减为 0 之前,阻塞队列里面的线程处于挂起状态,当计数器递减到 0 时会唤醒阻塞队列所有线程,这里的计数器是一个标志,可以表示一个任务一个线程,也可以表示一个倒计时器,CountDownLatch 可以解决那些一个或者多个线程在执行之前必须依赖于某些必要的前提业务先执行的场景,例如主线程要等待所有子线程执行完再执行,老师要等待所有考生都交卷再离开考场……
二、CountDownLatch 提供的API
方法 | 说明 |
countDown() | 使latch的值减1,如果减到了0,则会唤醒所有等待在这个latch上的线程。 |
await() | 使当前线程进入同步队列进行等待,直到latch的值被减到0或者当前线程被中断,当前线程就会被唤醒。 |
await(long timeout, TimeUnit unit) | 带超时时间的await() |
getCount() | 获得latch的数值 |
三、代码举例
public class TestCountDownLatchDemo {
public volatile static int i = 0;
public static void main(String[] args) {
final CountDownLatch latch = new CountDownLatch(2);
new Thread(() ->{
try {
System.out.println("正在执行");
Thread.sleep(1000);
i++;
System.out.println("执行完毕"+ i);
latch.countDown();
} catch (InterruptedException e) {
e.printStackTrace();
}
}).start();
new Thread(() ->{
try {
System.out.println("正在执行");
Thread.sleep(3000);
i ++;
System.out.println("执行完毕" + i);
latch.countDown();
} catch (InterruptedException e) {
e.printStackTrace();
}
}).start();
try {
System.out.println("等待两个子线程执行完毕……");
latch.await();
System.out.println("2个子线程已经执行完成");
System.out.println("继续执行主线程……");
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
结果打印:
正在执行
等待两个子线程执行完毕……
正在执行
执行完毕1
执行完毕2
2个子线程已经执行完成
继续执行主线程……