Java并发编程：CountDownLatch的应用案例

原创柔光的暖阳◎ 2024-10-31 16:09 20阅读 0赞

Java并发编程中，`CountDownLatch` 是一个同步辅助类，在完成一组正在其他线程中执行的操作之前，它允许一个或多个线程一直等待。`CountDownLatch`通过一个计数器来实现，该计数器在初始化时被设置一个特定的值，每当一个任务执行完毕时，计数器的值就会减少，当计数器的值减少到0时，那些等待在这个`CountDownLatch`上的线程就会被唤醒。

以下是`CountDownLatch`的一个应用案例：

###场景描述假设我们有一个任务，需要等待多个线程完成各自的工作后，主线程才能继续执行。比如，我们有一个主线程需要等待三个子线程完成各自的数据加载任务后，才能进行数据分析。

###代码示例```javaimport java.util.concurrent.CountDownLatch;

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

// 创建并启动三个线程，每个线程完成一项任务 for (int i =0; i < threadCount; i++) {
 new Thread(new WorkerThread(latch, i)).start();
 }

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

//所有子线程完成后，主线程继续执行 System.out.println("所有数据加载完成，开始数据分析");
 }

static class WorkerThread implements Runnable {
 private final CountDownLatch latch;
 private final int threadNumber;

public WorkerThread(CountDownLatch latch, int threadNumber) {
 this.latch = latch;
 this.threadNumber = threadNumber;
 }

@Override public void run() {
 try {
//模拟数据加载任务 System.out.println("线程 " + threadNumber + "正在加载数据...");
Thread.sleep((long) (Math.random() *10000));
 System.out.println("线程 " + threadNumber + " 数据加载完成");
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
//任务完成后，计数器减1 latch.countDown();
 }
 }
 }
}
```###代码解释1. `CountDownLatch`被初始化为3，因为我们有三个线程需要等待。
2. 创建三个线程，每个线程执行`WorkerThread`的实例，该实例接收`CountDownLatch`对象和线程编号。
3.每个线程在完成其任务后调用`countDown()`方法，这会将`CountDownLatch`的计数减1。
4. 主线程调用`await()`方法，这会阻塞直到`CountDownLatch`的计数达到0，即所有子线程都调用了`countDown()`方法。

这个案例展示了如何使用`CountDownLatch`来确保主线程在所有子线程完成工作后才继续执行。