Hello! 欢迎来到小浪资源网!


WebClient如何根据接口返回类型灵活处理响应体?


WebClient如何根据接口返回类型灵活处理响应体?

webclient根据接口返回类型逐条或全部返回

在请求未知返回类型的接口时,需要根据不同类型灵活处理响应体。本文将介绍如何使用webclient根据接口返回类型进行逐条返回或一次性全部返回。

对于text/Event-stream(sse)类型,可以使用toentityflux方法逐条返回,而对于其他类型则可以一次性返回全部响应体。以下代码演示了如何在webclient中实现此功能:

return requestSpec     .retrieve()     .onStatus(HttpStatusCode::isError, clientResponse -> clientResponse.createException().flatMap(Mono::error))     .toEntityFlux(String.class)     .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 {             // 其他类型,一次性返回             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方法提供了访问响应完整信息的权限,包括响应头和响应体,使你能够根据响应类型灵活地处理数据。

相关阅读