Java修炼终极指南:218 挂钩任务状态

从 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 被取消了，那么就没有什么可做的了。我们只是在日志中报告它。