webclient根据响应类型处理流数据
问题：如何使webclient根据接口返回类型处理流数据？如果返回类型是text/event-stream，则逐条返回；否则，一次性返回所有数据。
回答：
使用webclient的toentityflux方法可以实现上述需求。该方法返回一个包含完整响应信息的mono，包括状态码、头信息和响应体。通过检查头信息中的content-type，我们可以确定响应类型并相应地处理数据。
以下代码示例演示了如何使用toentityflux方法：
return requestSpec
.retrieve()
.onStatus(HttpStatusCode::isError, clientResponse -> clientResponse.createException().flatMap(Mono::error))
.toEntityFlux(String.class) // 使用toEntityFlux代替exchangeToMono或exchangeToFlux
.flatMap(responseEntity -> {
HttpHeaders headers2 = responseEntity.getHeaders();
MediaType contentType2 = headers2.getContentType();
if (MediaType.TEXT_EVENT_STREAM.isCompatibleWith(contentType2)) {
// 如果是SSE，直接返回Flux作为响应体
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)); // 返回Mono，而不是Flux
}
})
.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方法，我们可以灵活地处理响应类型并实现逐条或一次性返回数据。
