#问题的由来
主线程等待所有线程都运行完后,再执行逻辑,这个需求很普遍。比如,在处理数据时,为了效率期间,起了10个线程,分别处理一块数据,这样能缩短处理时间,10个线程都执行完后,继续进行下边的逻辑(有点map/reduce的意思)。这里我们就需要用到java的线程同步类 CountDownLatch
,count down是计数的意思,latch是门闩的意思,合起来也就是计数的开关。我们看一下CountDownLatch的API。
#CountDownLatch API介绍
一个同步辅助类,在完成一组正在其他线程中执行的操作之前,它允许一个或多个线程一直等待。
用给定的计数 初始化 CountDownLatch。由于调用了 countDown() 方法,所以在当前计数到达零之前,await
方法会一直受阻塞。之后,会释放所有等待的线程,await
的所有后续调用都将立即返回。这种现象只出现一次——计数无法被重置。如果需要重置计数,请考虑使用 CyclicBarrier。CountDownLatch 是一个通用同步工具,它有很多用途。将计数 1 初始化的 CountDownLatch
用作一个简单的开/关锁存器,或入口:在通过调用 countDown() 的线程打开入口前,所有调用 await 的线程都一直在入口处等待。用
N 初始化的 CountDownLatch 可以使一个线程在 N 个线程完成某项操作之前一直等待,或者使其在某项操作完成 N
次之前一直等待。CountDownLatch 的一个有用特性是,它不要求调用 countDown
方法的线程等到计数到达零时才继续,而在所有线程都能通过之前,它只是阻止任何线程继续通过一个 await。
从中可以看出,最重要的几个步骤,
第一,创建门闩,
第二,开启门闩,阻塞所有线程,
第三,其它线程递减次数,减为0时,门闩打开,主线程继续执行放开门闩
#样例
public class TestCountDownLatch {
public static void main(String[] args) {
int count = 5;
final CountDownLatch latch = new CountDownLatch(count);//创建门闩
//execute thread
for(int i=0; i<count; i++){
new Thread(new Worker(latch)).start();
}
//
System.out.println("进入等待");
try {
latch.await();//开启门闩,阻塞所有线程,
} catch (InterruptedException e) {
e.printStackTrace();
}
//dosomething
System.out.println("都执行完了");
}
}
class Worker implements Runnable{
private final CountDownLatch latch;
public Worker(CountDownLatch latch) {
super();
this.latch = latch;
}
@Override
public void run() {
//do something
System.out.println(Thread.currentThread().getName());
try {
Thread.sleep(5000);
} catch (InterruptedException e) {
e.printStackTrace();
}
/*
其它线程递减次数,减为0时,门闩打开,主线程继续执行放开门闩。
每调用一次countdown()计数就减一次
*/
latch.countDown();//递减
}
}
输出结果
#后续
这里,我们没有往线程中传值、也没有获取线程的计算结果,所以比较简单。如果需要往线程中传值,使用构造函数方法传参或者公共方法传参;获取其它线程的计算结果,可以使用Future