从 JDK 19 开始,我们可以依赖 Future#state()。此方法基于众所周知的 get(), isDone(), 和 isCancelled() 计算 Future 的状态,并返回一个 Future.State 枚举项,如下所示:
- CANCELLED - 任务被取消了。
- FAILED - 任务异常完成(带有异常)。
- RUNNING - 任务仍在运行(尚未完成)。
- SUCCESS - 任务正常完成,有结果(无异常)。
在以下代码片段中,我们分析了加载测试团队成员的状态,并相应地采取行动:
public static TestingTeam buildTestingTeam()
throws InterruptedException {
List<String> testers = new ArrayList<>();
try (ExecutorService executor
= Executors.newVirtualThreadPerTaskExecutor()) {
List<Future<String>> futures = executor.invokeAll(
List.of(() -> fetchTester(Integer.MAX_VALUE),
() -> fetchTester(2),
() -> fetchTester(Integer.MAX_VALUE)));
futures.forEach(f -> {
logger.info(() -> "Analyzing " + f + " state ...");
switch (f.state()) {
case RUNNING -> throw new IllegalStateException(
"Future is still in the running state ...");
case SUCCESS -> {
logger.info(() -> "Result: " + f.resultNow());
testers.add(f.resultNow());
}
case FAILED ->
logger.severe(() -> "Exception: "
+ f.exceptionNow().getMessage());
case CANCELLED ->
logger.info("Cancelled ?!?");
}
});
}
return new TestingTeam(testers.toArray(String[]::new));
}
我们知道,当执行到达 switch 块时,Future 对象应该完全正常或异常。因此,如果当前 Future 状态是 RUNNING,那么这是一个非常奇怪的局面(可能是一个 bug),我们抛出一个 IllegalStateException。接下来,如果 Future 状态是 SUCCESS(fetchTester(2)),那么我们就有一个可以通过 resultNow() 获取的结果。这个方法是在 JDK 19 中添加的,当我们确定有结果时非常有用。resultNow() 方法立即返回,不等待(与 get() 相比)。如果状态是 FAILED(fetchTester(Integer.MAX_VALUE)),那么我们通过 exceptionNow() 记录异常。这个方法也是在 JDK 19 中添加的,它立即返回失败 Future 的底层异常。最后,如果 Future 被取消了,那么就没有什么可做的了。我们只是在日志中报告它。