JAVA笔记 | ResponseBodyEmitter等异步流式接口快速学习

Nu11PointerException2024-11-09 2:02

先简单记录下简单使用跟测试，后续再补充具体，最近有用到，简单来说就是后端(服务端)编写个发射器，实现一次请求，一直向前端客户端发射数据，直到发射器执行完毕，模拟ai一句一句回复的效果

ResponseBodyEmitter

测试代码

java 复制代码

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyEmitter;
import java.util.Date;
import java.util.concurrent.CompletableFuture;

/**
 * @Author: linjinde
 * @CreateTime: 2024-11-04
 */
@RestController
@RequestMapping("/dify")
public class DifyController {

    @GetMapping("/emitter")
    public ResponseBodyEmitter handle() {
        //创建ResponseBodyEmitter发射器，-1代表不超时
        ResponseBodyEmitter emitter = new ResponseBodyEmitter(-1L);
        CompletableFuture.runAsync(() -> {
            try {
                for (int i = 0; i < 10000; i++) {
                    System.out.println("bodyEmitter " + i);
                    //发送数据
                    emitter.send("bodyEmitter " + i + " @ " + new Date() + "\n");
                    //模拟两秒响应一次
                    Thread.sleep(2000);
                }
                emitter.complete();
            }catch (Exception e){
                // 发生异常时结束接口
                emitter.completeWithError(e);
            }

        });
        //postman或者网页无法直接现实响应体，因为发射器是设计面向客户端，要显示设计歌简单js客户端接收
        return emitter;
    }
}

请求测试

因为ResponseBodyEmitter是设计面向客户端持续发送数据，故postman跟网页正常情况下是不支持数据连续现实，需要javascript编写简单客户端才可以显示返回内容，对于后端来说很麻烦。

简单解决方法：这边就用cmd curl直接请求

测试完毕

后续继续补充SseEmitter等