CountDownLatch 学习小记

CountDownLatch 学习小记

CountDownLatch并发流程控制的工具，可以使一个或多个线程等待其他线程各自执行完毕后再执行。

CountDownLatch 定义了一个计数器，和一个阻塞队列， 当计数器的值递减为0之前，阻塞队列里面的线程处于挂起状态，当计数器递减到0时会唤醒阻塞队列所有线程，这里的计数器是一个标志，可以表示一个任务一个线程，也可以表示一个倒计时器，CountDownLatch可以解决那些一个或者多个线程在执行之前必须依赖于某些必要的前提业务先执行的场景。

CountDownLatch 是不能被重用的，如果需要重新计数，可以考虑使用CyclicBarrier 或者创建新实例。

1. 一个线程等待多个线程

public class CountDownLatchDemo {
    static CountDownLatch downLatch = new CountDownLatch(10);

    /**
     * 一个线程等待多个线程都执行完毕，在继续自己的工作
     */
    public static void main(String[] args) {
        for (int i = 0; i < 10; i++) {
            new Thread(new Runnable() {
                @Override
                public void run() {
                    try {
                        Thread.sleep(new Random().nextInt(5) * 1000);
                        System.out.println(Thread.currentThread().getName() + " 就位");
                        downLatch.countDown();
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                }
            }).start();
        }

        System.out.println("等待人员全部就位 ...");
        try {
            downLatch.await();
            System.out.println("人员全部就位 开始行动");
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

2. 一个线程等待多个线程，多个线程等待一个线程

public class CountDownLatchDemo2 {

    static CountDownLatch starterLatch = new CountDownLatch(10);
    static CountDownLatch athletesLatch = new CountDownLatch(1);

    /**
     * 一个线程等多个线程, 多个线程等一个线程
     * 模拟十个远动员百米赛跑
     * 1. 发令员等待运动员准备就绪
     * 2. 远动员准备就绪以后, 等待发令员发送开始信号
     * 3. 运动员收到开始信号开始跑步
     */
    public static void main(String[] args) {
        for (int i = 0; i < 10; i++) {
            int num = i + 1;
            new Thread(() -> {
                try {
                    Thread.sleep(RandomUtils.nextInt(1, 3) * 1000);

                    System.out.println(num + " 号远动员准备就绪等待发令员开始信号 ...");
                    starterLatch.countDown();

                    athletesLatch.await();
                    System.out.println(num + "员开始比赛 ...");
                    int time = RandomUtils.nextInt(8, 12);
                    Thread.sleep(time * 1000);
                    System.out.println(num + "号远动员用时 " + time + "s");

                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }).start();
        }

        System.out.println("等待运动员准备就绪 ... ");
        try {
            starterLatch.await();
            System.out.println("运动员就绪开始比赛 ...");
            athletesLatch.countDown();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}