CountDownLatch分析

1.类声明

算是最简单最简单的一个同步工具类了。。。哈哈哈

// 没有继承类，没有实现接口
public class CountDownLatch {
}

2.核心执行依旧靠Sync内部类（基本上所有工具类，包括Lock都是这个模式实现）

private static final class Sync extends AbstractQueuedSynchronizer {
        private static final long serialVersionUID = 4982264981922014374L;

        Sync(int count) {
            // 设置AQS的资源数量
            setState(count);
        }

        int getCount() {
            return getState();
        }

        // 非常重要的实现，只有当当前资源数量为0的时候才算是获取到锁
        protected int tryAcquireShared(int acquires) {
            return (getState() == 0) ? 1 : -1;
        }
        // 以共享模式释放一个资源， 哈哈这里很好玩，int releases是AQS生命的函数带有的参数，这里根本没用它，因为每次Countdown本来就只减一，但是感觉怪怪的。
        protected boolean tryReleaseShared(int releases) {
            // Decrement count; signal when transition to zero
            for (;;) { // 比较简单的CAS递减操作。
                int c = getState();
                if (c == 0)  // 已经是0了，直接返回false，AQS什么也不干
                    return false; 
                int nextc = c-1;
                if (compareAndSetState(c, nextc))
                    return nextc == 0; // 操作完成后资源数为0则返回true，不为0返回false，AQS什么也不干
            }
        }
    }

3.构造函数

public CountDownLatch(int count) {
        // 可以看到初始值不能小于0
        if (count < 0) throw new IllegalArgumentException("count < 0");
        // 实例化一个Sync
        this.sync = new Sync(count);
    }

4.await方法

public void await() throws InterruptedException {
        // 调用了AQS的尝试获取共享锁可中断的方法，因为
        // Sync在实现tryAcquireShared时，是以getState是否为0来判断获取到锁的，所以
        // 只要当前state还不为0，await县城将会被放进等待队列。
        sync.acquireSharedInterruptibly(1);
    }
// 调用了支持超时的获取锁方法
public boolean await(long timeout, TimeUnit unit)
        throws InterruptedException {
        return sync.tryAcquireSharedNanos(1, unit.toNanos(timeout));
    }

AQS中acquireSharedInterruptibly方法

public final void acquireSharedInterruptibly(int arg)
            throws InterruptedException {
        if (Thread.interrupted())
            throw new InterruptedException();
        if (tryAcquireShared(arg) < 0) // (getState() == 0) ? 1 : -1；小于0代表state还大于0
            doAcquireSharedInterruptibly(arg);
    }

5.countDown方法

public void countDown() {
       // 调用AQS的释放共享资源方法
       sync.releaseShared(1);
   }

AQS的releaseShared源码

public final boolean releaseShared(int arg) {
// 当前扣减后剩余量为0时，tryReleaseShared返回true，将会执行doReleaseShared唤醒await线程
// 就是说新建闭锁时初始化资源数为5，但是程序中有6个线程去做countDown，那么最后一个countDown的线程将不会造成任何影响
        if (tryReleaseShared(arg)) { 
            doReleaseShared();
            return true;
        }
        return false;
    }