用CountDownLatch解决并发程序设计中的常见问题

见解分享

2023-10-13 13:18:43

CountDownLatch简介

CountDownLatch是一个同步工具，它允许一个线程等待其他线程完成任务。它有一个计数器，初始化时指定计数器的值，然后其他线程可以调用countDown()方法来递减计数器的值。当计数器减为0时，等待的线程将被唤醒。

CountDownLatch通常用于以下场景：

主线程等待所有子线程完成任务后，再继续执行。
一个线程等待其他线程完成任务后，再继续执行。
多个线程等待一个事件发生后，再继续执行。

CountDownLatch的用法

CountDownLatch的用法非常简单，只需要三个步骤：

创建一个CountDownLatch对象，并指定计数器的初始值。
其他线程调用countDown()方法来递减计数器的值。
等待的线程调用await()方法来等待计数器减为0。

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

import java.util.concurrent.CountDownLatch;

public class CountDownLatchDemo {

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

        // 创建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();
        }

        // 主线程调用await()方法来等待计数器减为0
        try {
            latch.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        // 计数器减为0后，主线程继续执行
        System.out.println("主线程执行完毕");
    }
}