前言
CountDownLatch是基于AQS实现的共享锁，用await和countdown方法配合完成其它线程执行完，再执行阻塞的线程的功能。所以这里一些AQS的知识就不过多的介绍，详细见我的AQS篇章http://08643.cn/p/54d372425e54
类结构
public class CountDownLatch {
private static final class Sync extends AbstractQueuedSynchronizer {
}
}
源码解析
constructor
//构造类，实际上去创建继承了AbstractQueuedSynchronizer 的Sync类
public CountDownLatch(int count) {
if (count < 0) throw new IllegalArgumentException("count < 0");
this.sync = new Sync(count);
}
- Sync
Sync(int count) {
//设置状态值，用来计数，当等于0的时候可以释放锁
setState(count);
}
- AQS的setState
protected final void setState(int newState) {
state = newState;
}
await(用来阻塞调用线程)
public void await() throws InterruptedException {
//调用的是AQS的acquireSharedInterruptibly方法
sync.acquireSharedInterruptibly(1);
}
- acquireSharedInterruptibly
这个详见我的AQS篇章，我大概说下流程
判断是否是中断标志，是的话直接抛出异常结束方法了，这样就取消了资源的获取，至于异常线程怎么处理咱们这就不用管啦。如果没有获得资源则进入阻塞状态，当被唤醒了，是被中断的状态，则直接抛出异常结束自旋方法，当然会进入取消获取节点阶段，移除自身节点修改状态，并且唤醒下一个节点
//这个详见我的AQS篇章
public final void acquireSharedInterruptibly(int arg)
throws InterruptedException {
if (Thread.interrupted())
throw new InterruptedException();
if (tryAcquireShared(arg) < 0)
doAcquireSharedInterruptibly(arg);
}
- tryAcquireShared(CountDownLatch的Sync类对其进行重写了，只是判断获取的状态计数值，是0则表示获得资源)
//在Sync类
//尝试获取资源(共享锁)，当状态是0代表获锁成功
protected int tryAcquireShared(int acquires) {
return (getState() == 0) ? 1 : -1;
}
countDown(用来释放相应的线程)
//将计数状态减一
public void countDown() {
sync.releaseShared(1);
}
- releaseShared释放资源(共享锁)
//释放资源(共享锁)
public final boolean releaseShared(int arg) {
//尝试获取共享锁,在CountDownLatch的内部类进行重写
if (tryReleaseShared(arg)) {
//释放资源，在AQS篇讲过了就不做过多解释了
doReleaseShared();
return true;
}
return false;
}
- tryReleaseShared尝试获取共享锁,在CountDownLatch的内部类进行重写
//尝试获取共享锁,在CountDownLatch的内部类进行重写
protected boolean tryReleaseShared(int releases) {
//自旋直到状态变为0，也就是构造方法初始化的计数状态值变为0后表示获得锁
for (;;) {
int c = getState();
if (c == 0)
return false;
int nextc = c-1;
if (compareAndSetState(c, nextc))
return nextc == 0;
}
}
如果在理解了AQS的原理以后，看CountDownLatch源码其实没啥东西。
AQS源码见http://08643.cn/p/54d372425e54
觉得还不错的朋友可以点点赞哦！?。。?/p>
