1.8CountDownLatch史上最详细源码解析

前言

CountDownLatch是基于AQS实现的共享锁，用await和countdown方法配合完成其它线程执行完，再执行阻塞的线程的功能。所以这里一些AQS的知识就不过多的介绍，详细见我的AQS篇章https://www.jianshu.com/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源码见https://www.jianshu.com/p/54d372425e54
觉得还不错的朋友可以点点赞哦！！！！