CountDownLatch是一个非常实用的多线程控制工具类。其使用场景是一些线程等待另一些线程结束之后才执行。一个简单的例子是火箭发射程序。在火箭发射之前要进行检查任务,检查任务完成之后才进行火箭发射。使用CountDownLatch可以很容易完成程序。CountDownLatch的构造函数有一个int整数,用来标示执行任务之前需要执行的countDownLatch.countDown次数。
程序示例如下:
import java.util.Random;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
public class CountDownLatchDemo{
//latck.countDown调用10次之后,主任务才能执行
static final CountDownLatch latch = new CountDownLatch(10);
static final CheckTask task = new CheckTask();
//检查任务
static class CheckTask implements Runnable{
@Override
public void run() {
try{
Thread.sleep(new Random().nextInt(10)*1000);
System.out.println(Thread.currentThread().getName()+" check complete!");
//调用countDown,计数减一
latch.countDown();
}catch(InterruptedException e){
e.printStackTrace();
}
}
}
public static void main(String[] args) throws InterruptedException {
//创建大小为10的线程池,如果线程池大小为5,则只会创建5个线程
//5个线程执行10个任务,也就是线程会共用
ExecutorService pool = Executors.newFixedThreadPool(10);
for(int i=0;i<10;i++){
pool.submit(task); //提交检查任务
}
//等吧,等检查完再执行发射任务。
latch.await();
//都检查完了,发射
System.out.println("I am fire");
pool.shutdown();
}
}