JUC小册——CountDownLatch原理以及用法。

目录

一、CountDownLatch简介

二、CountDownLatch使用场景

1. 并行任务的协调

2. 完成初始化

3. 控制线程的执行顺序

三、CountDownLatch用法

---

一、CountDownLatch简介

CountDownLatch  是 Java 并发包中的一个同步工具类，位于  java.util.concurrent  包中。允许一个或多个线程等待直到一组操作完成。 CountDownLatch  的核心思想是通过一个计数器来控制线程的执行。 CountDownLatch  维护一个计数器，初始值为一个正整数。当计数器的值减为零时，表示所有线程的操作已完成，等待的线程将被唤醒。

CountDownLatch  的主要方法包括：

 await() ： 使当前线程等待，直到计数器的值为零。

 countDown()：将计数器的值减一。当计数器的值减为零时，所有等待的线程将被唤醒。

二、CountDownLatch使用场景

1. 并行任务的协调

当有多个线程执行任务，并希望在所有线程完成后再继续执行主线程时，可以使用  CountDownLatch 。

2. 完成初始化

在某些情况下，主线程需要等待多个初始化任务完成后再继续执行，比如在启动一个服务时等待所有依赖的组件初始化完毕。

3. 控制线程的执行顺序

可以使用  CountDownLatch  来控制线程的执行顺序。

三、CountDownLatch用法

下面使用CountDownLatch实现如何在多个线程完成任务后再继续执行主线程：

import java.util.concurrent.CountDownLatch;

public class CountDownLatchExample {

    private static final int THREAD_COUNT = 3;

    public static void main(String[] args) {
        CountDownLatch latch = new CountDownLatch(THREAD_COUNT);

        // 创建并启动多个线程
        for (int i = 0; i < THREAD_COUNT; i++) {
            final int threadIndex = i;
            new Thread(() -> {
                try {
                    // 模拟任务执行
                    System.out.println("线程 " + threadIndex + " 正在工作...");
                    Thread.sleep((long) (Math.random() * 1000)); // 随机睡眠
                    System.out.println("线程 " + threadIndex + " 已经完成.");
                } catch (InterruptedException e) {
                    e.printStackTrace();
                } finally {
                    // 任务完成，计数器减一
                    latch.countDown();
                }
            }).start();
        }

        try {
            // 主线程等待，直到计数器减为零
            latch.await();
            System.out.println("所有线程都完成，主线程开始启动.");
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

结果如下：

如上，在循环中创建多个线程，每个线程模拟一个任务的执行，每个线程在完成任务后调用  latch.countDown() ，将计数器减一，主线程调用  latch.await() ，等待计数器减到零，表示所有线程都已完成任务。