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;
    }

}
Logo

一站式 AI 云服务平台

更多推荐