如何根据接口返回类型自动处理响应？

import org.springframework.web.reactive.function.client.WebClient;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import reactor.core.publisher.ResponseEntity;

public class DynamicResponseHandling {

    public static Mono<ResponseEntity<String>> handleResponse(WebClient.RequestHeadersSpec<?> requestSpec) {
        return requestSpec
                .retrieve()
                .onStatus(HttpStatusCode::isError, clientResponse -> clientResponse.createException().flatMap(Mono::error))
                .toEntityFlux(String.class) // 使用 toEntityFlux 而不是 bodyToFlux
                .flatMap(responseEntity -> {
                    HttpHeaders headers2 = responseEntity.getHeaders();
                    MediaType contentType2 = headers2.getContentType();

                    if (MediaType.TEXT_EVENT_STREAM.isCompatibleWith(contentType2)) {
                        // 如果是 SSE，则直接返回 Flux<String> 作为响应体
                        return Mono.just(ResponseEntity.ok()
                                .headers(headers2)
                                .contentType(MediaType.TEXT_EVENT_STREAM)
                                .body(Flux.from(Objects.requireNonNull(responseEntity.getBody()))));
                    } else {
                        // 如果不是 SSE，则根据需要处理响应体
                        return Objects.requireNonNull(responseEntity.getBody()).reduce("", String::concat)
                                .map(body2 -> ResponseEntity.ok()
                                        .headers(headers2)
                                        .body(body2));
                    }
                })
                .onErrorResume(WebClientResponseException.class, e -> {
                    return Mono.just(ResponseEntity.status(e.getStatusCode())
                            .headers(e.getHeaders())
                            .body(e.getResponseBodyAsString()));
                })
                .onErrorResume(e -> {
                    // 处理其他异常，并将错误信息添加到响应头中
                    HttpHeaders errorHeaders = new HttpHeaders();
                    errorHeaders.set("error-status", "500");
                    errorHeaders.set("error-message", e.getMessage());
                    return Mono.just(ResponseEntity.status(500)
                            .headers(errorHeaders)
                            .body(Map.of(
                                    "status", "500",
                                    "error", "Internal Server Error",
                                    "message", e.getMessage()
                            )));
                });
    }

}