这段获取异常根因的Java代码是否会导致无限循环？

先看这段获取异常根因的Java代码：

public static Throwable getCause(@Nonnull Throwable t) {
    while ((t instanceof ExecutionException || t instanceof CompletionException) && t.getCause() != null) {
        t = t.getCause();
    }

    return t;
}

是否存在无限循环风险？能构造出触发场景吗？ #

存在无限循环的风险。Java的Throwable允许构造循环引用的异常链——比如异常A的cause指向异常B，异常B的cause又指向异常A，只要其中至少一个是ExecutionException或CompletionException，这段代码就会陷入死循环。

举个具体的触发场景：

// 构造循环引用的异常链
ExecutionException exA = new ExecutionException(null);
CompletionException exB = new CompletionException(null);
exA.initCause(exB);
exB.initCause(exA);

// 调用getCause会触发无限循环
getCause(exA);

这段代码运行后，循环会一直在exA和exB之间切换，永远无法终止。

更优的处理方式 #

1. 添加循环上限（可行） #

设置一个合理的循环次数上限（比如100次），超过上限就直接返回当前异常，避免死循环。这种方式实现简单，足够覆盖绝大多数真实业务场景（正常的异常嵌套深度不会这么深）。

优化代码示例：

public static Throwable getCause(@Nonnull Throwable t) {
    int depth = 0;
    final int MAX_DEPTH = 100;
    while ((t instanceof ExecutionException || t instanceof CompletionException) 
           && t.getCause() != null 
           && depth < MAX_DEPTH) {
        t = t.getCause();
        depth++;
    }
    return t;
}

2. 跟踪已遍历的异常（更严谨） #

用一个Set记录已经处理过的Throwable实例，每次循环前检查当前异常是否已经存在于集合中，若存在则说明出现循环引用，直接终止循环。这种方式能精准识别循环引用，不会因为嵌套深度略深就提前终止。

优化代码示例：

public static Throwable getCause(@Nonnull Throwable t) {
    Set<Throwable> visited = new HashSet<>();
    while ((t instanceof ExecutionException || t instanceof CompletionException) 
           && t.getCause() != null 
           && !visited.contains(t)) {
        visited.add(t);
        t = t.getCause();
    }
    return t;
}

内容的提问来源于stack exchange，提问作者Robert O'Neal