JUC并发工具-CountDownLatch

CountDownLatch是一个同步辅助类，通过AQS实现的一个闭锁。在其他线程完成它们的操作之前，允许一个多个线程等待。简单来说，CountDownLatch中有一个锁计数，在计数到达0之前，线程会一直等待。

public class CountDownLatch {
    /**
     * Synchronization control For CountDownLatch.
     * Uses AQS state to represent count.
     */
    //继承CAS实现的Sync
    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;
        }

        //释放锁，state - 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;
            }
        }
    }

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


    //释放锁，
    public void countDown() {
        sync.releaseShared(1);
    }


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

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

CountDownLatch内部维护Sync对象，tryAcquiredShared方法就是对资源的state剩余量进行判断是否获取锁，只要state不为0，线程就进入CLH阻塞，tryReleaseShared方法实际上就是state进行 -1 操作，到 0 时，说明已经到达lantch ,这是就可以调用doRelaseShared方法唤醒线程。

CountDownLatch类的实现完全依赖AQS，只要理解AQS，理解该类也就简单了。

AQS源码阅读：JUC源码阅读-AbstractQueuedSynchronizer