CountDownLatch是一个同步辅助类,在一组其他线程完成操作之前,它允许一个或多个线程等待。
1.Sync内部类
这个内部类继承了AQS,实现了AQS的共享模式下的几个模板方法
private static final class Sync extends AbstractQueuedSynchronizer {
private static final long serialVersionUID = 4982264981922014374L;
//初始化Sync,设置了state为等待需要的线程数量
Sync(int count) {
setState(count);
}
//返回当前state,也就是还需要等几个线程完成操作
int getCount() {
return getState();
}
//共享模式的tryAcquire,如果state为0,表明获取成功,否则获取失败
protected int tryAcquireShared(int acquires) {
return (getState() == 0) ? 1 : -1;
}
//共享模式的释放,如果state为0,返会false。否则就将state减一。并返回减完之后,是否是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;
}
}
}
await(),调用AQS的doAcquireSharedInterruptibly()
//调用AQS的doAcquireSharedInterruptibly。
//doAcquireSharedInterruptibly中调用子类的tryAcquireShared,就是state为0就表示获取成功
//获取成功之后就返回,调用await()的地方就可以继续执行
public void await() throws InterruptedException {
sync.acquireSharedInterruptibly(1);
}
countDown(),调用子类的tryReleaseShared(),然后如果成功就通知队列中的节点可以unpark检查state是不是为0了,如果state还不是0,就继续park。
public void countDown() {
sync.releaseShared(1);
}