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