1. 描述
实现一个像 ai问答
类似,一个接口,先处理好的先返回到客户端渲染,等下部分数据处理好再接着返回到客户端,直到所有数据处理结束,接口调用完成。
2. 代码实现
以下是一个简单的代码示例
@RestController
public class StreamResponseController {
@SneakyThrows
@GetMapping("/stream")
public ResponseEntity<Void> stream(HttpServletResponse response) {
response.setContentType("text/plain"); // 设置响应内容类型为纯文本
response.setCharacterEncoding("UTF-8"); // 设置响应字符编码为 UTF-8
//模拟一段一段数据
String textData1 = "这是一段文字数据。这是一段文字数据。这是一段文字数据。这是一段文字数据。这是一段文字数据。这是一段文字数据。这是一段文字数据。\n"; // 要返回的文字数据
String[] test = {textData1, textData1, textData1, textData1, textData1, textData1};
// 获取输出流
OutputStream outputStream = response.getOutputStream();
try {
for (String textData : test) {
//中文转码
outputStream.write(textData.getBytes(StandardCharsets.UTF_8));
//模拟每一段文字的处理时长
TimeUnit.SECONDS.sleep(3);
//当前文字数据输出到客户端
outputStream.flush();
}
// 将文字数据以流的形式写入到输出流中
} finally {
// 关闭输出流
outputStream.close();
}
return null;
}
}