CompletableFuture 合并操作

最新推荐文章于 2025-03-17 16:30:02 发布

tangyaya8

最新推荐文章于 2025-03-17 16:30:02 发布

阅读量5.3k

点赞数 2

分类专栏： javase 文章标签： stream

本文链接：https://blog.csdn.net/tangyaya8/article/details/109006788

版权

javase 专栏收录该内容

14 篇文章

订阅专栏

这篇博客探讨了在Java中如何使用CompletableFuture来处理批量异步请求下游RPC服务，并将其结果合并成一个List。文章通过模拟调用下游请求，展示了如何利用CompletableFuture.thenCombine与CompletableFuture.allOf方法来聚合请求。

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

背景

最近在使用CompletableFuture做异步代码的重构，其中遇到了一些问题，记录下来。

需求

批量请求下游RPC的List，然后合并为一个List

代码

1.模拟调用下游请求

    private CompletableFuture<List<Integer>> getIntegerList(int startIndex, int size) {
        return CompletableFuture.supplyAsync(() -> {
                    try {
                        TimeUnit.SECONDS.sleep(1);
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                    return Stream.iterate(startIndex, item -> item + 1).limit(size).collect(Collectors.toList());
                }
        );
    }

2.使用CompletableFuture.thenCombine来聚合

@Test
    public void testCompletableFutureWithCombine() {
        List<CompletableFuture<List<Integer>>> completableFutureList = new ArrayList<>(4);
        Instant startTime = Instant.now();
        for (int i = 1; i <= 30; i += 10) {
            CompletableFuture<List<Integer>> integerList = this.getIntegerList(i, 10);
            completableFutureList.add(integerList);
        }
        List<Integer> join = completableFutureList.stream()
                .reduce((fn1, fn2) -> fn1.thenCombine(fn2, (integers, integers2) -> Stream.of(integers, integers2).flatMap(Collection::stream).collect(Collectors.toList())))
                .orElse(CompletableFuture.completedFuture(Collections.emptyList()))
                .join();
        System.out.println(join);
        Instant endTime = Instant.now();

        System.out.println(Duration.between(startTime, endTime).getSeconds());
    }

在这里插入图片描述

3.使用CompletableFuture.allOf来聚合请求

   @Test
    public void testCompletableFutureForAll() {
        List<CompletableFuture<List<Integer>>> completableFutureList = new ArrayList<>(4);
        Instant startTime = Instant.now();
        for (int i = 1; i <= 30; i += 10) {
            CompletableFuture<List<Integer>> integerList = this.getIntegerList(i, 10);
            completableFutureList.add(integerList);
        }
        CompletableFuture<List<List<Integer>>> sequence = this.sequence(completableFutureList);
        CompletableFuture<List<Integer>> listCompletableFuture = sequence.thenApply(lists -> lists.stream().flatMap(Collection::stream).collect(Collectors.toList()));
        List<Integer> join = listCompletableFuture
                .join();
        System.out.println(join);
        Instant endTime = Instant.now();

        System.out.println(Duration.between(startTime, endTime).getSeconds());
    }

    public <T> CompletableFuture<List<T>> sequence(Collection<CompletableFuture<T>> futureList) {
        return CompletableFuture.allOf(futureList.toArray(new CompletableFuture<?>[0]))
                .thenApply(v -> futureList.stream()
                        .map(CompletableFuture::join)
                        .collect(Collectors.toList())
                );
    }

在这里插入图片描述