Java如何使用CountDownLatch实现任务分批执行_Java并发任务协调方案-java教程-PHP中文网

Java如何使用CountDownLatch实现任务分批执行_Java并发任务协调方案

P粉602998670

发布： 2025-11-26 02:33:16

原创

118人浏览过

CountDownLatch通过计数器实现线程同步，主线程调用await等待，各任务线程完成时调用countDown，计数归零后主线程继续执行，适用于分批任务协调。

java如何使用countdownlatch实现任务分批执行_java并发任务协调方案

在Java并发编程中，当需要协调多个线程的执行，确保某些操作在其他任务完成后再执行时，CountDownLatch 是一个非常实用的工具。它允许一个或多个线程等待，直到一组操作执行完毕。利用这一特性，可以轻松实现任务的分批执行与统一同步。

CountDownLatch 简介

CountDownLatch 是 java.util.concurrent 包中的一个同步辅助类，通过一个计数器实现线程间的协作。计数器初始值为任务的数量，每当一个任务完成，调用 countDown() 方法将计数减一。等待的线程调用 await() 方法阻塞，直到计数器归零，才继续执行。

分批执行任务的场景设计

假设我们需要处理一批数据，比如1000条记录，为了提高效率，将其分为10批，每批100条，由独立线程处理。主线程需等待所有批次处理完成后，再进行后续汇总操作。此时可使用 CountDownLatch 实现协调。

示例代码如下：

GAIPPT

AI PPT制作和美化神器

1215

查看详情

立即学习“Java免费学习笔记（深入）”；

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class BatchTaskWithCountDownLatch {
    public static void main(String[] args) throws InterruptedException {
        int batchSize = 10;
        CountDownLatch latch = new CountDownLatch(batchSize);
        ExecutorService executor = Executors.newFixedThreadPool(5);

        for (int i = 0; i < batchSize; i++) {
            final int batchId = i;
            executor.submit(() -> {
                try {
                    // 模拟批处理任务
                    System.out.println("开始执行第 " + batchId + " 批任务");
                    Thread.sleep(1000); // 模拟耗时
                    System.out.println("完成第 " + batchId + " 批任务");
                } finally {
                    latch.countDown(); // 任务完成，计数减一
                }
            });
        }

        // 主线程等待所有批次完成
        latch.await();
        System.out.println("所有批次任务已完成，开始汇总...");

        executor.shutdown();
    }
}

登录后复制