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