Java CompletableFuture超时控制的优雅实践

在Java并发编程中，CompletableFuture无疑是一个强大的工具，它提供了非阻塞的异步编程模型，使得代码更加简洁高效。然而，在实际应用中，我们常常需要面对超时控制的问题。如何在CompletableFuture中优雅地实现超时控制，成为了许多开发者关注的焦点。

超时控制的必要性

在异步编程中，超时控制是不可或缺的一环。它不仅能防止因长时间等待而导致的资源浪费，还能提高系统的健壮性和响应性。想象一下，如果一个远程服务调用长时间没有响应，我们的程序是否会陷入无尽的等待？显然，这是不可接受的。

传统超时控制的痛点

在CompletableFuture出现之前，我们通常使用Future结合ExecutorService来实现异步任务。然而，Future的超时控制并不那么直观。我们需要使用get(long timeout, TimeUnit unit)方法，并在捕获TimeoutException时进行处理。这不仅代码冗长，还容易引入新的问题。

Future<String> future = executor.submit(() -> {
    // 模拟长时间任务
    Thread.sleep(5000);
    return "Result";
});

try {
    String result = future.get(3, TimeUnit.SECONDS);
    System.out.println(result);
} catch (TimeoutException e) {
    System.out.println("Timeout occurred");
} catch (InterruptedException | ExecutionException e) {
    e.printStackTrace();
}

CompletableFuture的超时控制

CompletableFuture为我们提供了更加优雅的超时控制方式。通过orTimeout和completeOnTimeout方法，我们可以轻松实现超时逻辑。

使用`orTimeout`

orTimeout方法会在指定的时间内，如果任务没有完成，直接抛出TimeoutException。

CompletableFuture<String> completableFuture = CompletableFuture.supplyAsync(() -> {
    try {
        Thread.sleep(5000);
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    return "Result";
});

completableFuture.orTimeout(3, TimeUnit.SECONDS)
    .thenAccept(result -> System.out.println(result))
    .exceptionally(ex -> {
        System.out.println("Timeout occurred: " + ex.getMessage());
        return null;
    });

使用`completeOnTimeout`

completeOnTimeout方法则会在超时后，使用指定的值完成CompletableFuture。

CompletableFuture<String> completableFuture = CompletableFuture.supplyAsync(() -> {
    try {
        Thread.sleep(5000);
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    return "Result";
});

completableFuture.completeOnTimeout("Default Result", 3, TimeUnit.SECONDS)
    .thenAccept(result -> System.out.println(result));

实战应用场景

在实际项目中，超时控制的应用场景非常广泛。例如，在一个微服务架构中，我们可能需要调用多个服务来获取数据。如果某个服务响应过慢，我们希望能够在超时后进行降级处理。

CompletableFuture<String> serviceA = CompletableFuture.supplyAsync(() -> {
    // 调用服务A
    return "Data from A";
});

CompletableFuture<String> serviceB = CompletableFuture.supplyAsync(() -> {
    // 调用服务B
    try {
        Thread.sleep(5000);
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    return "Data from B";
});

CompletableFuture<String> result = serviceA.thenCombine(serviceB, (a, b) -> a + " & " + b)
    .completeOnTimeout("Fallback Result", 3, TimeUnit.SECONDS);

result.thenAccept(System.out::println);