JDK源码--CountDownLatch（AQS子类）

• CountDownLatch用于同步一或多个线程，强制它们等待由其他线程执行的一组操作完成。典型用法是将一个程序分为n个互相独立的可解决任务，并创建值为n的CountDownLatch。每个任务完成时都会调用countDown，等待问题被解决的任务调用await将自己挂起，直至计数结束。

初始化

public CountDownLatch(int count) { //参数为等待完成的线程数
    if (count < 0) throw new IllegalArgumentException("count < 0");
    this.sync = new Sync(count); //将AQS状态设置为count
}

任务挂起

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

public final void acquireSharedInterruptibly(int arg)
            throws InterruptedException {
    if (Thread.interrupted())
        throw new InterruptedException();
    if (tryAcquireShared(arg) < 0)
        doAcquireSharedInterruptibly(arg);
}

private void doAcquireSharedInterruptibly(int arg)
        throws InterruptedException {
    final Node node = addWaiter(Node.SHARED);
    boolean failed = true;
    try {
        for (;;) {
            final Node p = node.predecessor();
            if (p == head) {
                int r = tryAcquireShared(arg); //状态是否已经归0
                if (r >= 0) { //状态已经归0，所有被挂起的任务可以唤醒了
                    setHeadAndPropagate(node, r); //有界阻塞队列里的等待线程全部唤醒
                    p.next = null; // help GC
                    failed = false;
                    return;
                }
            }
            if (shouldParkAfterFailedAcquire(p, node) &&
                parkAndCheckInterrupt()) //挂起线程，并返回线程是否中断
                throw new InterruptedException();
        }
    } finally {
        if (failed)
            cancelAcquire(node);
    }
}

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

private final boolean parkAndCheckInterrupt() {
    LockSupport.park(this);
    return Thread.interrupted();
}

任务完成时释放状态

public void countDown() {
    sync.releaseShared(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;
    }
}

查看等待完成的任务数

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

int getCount() {
   return getState();
}