JUC之CountDownLatch的使用

CountDownLatch：也是用来一个或多个线程线程等待一组事件发生，它包括了一个计数器，在初始化时为正数，表示需要等待的事件数量，有两个方法一个await()，另一个是countDown()

countDown：递减计数器，每次调用减去1，直到0以后不再递减阻塞事件释放

 await：等待事件，等待countDown递减为0（或线程中断或者等待超时），则继续执行后续代码

下面说一下使用，可以用来处理主线程处理和子线程处理任务时间计算，等到主子都处理完毕时间才准确，所以可以在主线程阻塞着，直到子线程完毕进行时间计算

/**
 * @Author df
 * @Date 2022/5/24 15:16
 * @Version 1.0
 * CountDownLatch使用示例：
 * 设置一个数，当这个数倒数为0时则可以继续往下走，否择通过await()阻塞
 * 通过countDown()减去数量，可以弄成线程每执行一次减一，直到数据为0，释放
 */
public class CountDownLatchDemo {

    public long timeTasks(int nThread, Runnable task) throws InterruptedException {
        final CountDownLatch startGate = new CountDownLatch(1);
        final CountDownLatch endGate = new CountDownLatch(nThread);

        for (int i = 0; i < nThread; i++) {
            Thread t = new Thread() {
                @Override
                public void run() {
                    try {
                        startGate.await();
                        try {
                            task.run();
                        } finally {
                            endGate.countDown();
                        }
                    } catch (InterruptedException e) {
                    }
                }
            };
            t.start();
        }
        long start = System.currentTimeMillis();
        startGate.countDown();
        // 等待endGate数量都为0则表示释放等待主线程继续往下走
        endGate.await();
        System.out.println("子线程全部结束可以继续往下走了");
        long end = System.currentTimeMillis();
        long result = end - start;
        return result;

    }

    public static void main(String[] args) {
        CountDownLatchDemo countDownLatchDemo = new CountDownLatchDemo();
        try {
            long time = countDownLatchDemo.timeTasks(4, new Runnable() {
                @Override
                public void run() {
                    System.out.println("子线程模拟执行任务！");
                }
            });
            System.out.println("用时：" + time);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

startGate 的CountDownLatch 设置1个，主要用来控制什么时候开始计算时间，子线程开始就阻塞，，直到主线程把startGate.countDown();递减为0，继续阻塞在 endGate.await();，直到子线程都执行完毕，在finally里endGate.countDown()逐次递减为0，继续执行endGate.await();下面的代码(也就是主线程的代码)

结果如下：