SpringBoot3 流式响应 超时时间

在 Spring Boot 3 中，可以使用 `WebFlux` 模块实现流式响应。要设置超时时间，可以在 `ServerHttpResponse` 对象上调用 `setWriteHandler()` 方法并传入一个 `java.util.function.Consumer` 对象。这个 Consumer 对象接受一个 `org.reactivestreams.Subscription` 对象和一个 `java.time.Duration` 对象，可以在其中设置超时时间。

以下是一个示例代码：

@GetMapping(value = "/stream", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
public Flux<String> streamData(ServerHttpResponse response) {
    return Flux.interval(Duration.ofSeconds(1))
            .map(i -> "data " + i)
            .doOnSubscribe(subscription -> {
                response.setStatusCode(HttpStatus.OK);
                response.getHeaders().setContentType(MediaType.TEXT_EVENT_STREAM);
                response.setWriteHandler((outputStream) -> {
                    Consumer<Tuple2<Subscription, Duration>> timeoutHandler = (t) -> {
                        subscription.cancel();
                        try {
                            outputStream.write("timeout".getBytes());
                            outputStream.flush();
                            outputStream.close();
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    };
                    outputStream.onTimeout(Duration.ofSeconds(5), timeoutHandler);
                    outputStream.onClose(() -> {
                        subscription.cancel();
                    });
                    subscription.request(Long.MAX_VALUE);
                });
            });
}

在上面的代码中，我们使用 `Flux.interval()` 方法生成一个每秒钟产生一个递增数字的 Flux。然后，我们在 `doOnSubscribe()` 方法中设置响应头和写处理程序。在写处理程序中，我们设置了一个超时时间为 5 秒，并传入了一个超时处理程序。当超时发生时，我们取消订阅并向客户端发送 "timeout" 消息。如果客户端关闭连接，我们也会取消订阅。