这次分析CountDownLatch,相信大部分人都用过把!

CountDownLatch内部还是Sync对象,还是基础AQS(可见其重要性),首先看一下CountDownLatch初始化,CountDownLatch初始化一个state状态值,

这个值就是AQS中获取锁得state值,CountDownLatch首先是基于共享锁实现得,实现了AQS中得 tryAcquireShare()方法跟tryReleaseShared()方法。

java源码-CountDownLatch源码分析-LMLPHP

java源码-CountDownLatch源码分析-LMLPHP

首先看一下countDown()方法,调用AQS得releaseShared(1)方法,首先尝试获获取锁,如果state等于0,返回false。不需要唤醒等待队列,

如果更新state-1成功判断是否==0,如果等于0开始唤醒等待队列。这个方法比较简单。

java源码-CountDownLatch源码分析-LMLPHP

调用唤醒方法,先获取head节点,然后判断是否只有一个节点,或者head节点是否为空,如果都不是,然后获取等待状态,

判断是否是signal如果是先更新为0,然后唤醒next节点,如果此时只有一个head节点时候waitStatus=0,然后设置为PROPAGATE,等待下一个阻塞节点加入

然后修改为signal状态。如果head节点被唤醒的节点修改的时候,此循环会继续唤醒 header的next节点。这块的设计目前看还是有点模糊,这块有很多状态,

就是唤醒的线程获取到锁更新header节点的时候,还需要好好琢磨。

java源码-CountDownLatch源码分析-LMLPHP

在看一下await()方法.调用AQS中得tryAcquireShareNanos()或者acquireShareinterruptibly()方法,首先会tryAcqureShared()获取锁,

如果此时锁state==0返回-1此时进去doAcquireShareinterruptibly()阻塞,

java源码-CountDownLatch源码分析-LMLPHP

添加一个share的Node节点到节点尾,然后获取其前驱节点,如果前驱节点是head节点,然后再次获取锁,如果获取到则设置header节点为

当前node节点,如果不是head节点,则挂起当前节点。

java源码-CountDownLatch源码分析-LMLPHP

setHeadAndpropagate()方法设置node节点为header头,并唤醒next节点,使节点唤醒传播下去。

java源码-CountDownLatch源码分析-LMLPHP

05-22 23:09