【Java并发编程】CountDownLatch

妖狐艹你老母 2024-03-26 17:45 87阅读 0赞

CountDownLatch是JUC提供的解决方案  
CountDownLatch 可以保证一组子线程全部执行完牛后再进行主线程的执行操作。例如，主线程启动前，可能需要启动并执行若干子线程，这时就可以通过 CountDownLatch 来进行控制。  
CountDownLatch是通过一个线程个数的计数器实现的同步处理操作，在初始化时可以为CountDownLatch设置一个线程执行总数，这样每当一个子线程执行完毕后都要执行减1操作，当所有的子线程都执行完毕后，CountDownLatch中保存的计数为0，则主线程恢复执行。

![CountDownLatch 操作流程][CountDownLatch]

CountDownLatch类常用方法

<table> 
 <thead> 
  <tr> 
   <th align="left">方法</th> 
   <th align="left">描述</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td align="left">public CountDownLatch(int count)</td> 
   <td align="left">定义等待子线程总数</td> 
  </tr> 
  <tr> 
   <td align="left">public void await() throws InterruptedException</td> 
   <td align="left">主线程阻塞，等待子线程执行完毕</td> 
  </tr> 
  <tr> 
   <td align="left">public void countDown()</td> 
   <td align="left">子线程执行完后减少等待数量</td> 
  </tr> 
  <tr> 
   <td align="left">public long getCount()</td> 
   <td align="left">获取当前等待数量</td> 
  </tr> 
 </tbody> 
</table>

实现代码：

public static void main(String[] args) throws Exception {
        
        int n = 3;
        String[] tasks = {
        "发短信完毕", "发微信完毕", "发QQ完毕"};
        int[] executeTimes = new int[]{
        2, 5, 1};
        CountDownLatch countDownLatch = new CountDownLatch(n);
        ExecutorService executorService = Executors.newFixedThreadPool(n);
    
        long start = System.currentTimeMillis();
        for (int i = 0; i < n; i++) {
        
            int finalI = i;
            executorService.submit(() -> {
        
                try {
        
                    TimeUnit.SECONDS.sleep(executeTimes[finalI]);
                    System.out.println(tasks[finalI]);
                } catch (InterruptedException e) {
        
                    e.printStackTrace();
                } finally {
        
                    countDownLatch.countDown();
                }
            });
        }
        countDownLatch.await();
        System.out.println("所有消息都发送完毕了，执行主线程任务。\n耗时ms：" + (System.currentTimeMillis() - start));
        // 不要忘记关闭线程池，不然会导致主线程阻塞无法退出
        executorService.shutdown();
    }

本程序利用 CountDownLatch 定义了要等待的子线程数量，这样在该统计数量不为0的时候，主线代码暂时挂起，直到所有的子线程执行完毕（调用countDown()方法）后主线程恢复执行。

[CountDownLatch]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/26/2b4d8349664445e5ab79f9758c3abd3c.png