二十七. 并发实战-CountDownLatch的使用

布满荆棘的人生 2024-03-27 08:49 58阅读 0赞

### 前言 ###

关于`CountDownLatch`，官方注释中给出的说明如下。

> **一种同步辅助工具，允许一个或多个线程等待，直到在其它线程中执行的一组操作完成**

本篇文章，将对`CountDownLatch`的使用进行说明。

### 正文 ###

演示示例如下所示。

public class CountDownLatchTest {
        
    
        private long[] nums;
        private int sum1 = 0;
        private int sum2 = 0;
    
        @Before
        public void setUp() throws Exception {
        
            nums = new long[200];
            for (int i = 0; i < 200; i++) {
        
                nums[i] = i;
            }
        }
    
        @Test
        public void 主线程等待子线程执行完任务后再继续往下执行() throws Exception {
        
            // 创建线程池
            ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(2, 4,
                    60, TimeUnit.SECONDS, new ArrayBlockingQueue<>(300));
    
            // 创建CountDownLatch对象，count值为2
            CountDownLatch countDownLatch = new CountDownLatch(2);
    
            // 任务1计算nums数组的0-99索引的数字和
            Runnable fistRunnable = new Runnable() {
        
                int count = 0;
    
                @Override
                public void run() {
        
                    for (int i = 0; i < 100; i++) {
        
                        count += nums[i];
                    }
                    sum1 = count;
                    countDownLatch.countDown();
                }
            };
            // 任务2计算nums数组的100-199索引的数字和
            Runnable secondRunnable = new Runnable() {
        
                int count = 0;
    
                @Override
                public void run() {
        
                    for (int i = 100; i < 199; i++) {
        
                        count += nums[i];
                    }
                    sum2 = count;
                    countDownLatch.countDown();
                }
            };
    
            // 线程池执行任务
            threadPoolExecutor.execute(fistRunnable);
            threadPoolExecutor.execute(secondRunnable);
    
            // 等待任务1和任务2执行完毕
            countDownLatch.await();
    
            // 打印结果
            System.out.println("第一个任务的执行结果是：" + sum1);
            System.out.println("第二个任务的执行结果是：" + sum2);
        }
    
    }

主要的功能就是给定一个200长度的数组**nums**，然后让线程1计算数组索引0-99的累加和**sum1**，线程2计算数组索引100-199的累加和**sum2**，主线程中等待线程1和线程2计算完毕后，打印**sum1**和**sum2**的值。

运行测试程序，打印结果如下。

![在这里插入图片描述][0247e4f0e5f349b9b8987f8affc31413.jpeg_pic_center]

### 总结 ###

`CountDownLatch`的使用方式总结如下。

1.  创建`CountDownLatch`对象，创建时指定**count**数量，**count**数量表明`CountDownLatch`的`countDown()`方法被调用多少次，才唤醒在`CountDownLatch`的`await()`方法上阻塞的线程，通常，**count**值需要设置为大于0的整数值（设置为0也可以，但是没意义）；
2.  需要等待其它线程执行完任务的线程需要调用`CountDownLatch`的`await()`方法，以实现阻塞等待；
3.  执行任务的线程执行完任务后，需要调用`CountDownLatch`的`countDown()`方法来减少**count**值；
4.  当**count**值减少到0时，唤醒所有在`CountDownLatch`的`await()`方法上等待的线程。

[0247e4f0e5f349b9b8987f8affc31413.jpeg_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/27/a86c1e98fe0d4aa0b7d2c6407009f664.png