根据接口返回类型自动处理响应
你在请求接口时遇到一个问题，因为你不知道接口返回的类型。对于这种情况，你需要根据返回类型来动态处理响应。
实现此功能的关键在于使用响应头信息，特别是 content-type 头。
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() ))); }); } }
通过使用 toentityflux，你可以获取包含 响应体、状态码 和 头信息 的完整响应。然后，你可以使用头信息，特别是 content-type，来确定返回的类型，并相应地处理响应。
具体来说，如果你在头信息中检测到 text/event-stream，则这意味着接口返回了服务器端事件（sse）。在这种情况下，返回一个 flux
否则，你可以将响应体作为一次性值返回，例如字符串。
这种方法提供了根据接口返回类型动态处理响应的灵活性。
