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上了(链接见本文末)，这里先给出一个所涉及内容图：

---

声明：本文是学习笔记，主要学习自以下视频。

---

^_^ 如有不当之处，欢迎指正

^_^ 学习视频
             《Java多线程与并发实战视频课程》，齐毅 

^_^ 多线程一栏所有测试示例代码，托管链接
               https://github.com/JustryDeng/PublicRepository

^_^ 本文已经被收录进《程序员成长笔记(三)》，笔者JustryDeng