java 多线程 CountDownLatch使用

Java中的CountDownLatch是一个同步工具类，它可以用来协调多个线程之间的同步操作。它通过一个计数器来实现，计数器的初始值为线程的数量。

使用CountDownLatch的基本步骤如下：

1. 创建一个CountDownLatch对象，指定计数器的初始值。
2. 在主线程中调用CountDownLatch的await()方法，让主线程等待计数器变为0。
3. 在子线程中执行任务，当任务执行完毕后，调用CountDownLatch的countDown()方法将计数器减1。
4. 主线程在调用await()方法后会一直等待，直到计数器减为0，然后继续执行。

以下是一个简单的示例：

import java.util.concurrent.CountDownLatch;

public class CountDownLatchExample {
    public static void main(String[] args) throws InterruptedException {
        int threadCount = 5;
        CountDownLatch latch = new CountDownLatch(threadCount);

        for (int i = 0; i < threadCount; i++) {
            Thread thread = new Thread(new Worker(latch));
            thread.start();
        }

        // 主线程等待所有子线程执行完毕
        latch.await();

        System.out.println("All workers finished their tasks.");
    }

    static class Worker implements Runnable {
        private final CountDownLatch latch;

        public Worker(CountDownLatch latch) {
            this.latch = latch;
        }

        @Override
        public void run() {
            try {
                // 模拟耗时任务
                Thread.sleep(1000);
                System.out.println("Task executed by " + Thread.currentThread().getName());

                // 任务执行完毕后调用countDown()方法
                latch.countDown();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}

上述示例中，创建了5个子线程来执行任务，每个子线程执行完任务后都调用了CountDownLatch的countDown()方法将计数器减1。主线程调用await()方法等待计数器变为0，当所有子线程执行完毕后，主线程继续执行，输出"All workers finished their tasks"。