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

前言

关于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的值。

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

总结

CountDownLatch的使用方式总结如下。

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