用Future与CountDownLatch实现多线程执行多个异步任务，任务全部完成后返回结果

最新推荐文章于 2024-07-11 15:09:28 发布

0oIronhide

最新推荐文章于 2024-07-11 15:09:28 发布

阅读量1.7k

点赞数

分类专栏： Java 多线程/并发文章标签： java

本文链接：https://blog.youkuaiyun.com/qq_38599840/article/details/120708245

版权

Java 多线程/并发专栏收录该内容

6 篇文章

订阅专栏

本文介绍了如何使用Java的Future和CountDownLatch实现异步任务在多线程中并发执行，并在所有任务完成后返回结果。通过ThreadPoolExecutor管理线程池，展示了如何创建、提交任务和处理Future结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.*;

/**
 * 用Future与CountDownLatch实现多线程执行多个异步任务，任务全部完成后返回结果
 */
public class TestFuture {
    public static void main(String[] args) {
        TestFuture testFuture = new TestFuture();
        // 初始为三个任务数
        int taskCount = 3;
        List<Future<String>> futures = new ArrayList<>(taskCount);
        // CountDownLatch作为递减计数器，一个线程完成了一个任务，计数器减一，减为0时表示任务全部完成
        CountDownLatch downLatch = new CountDownLatch(taskCount);

        ExecutorService executorService = new ThreadPoolExecutor
                (taskCount, 5, 5, TimeUnit.SECONDS, new ArrayBlockingQueue<>(10));

        for (int i = 0; i < taskCount; i++) {
            // 开启三个异步任务
            Future<String> future = executorService.submit(testFuture.executeTask(i, downLatch));
            futures.add(future);
        }

        System.out.println("do other things");

        try {
            // 在downLatch不为0时，主线程会在此阻塞
            downLatch.await();
            executorService.shutdown();
            // 通过future获取结果
            List<String> result = testFuture.getFutureResult(futures);
            for (String s : result) {
                System.out.println(s);
            }
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

    public Callable<String> executeTask(int i, CountDownLatch downLatch) {
        return () -> {
            Thread.sleep(i * 1000L);
            downLatch.countDown();
            return "Result " + Thread.currentThread().getName();
        };
    }

    public List<String> getFutureResult(List<Future<String>> futures) {
        List<String> result = new ArrayList<>(3);
        for (Future<String> future : futures) {
            if (future.isDone() && !future.isCancelled()) {
                try {
                    result.add(future.get());
                } catch (InterruptedException | ExecutionException e) {
                    e.printStackTrace();
                }
            }
        }
        return result;
    }
}