JUC源码系列-CountDownLatch源码研读
前言
CountDownLatch是一个很有用的工具,latch是门闩的意思,该工具是为了解决某些操作只能在一组操作全部执行完成后才能执行的情景。例如,小组早上开会,只有等所有人到齐了才能开;再如,游乐园里的过山车,一次可以坐10个人,为了节约成本,通常是等够10个人了才开。CountDown
是倒数计数,所以CountDownLatch的用法通常是设定一个大于0的值,该值即代表需要等待的总任务数,每完成一个任务后,将总任务数减一,直到最后该值为0,说明所有等待的任务都执行完了,“门闩”此时就被打开,后面的任务可以继续执行。
CountDownLatch本身是基于共享锁实现的,如果你还不了解共享锁,建议先读一下共享锁的获取与释放,然后再继续往下看。
核心属性
CountDownLatch主要是通过AQS的共享锁机制实现的,因此它的核心属性只有一个sync,它继承自AQS,同时覆写了tryAcquireShared
和tryReleaseShared
,以完成具体的实现共享锁的获取与释放的逻辑。
private final Sync sync;
/*** Synchronization control For CountDownLatch.* Uses AQS state to represent count.*/
private static final class Sync extends AbstractQueuedSynchronizer {private static final long serialVersionUID = 4982264981922014374L;Sync(int count) {setState(count);}int getCount() {return getState();}protected int tryAcquireShared(int acquires) {return (getState() == 0) ? 1 : -1;}protected boolean tryReleaseShared(int releases) {// Decrement count; signal when transition to zerofor (;;) {int c = getState();if (c == 0)return false;int nextc = c-1;if (compareAndSetState(c, nextc))return nextc == 0;}}
}
构造函数
public CountDownLatch(int count) {if (count < 0) throw new IllegalArgumentException("count < 0");this.sync = new Sync(count);
}
在构造函数中,我们就是简单传入了一个不小于0的任务数,由上面Sync的构造函数可知,这个任务数就是AQS的state的初始值。
核心方法
CountDownLatch最核心的方法只有两个,一个是countDown方法,每调用一次,就会将当前的count减一,当count值为0时,就会唤醒所有等待中的线程;另一个是await方法,它有两种形式,一种是阻塞式,一种是带超时机制的形式,该方法用于将当前等待“门闩”开启的线程挂起,直到count值为0,这一点很类似于条件队列,相当于等待的条件就是count值为0,然而其底层的实现并不是用条件队列,而是共享锁。
countDown()
public void countDown() {sync.releaseShared(1);
}
前面说过,countDown()方法的目的就是将count值减一,并且在count值为0时,唤醒所有等待的线程,它内部调用的其实是释放共享锁的操作:
public final boolean releaseShared(int arg) {if (tryReleaseShared(arg)) {doReleaseShared();return true;}return false;
}
该方法由AQS实现,但是tryReleaseShared方法由Sync类自己实现:
protected boolean tryReleaseShared(int releases) {// Decrement count; signal when transition to zerofor (;;) {int c = getState();if (c == 0)return false;int nextc = c-1;if (compareAndSetState(c, nextc))return nextc == 0;}
}
该方法的实现很简单,就是获取当前的state值,如果已经为0了,直接返回false;否则通过CAS操作将state值减一,之后返回的是nextc == 0
,由此可见,该方法只有在count值原来不为0,但是调用后变为0时,才会返回true,否则返回false,并且也可以看出,该方法在返回true之后,后面如果再次调用,还是会返回false。也就是说,调用该方法只有一种情况会返回true,那就是state值从大于0变为0值时,这时也是所有在门闩前的任务都完成了。
在tryReleaseShared
返回true以后,将调用doReleaseShared
方法唤醒所有等待中的线程,该方法我们在前面的文章中已经详细分析过了,这里就不再赘述了。
值得一提的是,我们其实并不关心releaseShared
的返回值,而只关心tryReleaseShared
的返回值,或者只关心count到0了没有,这里更像是借了共享锁的“壳”,来完成我们的目的,事实上我们完全可以自己设一个全局变量count来实现相同的效果,只不过对这个全局变量的操作也必须使用CAS。
await()
与Condition的await()方法的语义相同,该方法是阻塞式地等待,并且是响应中断的,只不过它不是在等待signal操作,而是在等待count值为0:
public void await() throws InterruptedException {sync.acquireSharedInterruptibly(1);
}
可见,await方法内部调用的是acquireSharedInterruptibly方法,相当于借用了获取共享锁的“壳”:
public final void acquireSharedInterruptibly(int arg) throws InterruptedException {if (Thread.interrupted())throw new InterruptedException();if (tryAcquireShared(arg) < 0)doAcquireSharedInterruptibly(arg);
}
我们来回忆一下独占模式下对应的方法:
public final void acquireInterruptibly(int arg) throws InterruptedException {if (Thread.interrupted())throw new InterruptedException();if (!tryAcquire(arg))doAcquireInterruptibly(arg);
}
可见,两者用的是同一个框架,只是这里:
tryAcquire(arg)
换成了tryAcquireShared(arg)
(子类实现)doAcquireInterruptibly(arg)
换成了doAcquireSharedInterruptibly(arg)
(AQS提供)
我们先来看看Sync子类对于tryAcquireShared的实现:
protected int tryAcquireShared(int acquires) {return (getState() == 0) ? 1 : -1;
}
该方法似乎有点挂羊头卖狗肉的感觉——所谓的获取共享锁,事实上并不是什么抢锁的行为,没有任何CAS操作,它就是判断当前的state值是不是0,是就返回1,不是就返回-1。
值得注意的是,在共享锁的获取与释放中我们特别提到过tryAcquireShared
返回值的含义:
- 如果该值小于0,则代表当前线程获取共享锁失败
- 如果该值大于0,则代表当前线程获取共享锁成功,并且接下来其他线程尝试获取共享锁的行为很可能成功
- 如果该值等于0,则代表当前线程获取共享锁成功,但是接下来其他线程尝试获取共享锁的行为会失败
所以,当该方法的返回值不小于0时,就说明抢锁成功,可以直接退出了,所对应的就是count值已经为0,所有等待的事件都满足了。否则,我们调用doAcquireSharedInterruptibly(arg)
将当前线程封装成Node,丢到sync queue
中去阻塞等待:
private void doAcquireSharedInterruptibly(int arg) throws InterruptedException {final Node node = addWaiter(Node.SHARED);boolean failed = true;try {for (;;) {final Node p = node.predecessor();if (p == head) {int r = tryAcquireShared(arg);if (r >= 0) {setHeadAndPropagate(node, r);p.next = null; // help GCfailed = false;return;}}if (shouldParkAfterFailedAcquire(p, node) &&parkAndCheckInterrupt())throw new InterruptedException();}} finally {if (failed)cancelAcquire(node);}
}
在前面我们介绍共享锁的获取时,已经分析过了doAcquireShared
方法,只是它是不抛出InterruptedException
的,doAcquireSharedInterruptibly(arg)
是它的可中断版本,我们可以直接对比一下:
可见,它们仅仅是在对待中断的处理方式上有所不同,其他部分都是一样的,由于doAcquireShared
在前面的文章中我们已经详细分析过了,这里就不再赘述了。
await(long timeout, TimeUnit unit)
相较于await()
方法,await(long timeout, TimeUnit unit)
提供了超时等待机制:
public boolean await(long timeout, TimeUnit unit) throws InterruptedException {return sync.tryAcquireSharedNanos(1, unit.toNanos(timeout));
}
public final boolean tryAcquireSharedNanos(int arg, long nanosTimeout) throws InterruptedException {if (Thread.interrupted())throw new InterruptedException();return tryAcquireShared(arg) >= 0 || doAcquireSharedNanos(arg, nanosTimeout);
}
private boolean doAcquireSharedNanos(int arg, long nanosTimeout) throws InterruptedException {if (nanosTimeout <= 0L)return false;final long deadline = System.nanoTime() + nanosTimeout;final Node node = addWaiter(Node.SHARED);boolean failed = true;try {for (;;) {final Node p = node.predecessor();if (p == head) {int r = tryAcquireShared(arg);if (r >= 0) {setHeadAndPropagate(node, r);p.next = null; // help GCfailed = false;return true;}}nanosTimeout = deadline - System.nanoTime();if (nanosTimeout <= 0L)return false;if (shouldParkAfterFailedAcquire(p, node) &&nanosTimeout > spinForTimeoutThreshold)LockSupport.parkNanos(this, nanosTimeout);if (Thread.interrupted())throw new InterruptedException();}} finally {if (failed)cancelAcquire(node);}
}
注意,在tryAcquireSharedNanos
方法中,我们用到了doAcquireSharedNanos
的返回值,如果该方法因为超时而退出时,则将返回false。由于await()方法是阻塞式的,也就是说没有获取到锁是不会退出的,因此它没有返回值,换句话说,如果它正常返回了,则一定是因为获取到了锁而返回; 而await(long timeout, TimeUnit unit)
由于有了超时机制,它是有返回值的,返回值为true则表示获取锁成功,为false则表示获取锁失败。doAcquireSharedNanos
的这个返回值有助于我们理解该方法究竟是因为获取到了锁而返回,还是因为超时时间到了而返回。
至于doAcquireSharedNanos
的实现细节,由于他和doAcquireSharedInterruptibly
相比只是多了一个超时机制:
代码本身很简单,就不赘述了。
实战
接下来我们来学习一个使用CountDownLatch
的实际例子,Java的官方源码已经为我们提供了一个使用的示例代码:
class Driver { // ...void main() throws InterruptedException {CountDownLatch startSignal = new CountDownLatch(1);CountDownLatch doneSignal = new CountDownLatch(N);for (int i = 0; i < N; ++i) // create and start threadsnew Thread(new Worker(startSignal, doneSignal)).start();doSomethingElse(); // don't let run yetstartSignal.countDown(); // let all threads proceeddoSomethingElse();doneSignal.await(); // wait for all to finish}
}class Worker implements Runnable {private final CountDownLatch startSignal;private final CountDownLatch doneSignal;Worker(CountDownLatch startSignal, CountDownLatch doneSignal) {this.startSignal = startSignal;this.doneSignal = doneSignal;}public void run() {try {startSignal.await();doWork();doneSignal.countDown();} catch (InterruptedException ex) {} // return;}void doWork() { ...}
}
在这个例子中,有两个“闸门”,一个是CountDownLatch startSignal = new CountDownLatch(1)
,它开启后,等待在这个“闸门”上的任务才能开始运行;另一个“闸门”是CountDownLatch doneSignal = new CountDownLatch(N)
, 它表示等待N个任务都执行完成后,才能继续往下。
Worker实现了Runnable接口,代表了要执行的任务,在它的run方法中,我们先调用了startSignal.await()
,等待startSignal
这一“闸门”开启,闸门开启后,我们就执行自己的任务,任务完成后再执行doneSignal.countDown()
,将等待的总任务数减一。
代码本身的逻辑非常简单好懂,这里不赘述了。
总结
- CountDownLatch相当于一个“门栓”,一个“闸门”,只有它开启了,代码才能继续往下执行。通常情况下,如果当前线程需要等其他线程执行完成后才能执行,我们就可以使用CountDownLatch。
- 使用
CountDownLatch#await
方法阻塞等待一个“闸门”的开启。 - 使用
CountDownLatch#countDown
方法减少闸门所等待的任务数。 - CountDownLatch基于共享锁实现。
- CountDownLatch是一次性的,“闸门”开启后,无法再重复使用,如果想重复使用,应该用[CyclicBarrier]()
JUC源码系列-CountDownLatch源码研读相关推荐
- Java集合Collection源码系列-ArrayList源码分析
Java集合系列-ArrayList源码分析 文章目录 Java集合系列-ArrayList源码分析 前言 一.为什么想去分析ArrayList源码? 二.源码分析 1.宏观上分析List 2.方法汇 ...
- MVC系列——MVC源码学习:打造自己的MVC框架(一:核心原理)(转)
阅读目录 一.MVC原理解析 1.MVC原理 二.HttpHandler 1.HttpHandler.IHttpHandler.MvcHandler的说明 2.IHttpHandler解析 3.Mvc ...
- MVC系列——MVC源码学习:打造自己的MVC框架(一:核心原理)
前言:最近一段时间在学习MVC源码,说实话,研读源码真是一个痛苦的过程,好多晦涩的语法搞得人晕晕乎乎.这两天算是理解了一小部分,这里先记录下来,也给需要的园友一个参考,奈何博主技术有限,如有理解不妥之 ...
- SpringMVC源码系列:HandlerMapping
SpringMVC源码系列:HandlerMapping SpringMVC源码系列:AbstractHandlerMapping HandlerMapping接口是用来查找Handler的.在Spr ...
- latch.await java有什么作用_java相关:CountDownLatch源码解析之await()
java相关:CountDownLatch源码解析之await() 发布于 2020-6-18| 复制链接 摘记: CountDownLatch 源码解析-- await(),具体内容如下上一篇文章说 ...
- 美工一流的个人网站源码系列(2),不漂亮你可以不下载!
美工一流的个人网站源码系列(2),不漂亮你可以不下载! 后台用户名和密码都是admin 下载地址: [url]http://down.599cn.com/599cndown/aspdown/soft2 ...
- c++ map 获取key列表_好未来Golang源码系列一:Map实现原理分析
分享老师:学而思网校 郭雨田 一.map的结构与设计原理 golang中map是一个kv对集合.底层使用hash table,用链表来解决冲突 ,出现冲突时,不是每一个key都申请一个结构通过链表串起 ...
- Spring源码系列:依赖注入(二)createBean
在Spring源码系列:依赖注入(一)(AbstractBeanFactory-getBean)最后说道getBean是依赖注入的起点,bean的创建都是通过createBean来完成具体的创建的.c ...
- 源码解读_入口开始解读Vue源码系列(二)——new Vue 的故事
作者:muwoo 转发链接:https://github.com/muwoo/blogs/blob/master/src/Vue/2.md 目录 入口开始解读Vue源码系列(一)--造物创世 入口开始 ...
最新文章
- 【收藏】Sentinel 高可用流量管理框架、服务熔断降级等
- Paper Reading × 杭州丨揭秘自动驾驶的现在与未来
- 扎克伯格公开信:Facebook拥有五大核心价值
- Python-程序模块化
- 华为交换机 查ip冲突_华为交换机:如何解决网络中IP地址发生冲突故障?
- grafana导入json文件没有数据_基于SpringBoot将Json数据导入到数据库
- layui里面layer弹窗渲染layerdate,时间控件不出现,闪现问题,F12才出来
- 计算机宏如何设置方法,excel 如何启用宏的方法,以及如何设置excel启用宏
- windows或office原版镜像下载地址及msdn.itellyou上面列的项目的详细解读
- vue利用 canvas 在图片上加文字
- java即时通讯框架_Spring Boot 开发私有即时通信系统(WebSocket)
- 力扣 345. 反转字符串中的元音字母
- SpringBoot与拦截器
- ps中图层放到顶层的快捷键是什么?
- 结盟vivo-逐浪字库上线vivo主题市场
- Qt编写自定义控件一开关按钮
- 解决ul的li横向排列换行的问题
- sakai开发环境及运行环境搭建
- 生兔子问题(递归思想)
- VMware 克隆虚拟机
热门文章
- 2022-02-25 android java file:///storage/emulated/0路径通过Uri.parse(uriString).g形式转换/storage/emulated/0
- java毕业设计夕阳红养老院系统源码+lw文档+mybatis+系统+mysql数据库+调试
- 计算机毕设(附源码)JAVA-SSM火车订票系统
- html的div插入图片,html插入图片示例(html添加图片)
- mfc快捷键设置 | vc++6.0热键设置
- 交通拥堵、安全隐患、能源浪费…数字孪生技术助力城市管理现代化
- 结构型模式 - 桥接模式Bridge
- 用计算机fx 95算平均数,CASIO 卡西欧 FX95ES PLUS 函数计算器 的平均数怎么算
- 女人的心思男人你要懂
- 中年程序员跳槽常见的6个原因!