Java并发：掌握CountDownLatch，轻松处理并发难题

后端

2023-12-28 18:54:54

在软件开发中，并发的使用越来越广泛，处理并发的复杂性和出错率也越来越高。作为 Java 并发编程的核心组件之一，CountDownLatch 可以帮助您简化并发操作，提高代码的可读性和可靠性。

CountDownLatch 的基本原理是使用一个计数器来跟踪正在执行的任务数量。当所有任务都完成后，计数器将减为 0，并且等待该计数器的线程将被唤醒。这种机制可以确保在所有任务完成之前，后续的处理不会开始。

CountDownLatch 的主要方法包括：

await(): 等待计数器减为 0。如果计数器尚未减为 0，则当前线程将被阻塞，直到计数器减为 0。
countDown(): 将计数器减 1。当计数器减为 0 时，所有等待该计数器的线程将被唤醒。
getCount(): 获取计数器的当前值。

CountDownLatch 可以用于各种并发场景，例如：

等待所有子线程完成任务后再继续执行主线程。
等待所有数据加载完成后再显示用户界面。
等待所有服务启动完成后再开始处理请求。

通过使用 CountDownLatch，您可以轻松地协调多个线程之间的操作，从而提高并发程序的可靠性和可维护性。

以下是一个使用 CountDownLatch 的示例：

import java.util.concurrent.CountDownLatch;

public class CountDownLatchExample {

    public static void main(String[] args) {
        // 创建一个 CountDownLatch 对象，初始值为 3
        CountDownLatch latch = new CountDownLatch(3);

        // 创建三个线程，每个线程都将调用 countDown() 方法来减少计数器
        for (int i = 0; i < 3; i++) {
            new Thread(() -> {
                System.out.println("线程 " + Thread.currentThread().getName() + " 开始执行");
                try {
                    Thread.sleep(1000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                latch.countDown();
                System.out.println("线程 " + Thread.currentThread().getName() + " 执行结束");
            }).start();
        }

        // 等待计数器减为 0，然后继续执行主线程
        try {
            latch.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        System.out.println("所有线程执行结束，主线程继续执行");
    }
}