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方法提供了访问响应完整信息的权限，包括响应头和响应体，使你能够根据响应类型灵活地处理数据。