今天讲一个倒计时器工具,叫CountDownLatch。需要这个工具的场景大概有:当所有的小任务都完成之后,再启动大任务。
先看代码:
public class CountDownLatchDemo {
static final CountDownLatch LATCH = new CountDownLatch(10);
public static class Task implements Runnable {
final static AtomicInteger incr = new AtomicInteger();
@Override
public void run() {
try {
Thread.sleep(new Random().nextInt(10)*100);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("火箭检测任务-" + incr.getAndIncrement() + " 完成");
LATCH.countDown();
}
}
public static void main(String[] args) throws InterruptedException {
ExecutorService executorService = Executors.newFixedThreadPool(10);
for (int i = 0; i < 10; i++) {
executorService.execute(new Task());
}
LATCH.await();
System.out.println("发射火箭");
executorService.shutdown();
}
}
这里模拟了火箭发射的场景,所有预备任务完成,才能发射火箭。
CountDownLatch比较简单,一个构造加上两个主要方法:
- 构造函数需要入参一个总数,表示有几个小任务需要执行
- countDown() 每次执行完小任务都要调用这个方法,用来倒计时
- await() 没有倒计时到0,就会一直阻塞。当倒计时完成,则继续执行后续代码