背景
使用 java 语言实现sse协议客户端消息接收,完成大模型流式推理的协议转换。
核心:基于 Spring 5 + 实现,关键类 WebClient
,代码如下:
/**
* @Author ouyangrongtao
* @Date 2024-05-30 13:54
* @Description SSE 客户端测试
**/
public class SSETest {
public static void main(String[] args) throws InterruptedException {
WebClient client = WebClient.create("http://localhost:8080");
Flux<ServerSentEvent> eventFlux = client.post()
.uri("/sse")
.retrieve()
.bodyToFlux(ServerSentEvent.class);
eventFlux.subscribe(event -> {
System.out.println("Received event: " + event.data());
});
// 响应等待
Thread.sleep(Integer.MAX_VALUE);
}
}