JUC工具类: CountDownLatch详解

小蓝

2024 年 03 月 29 日

52 次浏览

暂无评论

2770字数

Linux 技术杂烩

Java并发编程中的 java.util.concurrent包提供了多种并发编程工具类，其中 CountDownLatch是一个非常有用的类，它允许一个或多个线程等待一系列指定操作的完成。

CountDownLatch简介

CountDownLatch是一个同步辅助类，它允许一个线程或多个线程等待，直到其他线程的一组操作执行完成。CountDownLatch有一个计数器，该计数器初始化为一个正整数，表示需要等待的事件数量。CountDownLatch提供了 countDown方法来表示一个事件已经发生了，每调用一次，计数器的值就减一。当计数器的值变为零时，所有等待的线程就会被释放，这意味着它们可以继续执行。

CountDownLatch的用法

CountDownLatch的常用于可以被拆分为一组独立子任务的场景，每个子任务执行完成后调用 countDown方法。这常见于并行处理任务或等待初始化操作完成。

示例场景

假设我们有一个任务，需要等待一个服务的多个组件完成初始化后才能启动。我们可以使用 CountDownLatch来确保所有组件都初始化完成后，主任务才开始执行。

CountDownLatch的具体实现

以下是一个 CountDownLatch的简单实现例子：

import java.util.concurrent.CountDownLatch;

public class ServiceStarter {
    public static void main(String[] args) {
        final CountDownLatch latch = new CountDownLatch(3); // 假设有3个服务组件
        for (int i = 1; i <= 3; i++) {
            final int componentId = i;
            new Thread(() -> {
                try {
                    // 模拟服务组件初始化
                    Thread.sleep((long) (Math.random() * 1000));
                    System.out.println("Component " + componentId + " has started.");
                } catch (InterruptedException e) {
                    Thread.currentThread().interrupt();
                } finally {
                    latch.countDown();
                }
            }).start();
        }

        try {
            latch.await(); // 等待所有服务组件启动
            System.out.println("All services are up, application is starting now!");
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
    }
}

上述代码中，我们启动了三个线程分别表示三个需要初始化的服务组件。每个线程启动后，就开始执行初始化操作，完成后调用 countDown方法。主线程中的 latch.await()会一直阻塞，直到计数器的值减到0。