CountDownLatch源码分析及队列同步器介绍

之前有看过AbstractQueueSynchronized（队列同步器）的使用方式，这里大家可以看看我的另一篇文章《自定义同步组件构建实例及AbstractQueuedSynchronizer使用》。
下面贴出CountDownLatch的源码（源码注释被我减去了）：

public class CountDownLatch {
    
    /**
     * 这里继承队列同步器，并重写tryAcquireShared(int acquires)、tryReleaseShared(int releases)方法
     */
    private static final class Sync extends AbstractQueuedSynchronizer {
        private static final long serialVersionUID = 4982264981922014374L;

        Sync(int count) {
            setState(count);
        }

        int getCount() {
            return getState();
        }
        
        /**
         * 只有当CountDownLatch里面的计数器为0时，才会返回1
         * 在AbstractQueuedSynchronizer里面tryAcquireShared(int acquires)表示共享式获取同步状态，
         * 只有返回值大于等于0的时候表示获取成功，反之则表示获取失败
         */
        protected int tryAcquireShared(int acquires) {
            return (getState() == 0) ? 1 : -1;
        }
        
        /**
         * 在AbstractQueuedSynchronizer里面tryAcquireShared(int acquires)表示共享式释放同步状态
         * 每成功调用一次这个方法Sync的实例的status值就会减一，当status的值减为0时，则不会再减。
         */
        protected boolean tryReleaseShared(int releases) {
            // Decrement count; signal when transition to zero
            for (;;) {
                int c = getState();
                if (c == 0)
                    return false;
                int nextc = c-1;
                if (compareAndSetState(c, nextc))
                    return nextc == 0;
            }
        }
    }

    private final Sync sync;

    public CountDownLatch(int count) {
        if (count < 0) throw new IllegalArgumentException("count < 0");
        this.sync = new Sync(count);
    }

    public void await() throws InterruptedException {
        sync.acquireSharedInterruptibly(1);
    }

    public boolean await(long timeout, TimeUnit unit)
        throws InterruptedException {
        return sync.tryAcquireSharedNanos(1, unit.toNanos(timeout));
    }

    /**
     * countDown:每调用一次countDown()方法都会使sync成员变量status减一（直到status为0），status为0，则会让调用await()方法的地方不在阻塞，
     * 从而达到可以等待多个并发事件完成的目标
     * @author JM 
     * 2017-2-23 下午10:29:25   
     * void
     */
    public void countDown() {
        sync.releaseShared(1);
    }

    public long getCount() {
        return sync.getCount();
    }

    public String toString() {
        return super.toString() + "[Count = " + sync.getCount() + "]";
    }
}

看完上面的源码，其实说来说去CountDownLatch就是一个自定义的同步组件。

现在来分析一下CountDownLatch实现“等待多个并发事件完成”的过程：
在new CountDownLatch(int count)的时候就会将CountDownLatch内部的Sync初始化，并将sync的status值初始化为count，每次调用CountDownLatch对象的countDown()方法都会将sync的status的值减一，当status值减为0时，线程内因为调用await()而阻塞的地方将会重新执行。从而达到“等待多个并发事件完成”的效果