AQS系列之AbstractQueuedSynchronizer基础分析
AQS系列之以排斥锁分析
AQS系列之共享锁解析
AQS系列之共享锁的应用解析:Semaphore
AQS系列之共享锁的应用解析:CountDownLatch
一、简介
CountDownLatch即倒排计数,是AQS共享锁的另一个经典应用。其应用主要是一个(或多个)线程等待一系列线程完成某些操作后才继续向下执行的场景。
换种程序上的描述:A线程申请资源await,进行阻塞等待,一系列线程进行某些操作(共state个),每完成一个释放一次资源coutDown。所有操作完成后,A线程资源获取成功,继续向下执行。
CountDown同样依赖内部类Sync继承AQS实现线程管理,只不过其申请资源的场景只是用于判断一系列线程的操作有没有完成,即使同时有多个线程申请资源,也不会存在互相的竞争,因此没有必要区分是否公平的策略。
CountDown只有一个构造方法public CountDownLatch(int count),用于设置AQS的state,这儿一般称呼为count,即任务数。
事实上,理解AQS的实现后,再阅读CountDownLatch的代码逻辑相当简单,可以算是AQS最简单的应用了。
二、获取资源
protected int tryAcquireShared(int acquires) {
return (getState() == 0) ? 1 : -1;
}
可以看到很简单,判断count是否为0,即任务是否已经全部完成。
这儿只有1和-1,没有0.因为成功的时候其他线程申请资源也是成功的,符合tryAcquireShared对返回值的定义。
三、释放资源
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;
}
}
释放资源的逻辑也是很明显,对count进行减1。