CountDownLatch
是 Java 并发包中的一个类,它可以用来协调多个线程之间的同步操作。CountDownLatch
可以将一个线程等待多个线程完成某个操作之后再执行,或者让多个线程等待一个线程完成某个操作之后再执行。
CountDownLatch
使用一个计数器来实现,计数器的初始值可以设置为任意值,当某个线程完成了一个操作后,计数器的值就会减去 1,当计数器的值变为 0 时,等待线程就会被唤醒执行。
CountDownLatch 的基本用法
1、创建 CountDownLatch
对象,并指定计数器的初始值。
CountDownLatch latch = new CountDownLatch(3);
上面的代码创建了一个计数器初始值为 3 的 CountDownLatch
对象。
2、在需要等待的线程中使用 await()
方法等待计数器。
try {
latch.await();
} catch (InterruptedException e) {
e.printStackTrace();
}
上面的代码将会等待计数器的值变为 0。
3、在需要等待的其他线程中,当完成某个操作时,使用 countDown()
方法将计数器的值减去 1。
latch.countDown();
上面的代码将会让计数器的值减去 1。
CountDownLatch 的应用场景
CountDownLatch
可以用于多个线程之间的同步操作,例如在一个任务中需要等待多个子任务全部完成后再执行。另外,它也可以用于等待某个操作完成后再执行其他操作,例如等待读取文件完成后再执行数据处理和分析等操作。
下面是一个使用 CountDownLatch
的示例代码:
public class Example {
public void doTask() {
CountDownLatch latch = new CountDownLatch(3); // 创建计数器,初始值为3
ExecutorService executor = Executors.newFixedThreadPool(3); // 创建线程池,线程数为3
// 提交3个线程执行任务
executor.submit(() -> {
// 执行任务1
latch.countDown(); // 标记任务完成
});
executor.submit(() -> {
// 执行任务2
latch.countDown(); // 标记任务完成
});
executor.submit(() -> {
// 执行任务3
latch.countDown(); // 标记任务完成
});
executor.shutdown(); // 关闭线程池
try {
latch.await(); // 等待计数器归零
} catch (InterruptedException e) {
e.printStackTrace();
}
// 所有任务完成后执行其他操作
System.out.println("All tasks completed.");
}
}
上面的代码中,创建了一个计数器初始值为3的 CountDownLatch
对象,然后提交了3个线程执行任务,并在每个任务完成时调用了 countDown()
方法标记任务完成。最后,在主线程中调用了 await()
方法等待计数器归零,当所有任务完成后,输出 "All tasks completed."。
总结
CountDownLatch
是一个非常有用的并发工具类,可以用于协调多个线程之间的同步操作。使用 CountDownLatch
需要注意计数器的初始值和计数器的变化,以及在等待线程中使用 await()
方法等待计数器,在其他线程中使用 countDown()
方法标记任务完成。CountDownLatch
的应用场景很多,例如需要等待多个子任务全部完成后再执行,或者等待某个操作完成后再执行其他操作。