CountDownLatch 是一个通用同步工具,它有很多用途,将计数 1 初始化的 CountDownLatch 用作一个简单的开/关锁存器,或阀门:在通过调用 countDown() 的线程打开入口前,所有调用 await 的线程都一直在入口处等待。用 N 初始化的 CountDownLatch 可以使一个线程在 N 个线程完成某项操作之前一直等待。
CountDownLatch 的一个有用特性是,它不要求调用 countDown 方法的线程等到计数到达零时才继续,而在所有线程都能通过之前,它只是阻止任何线程继续通过一个 await。
我们通过两段代码比较来看出CountDownLatch的作用,首先我们先看看一个多线程简单的代码:
public class Driver {
public static void main(String[] args) throws InterruptedException {
int N = 50;
for (int i = 0; i < N; i++) {//创建并开启线程
new Thread(new Worker()).start();
}
execute(N);
execute1(N);
}
public static void execute(int N){
System.out.println("execute:我运行了"+N);
}
public static void execute1(int N){
System.out.println("execute1:我运行了=============="+N);
}
}
class Worker implements Runnable{
Worker(){
}
@Override
public void run() {
execute2();
execute3();
}
public void execute2(){
System.out.println("execute2:我运行了..............");
}
public void execute3(){
System.out.println("execute3:我运行了------------------");
}
}
从以上代码当中我们可以看到控制台打印以下执行结果:
我们不难发现在execute、execute1方法穿插在多线程中间执行,没有任何规律可言。
我们接着看第二段代码演示:
import java.util.concurrent.CountDownLatch;
public class Driver {
public static void main(String[] args) throws InterruptedException {
int N = 50;
CountDownLatch startSignal = new CountDownLatch(1);
CountDownLatch doneSignal = new CountDownLatch(N);
for (int i = 0; i < N; i++) {//创建并开启线程
new Thread(new Worker(startSignal,doneSignal)).start();
}
execute(N);
startSignal.countDown();//使当前线程在锁存器倒计数至零之前一直等待,除非线程被中断
execute1(N);
doneSignal.await();//递减锁存器的计数,如果计数到达零,则释放所有等待的线程
}
public static void execute(int N){
System.out.println("execute:我运行了"+N);
}
public static void execute1(int N){
System.out.println("execute1:我运行了=============="+N);
}
}
class Worker implements Runnable{
private final CountDownLatch startSignal;
private final CountDownLatch doneSignal;
Worker(CountDownLatch startSignal,CountDownLatch doneSignal){
this.startSignal = startSignal;
this.doneSignal = doneSignal;
}
@Override
public void run() {
try {
startSignal.await();
execute2();
doneSignal.countDown();
} catch (InterruptedException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
execute3();
}
public void execute2(){
System.out.println("execute2:我运行了..............");
}
public void execute3(){
System.out.println("execute3:我运行了------------------");
}
}
控制台打印效果如下:
我们可以从控制台看到,startSignal.await()一直阻碍着run()方法中的线程运行,直到startSignal.countDown()等于0后,阀门被打开,run()方法中的线程才开始运行,直到把运行doneSignal.countDown()为0之后(也就是50次以后),doneSignal.await()方法才释放了主线程。