JUC之CountDownLatch倒计时锁
CountDownLatch倒计时锁特别适合处理“总-分任务”，例如:多线程计算后的数据汇总、多线程导入多个sheet然后数据汇总等。
CountDownLatch类位于java.util.concurrent包下(即:JUC包下),利用它可以实现类似计数器的功能。比如:任务A,他要等待其他几个任务执行完毕之后才能执行,此时就可以利用CountDownLatch倒计时锁来实现这个功能。
执行原理(白话比喻说明):
假设现在在main线程上开了5个线程,main线程对这5个线程分配任务说你们把自己所负责的田里秧子插完,每人就可以来我这里领取一百块钱(main线程提前准备了五百元钱在包里)。当main线程的“包里”准备的五百元钱被领取完的时候,那么main线程就知道他们都干完分配给他们的活了;这时main线程有什么后续的活就可以开始干了。
代码示例:
import java.util.Random;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
/**
* CountDownLatch倒计时锁 使用示例
*
* @author JustryDeng
* @date 2018/10/15 23:08
*/
public class CountDownLatchDemo {
/** 创建CountDownLatch实例,其大小一般与线程任务数量一致 */
private static CountDownLatch countDownLatch = new CountDownLatch(5);
public static void main(String[] args) throws InterruptedException {
// 创建定长线程池
ExecutorService executorService = Executors.newFixedThreadPool(10);
for (int i = 1; i <= 5; i++) {
// 使用lambel表达式简单实现Runnable接口的run方法
executorService.execute(() -> {
int consumeTime = new Random().nextInt(10) + 1;
String name = Thread.currentThread().getName();
System.out.println(name + ":我要" + consumeTime + "秒才能完成任务!");
try {
Thread.sleep(consumeTime * 1000);
System.out.println(name + "完成任务啦!");
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
// 倒计时减一
// 注.countDown()方法是线程安全的
countDownLatch.countDown();
}
});
}
// 当countDownLatch的值不为0时,此线程阻塞，继续等待;直到countDownLatch的值=0时,此线程才往下执行
// 这里是.await()方法,而不是.wait()方法
countDownLatch.await();
// 关闭线程池
executorService.shutdown();
System.out.println(Thread.currentThread().getName() + ":终于该本线程出手了!");
}
}
注:还有一个.await(long timeout, TimeUnit unit)方法,可以设置等待长时间,如果在指定时间内倒计时锁还没有
被减为0，那么就原来阻塞的当前线程就不“等”了，继续执行后面的逻辑。
追注:更多细节推荐阅读源码。
运行主函数,(某次的)输出结果为:
笔者将本人多线程一栏中博客涉及到的所有代码示例(Lock分开放在一个专门的项目、synchronized的代码附在该文章末尾)，放在GIT上了(链接见本文末)，这里先给出一个所涉及内容图：