结合CompletableFuture与Spring的Sleuth结合工具类与allOf以及anyOf

干货满满张哈希

发布于 2021-04-12 14:25:57

5590

发布于 2021-04-12 14:25:57

系列目录：

之前实现的CompletableFutureWithSpan，不能直接使用anyOf或者allOf。因为查看源码：

public static CompletableFuture allOf(CompletableFuture... cfs) {
    return andTree(cfs, 0, cfs.length - 1);
}

这里的andTree是内部方法，其实就是递归并联所有传入的CompletableFuture。这个递归如何截止呢？主要由以下的方法结束

final void bipush(CompletableFuture b, BiCompletion c) {
    if (c != null) {
        while (result == null) {
            if (tryPushStack(c)) {
                if (b.result == null)
                    b.unipush(new CoCompletion(c));
                else if (result != null)
                    c.tryFire(SYNC);
                return;
            }
        }
        b.unipush(c);
    }
}

非常遗憾，这个方法是final的，修改的field也是内部field，CompletableFutureWithSpan是基于代理实现的，所以直接用原有的allOf还有anyOf是不可行的，继承覆盖bipush也不可行。

只好实现自己的：

public static CompletableFutureWithSpan allOf(Tracer tracer, CompletableFutureWithSpan... cfs) {
        //需要转换
        CompletableFuture[] completableFutures = Arrays.stream(cfs).map(completableFutureWithSpan -> completableFutureWithSpan.completableFuture).collect(Collectors.toList()).toArray(new CompletableFuture[0]);
        return from(CompletableFuture.allOf(completableFutures), tracer);
}


public static CompletableFuture anyOf(Tracer tracer, CompletableFutureWithSpan... cfs) {
    //需要转换
    CompletableFuture[] completableFutures = Arrays.stream(cfs).map(completableFutureWithSpan -> completableFutureWithSpan.completableFuture).collect(Collectors.toList()).toArray(new CompletableFuture[0]);
    return from(CompletableFuture.anyOf(completableFutures), tracer);
}

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2019/11/22 ，如有侵权请联系 cloudcommunity@tencent.com 删除

spring