
根据响应类型处理webclient响应
问题:
如何根据接口返回类型是text/event-stream则逐条返回,否则一次性全部返回?
魔乐社区
天翼云和华为联合打造的AI开发者社区,支持AI模型评测训练、全流程开发应用
102 查看详情
回答:
使用webclient的toentityflux,而不要使用exchangetomono或exchangetoflux。
WebClient webClient = WebClient.create();Mono<ResponseEntity> response = webClient.get() .uri("http://example.com") .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,逐条返回 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() ))); });
以上就是WebClient响应处理:如何根据返回类型区分逐条和一次性返回数据?的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/373645.html
微信扫一扫
支付宝扫一扫