执子之手与子偕老 -- CountDownLatch

一：风骚概述

当方法A包含五个模块，如果串行执行将会消耗五个方法耗时之和。如果让五个模块在多线程作用执行，主线程等待子线程执行完毕后再返回，耗时将是最长子线程耗时。等待线程执行猛然想到join()，但是这里需要各个模块并发执行，重点就在于主线程与子线程执行状态的通信。JDK中提供并发工具类CountDownLatch实现该场景，若不具备AQS知识请绕道避免浪费时间

二：线程等待

API十分简单，若想让主线程陷入等待只需要调用await()即可，该方法具备超时等待重载。初始化CountDownLatch实例设定共享锁数量，当主线程进入等待前判断共享锁数量是否等于0决定是否执行后续逻辑。注意这里是共享锁实现，意味着可以有不同线程共用一个CountDownLatch实例陷入等待

2.1 初始化共享锁

看到Sync应该马上就知道内部类继承AQS实现系列基本方法，这个count入参就会作为state锁计数器的初始数量

2.2 阻塞线程

await()方法调用AQS模板方法acquireSharedInterruptibly()，该方法中通过tryAcquire()基本方法判断锁数量是否等于0，若不等于0则调用方法doAcquireSharedInterruptibly()使得线程睡眠

看一下CountDownLatch中内部类Sync实现的基本方法tryAcquireShared()，直接获取state属性值与0比较判断

至于doAcquireSharedInterruptibly()则是调用前面ReentrantLock中讲的shouldParkAfterFailedAcquire()与parkAndCheckInterrupt()进行等待节点状态更改与线程睡眠的操作

三：线程唤醒

前面主线程已经调用await()陷入了睡眠加入了等待队列，什么时候唤醒线程？自然是子线程执行完毕逻辑后调用countDown()释放锁，当锁释放到最后一个即state属性值等于0时再从队列中唤醒等待主线程

3.1 锁释放

锁的释放首先依赖的就是Sync实现的基本方法tryReleaseShared()，原子操作让state属性值-1，并返回state属性值是否等于0即锁释放完毕

3.2 线程唤醒

doReleaseShared()主要负责当锁释放完毕唤醒睡眠主线程，唤醒方法就是unparkSuccessor()调用LockSupport.unpark()，与独占锁等实现保持一致

四：实现总结

整体来看CountDownLatch实现也是借助于AQS实现共享锁后进行线程状态通信，如果掌握AQS相关知识是比较容易看懂源码的。实例化CountDownLatch设置锁数量，主线程通过锁数量判断是否睡眠或执行，子线程通过修改state属性值与主线程通信

五：Debug源码

下图中断点可以查看到子线程修改state属性值，当最后一个子线程执行释放锁之后循环唤醒等待主线程的效果。个人认为是最佳断点位置，下面提供Debug测试输出信息代码，配合信息输出更直观追踪CountDownLatch实现

@SneakyThrows
    public static void main (String[] args) {
        CountDownLatch cdl = new CountDownLatch(5);
        for (int i = 0; i < 5; i++) {
            new Thread(new TestClass2(cdl)).start();
        }
        for (int i = 0; i < 5; i++) {
            new Thread(new TestClass(cdl)).start();
        }
        cdl.await();
        System.out.println("执行完毕");
    }
    
    @AllArgsConstructor
    static class TestClass implements Runnable{
        
        private CountDownLatch countDownLatch;
    
        @Override
        public void run () {
            countDownLatch.countDown();
            System.out.println("解开锁"+ number.getAndIncrement());
        }
    }
    
    @AllArgsConstructor
    static class TestClass2 implements Runnable{
        
        private CountDownLatch countDownLatch;
        
        @Override
        public void run () {
            System.out.println("锁了");
            try {
                countDownLatch.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            System.out.println("解开锁执行");
        }
    }