当前位置：首页 > news >正文

Spring Boot 实现流式响应（兼容 2.7.x）

news 来源：原创 2025/6/7 7:08:47

在实际开发中，我们可能会遇到一些流式数据处理的场景，比如接收来自上游接口的 Server-Sent Events（SSE） 或 流式 JSON 内容，并将其原样中转给前端页面或客户端。这种情况下，传统的 RestTemplate 缓存机制会导致数据必须等待全部接收完毕后再处理，违背了流式传输的初衷。

本文将介绍如何在 Spring Boot 2.7.x 中使用 RestTemplate 实现一个流式响应的中转接口。

一、关键点说明

禁用请求缓冲：
通过 SimpleClientHttpRequestFactory#setBufferRequestBody(false) 禁用缓冲，确保流式传输生效。
设置响应为 SSE 格式：
设置 HttpServletResponse 的响应头为 text/event-stream，支持前端基于 EventSource 的实时响应。
使用 RestTemplate.execute 方法：
通过 RestTemplate.execute() 方法自定义 RequestCallback 和 ResponseExtractor 实现对输入输出流的精细控制。

二、完整代码实现

/*** 处理流式响应的HTTP请求方法（流式响应, Spring Boot 2.7.x 兼容）** @param requestBody 请求体内容* @param url         请求URL* @param httpMethod  请求方法* @param response    HttpServletResponse，用于直接返回流数据*/
public static void executeStreamingRequest(Map<String, Object> requestBody,String url,HttpMethod httpMethod,HttpServletResponse response) throws IOException {// 1. 配置RestTemplate（启用流式）SimpleClientHttpRequestFactory requestFactory = new SimpleClientHttpRequestFactory();requestFactory.setBufferRequestBody(false); // 关键：禁用请求缓冲requestFactory.setConnectTimeout(30000);requestFactory.setReadTimeout(60000);RestTemplate restTemplate = new RestTemplate(requestFactory);// 2. 设置请求头HttpHeaders headers = new HttpHeaders();headers.setContentType(MediaType.APPLICATION_JSON);// 3. 执行流式请求try {log.info("【开始流式请求】URL: {}", url);restTemplate.execute(URI.create(url),httpMethod,new RequestCallback() {@Overridepublic void doWithRequest(ClientHttpRequest request) throws IOException {request.getHeaders().putAll(headers);if (requestBody != null) {new ObjectMapper().writeValue(request.getBody(), requestBody);}}},new ResponseExtractor<Void>() {@Overridepublic Void extractData(ClientHttpResponse clientResponse) throws IOException {// 设置响应头response.setContentType("text/event-stream");response.setCharacterEncoding("UTF-8");// 流式传输try (InputStream is = clientResponse.getBody();OutputStream os = response.getOutputStream()) {byte[] buffer = new byte[8192];int bytesRead;while ((bytesRead = is.read(buffer)) != -1) {os.write(buffer, 0, bytesRead);os.flush(); // 立即刷新}}return null;}});} catch (RestClientException e) {log.error("流式请求失败", e);response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);response.getWriter().write("服务调用失败: " + e.getMessage());}
}