CompletableFuture的join最大等待时间

### CompletableFuture 中 `join` 方法的最大等待时间设置方式 在 Java 的 `CompletableFuture` 类中，`join()` 是一种用于获取异步任务结果的方法。然而，默认情况下，`join()` 方法是一个阻塞操作，它会一直等待直到任务完成或者抛出异常。为了实现最大等待时间的功能，可以通过组合其他 API 来达到目的。 #### 使用 `orTimeout` 实现超时控制 (Java 9+) 从 Java 9 开始，`CompletableFuture` 提供了一个内置的 `orTimeout(long timeout, TimeUnit unit)` 方法，可以直接用来设置超时时间[^5]。如果任务未能在指定时间内完成，则会抛出 `CompletionException`，其中嵌套的是 `TimeoutException`。 以下是基于 `orTimeout` 的代码示例： ```java import java.util.concurrent.CompletableFuture; import java.util.concurrent.TimeUnit; public class TimeoutExample { public static void main(String[] args) throws Exception { CompletableFuture<String> future = CompletableFuture.supplyAsync(() -> { try { Thread.sleep(3000); // Simulate a long-running task } catch (InterruptedException e) { throw new RuntimeException(e); } return "Result"; }); CompletableFuture<String> timedFuture = future.orTimeout(2, TimeUnit.SECONDS); try { String result = timedFuture.join(); // Will throw CompletionException with TimeoutException as cause after 2 seconds. System.out.println(result); } catch (Exception ex) { System.err.println("Task timed out or failed: " + ex.getCause()); } } } ``` --- #### 使用 `completeOnTimeout` 实现超时返回默认值 (Java 9+) 除了直接抛出异常外，还可以使用 `completeOnTimeout(T value, long timeout, TimeUnit unit)` 方法，在超时时返回一个默认值而不是中断任务。 以下是一个例子： ```java import java.util.concurrent.CompletableFuture; import java.util.concurrent.TimeUnit; public class DefaultTimeoutExample { public static void main(String[] args) throws Exception { CompletableFuture<String> future = CompletableFuture.supplyAsync(() -> { try { Thread.sleep(3000); // Simulate a long-running task } catch (InterruptedException e) { throw new RuntimeException(e); } return "Real Result"; }); CompletableFuture<String> timedFuture = future.completeOnTimeout("Default Value", 2, TimeUnit.SECONDS); String result = timedFuture.join(); System.out.println("Result is: " + result); // Prints "Default Value" } } ``` --- #### 自定义超时逻辑 (适用于 Java 8 和更高版本) 对于 Java 8 用户，由于缺少上述方法，可以借助 `ScheduledThreadPoolExecutor` 手动实现超时控制[^1]。具体做法是创建一个新的 `CompletableFuture` 并在其上触发超时异常，随后通过 `applyToEither` 或 `acceptEither` 判断哪个事件先发生——任务完成还是超时。 下面是一段自定义超时逻辑的代码： ```java import java.util.concurrent.*; import java.util.function.Supplier; public class CustomTimeoutExample { public static <T> CompletableFuture<T> completeWithTimeout( Supplier<CompletableFuture<T>> supplier, long timeoutMillis) { CompletableFuture<T> originalFuture = supplier.get(); CompletableFuture<Void> timeoutFuture = new CompletableFuture<>(); ScheduledExecutorService scheduler = Executors.newSingleThreadScheduledExecutor(); scheduler.schedule(timeoutFuture::completeExceptionally, timeoutMillis, TimeUnit.MILLISECONDS); return originalFuture.applyToEither( timeoutFuture.whenComplete((r, t) -> scheduler.shutdown()), Function.identity() ); } public static void main(String[] args) { CompletableFuture<String> future = completeWithTimeout( () -> CompletableFuture.supplyAsync(() -> { try { Thread.sleep(3000); // Simulate delay } catch (InterruptedException e) { throw new RuntimeException(e); } return "Success!"; }), 2000L // Timeout set to 2 seconds ); try { String result = future.join(); System.out.println("Result: " + result); } catch (Throwable throwable) { System.err.println("Operation timed out."); } } } ``` --- #### 总结 - **Java 9 及以上**：推荐使用 `orTimeout` 或 `completeOnTimeout` 这些内置方法来简化开发过程。 - **Java 8**：需手动结合 `ScheduledThreadPoolExecutor` 创建定时器，并配合 `applyToEither` 或 `acceptEither` 完成超时检测。 --- ###