CountDownLatch与AQS

1 CountDownLatch初始化

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

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

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

        int getCount() {
            return getState();
        }

        protected int tryAcquireShared(int acquires) {
            return (getState() == 0) ? 1 : -1;
        }

        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;
            }
        }
    }

2 countDown()

    public void countDown() {
        sync.releaseShared(1);
    }

AQS#releaseShared(int arg)

    public final boolean releaseShared(int arg) {
    	// 对计数器减1操作
        if (tryReleaseShared(arg)) {
        	// 如果计数器为0，唤醒被await方法阻塞对线程
            doReleaseShared();
            return true;
        }
        return false;
    }

第一节中Sync#tryReleaseShared(args)是CAS操作减1

3 await()

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

AQS#acquireInterruptibly(int arg)

    public final void acquireInterruptibly(int arg)
            throws InterruptedException {
        if (Thread.interrupted())
            throw new InterruptedException();
        // 判断state是否为0，不为0则阻塞当前线程
        if (!tryAcquire(arg))
            doAcquireInterruptibly(arg);
    }

第一节中Sync#tryAcquire(args)是判断state是否为0