在请求未知返回类型的接口时,需要根据不同类型灵活处理响应体。本文将介绍如何使用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方法提供了访问响应完整信息的权限,包括响应头和响应体,使你能够根据响应类型灵活地处理数据。