天天看点

等待多线程完成的CountDownLatch(带示例)

开始磨刀霍霍向多线程了,这期是 CountDownLatch 的一个小示例。

定义:CountDownLatch 允许一个或多个线程等待其他线程完成操作。

应用需求举例:假设有4个线程,A、B、C、D,线程 D 需要在 A、B、C 执行完之后再执行。

应用需求分析:如上描述,如果想让线程 D 最后执行,结合之前的学习,我们可以采用 join() 方法来实现,比如在 A 线程调 B.join(),B 线程调用 C.join() 等等,我们来回忆一下 join() 方法:一个线程调用另一个线程的 join() 方法时,当前线程阻塞,等待被调用 join() 方法的线程执行完毕才能继续执行,这样可以保证线程执行顺序。

如下为 join() 方法实现:

执行结果如下:

显然 join() 方法是可以完成 D 线程最后执行的要求,但是却失去了多线程并行执行的意义,怎么讲?

因为线程 A、B、C 无法完成同步运行,本质上还是串行,join() 方法其内部的实现是基于等待通知机制。

为了解决这个问题,我们可以利用 CountdownLatch 来实现这类通信方式,用法如下:

CountdownLatch 构造函数接收一个 int 类型的参数作为计数器,如果想等待 N 个点,就传入 N

调用 CountdownLatch 的 countDown() 方法时,计数器 N 就会减 1

在等待线程中调用 await() 方法,会进入等待状态,直到计数器 N 变为 0 后再执行;

CountdownLatch 方式实现:

应用场景模拟:导出用户账单明细,按年月日三个维度进行计算并显示,年月日分别放入 3 个 Excel 中,最终将Excel 合并为一个 zip 压缩文件返回给用户。

继续阅读