使用Spring SseEmitter实现服务端推送的示例代码

@RestController
@RequestMapping("/api/sse")
public class SseController {

    // 用于存储已连接的客户端，实际生产中建议使用专门的服务类管理
    private static final Map<String, SseEmitter> emitters = new ConcurrentHashMap<>();

    @GetMapping(value = "/subscribe/{userId}", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
    public SseEmitter subscribe(@PathVariable String userId) {
        // 设置超时时间（单位毫秒），0 表示永不超时
        SseEmitter emitter = new SseEmitter(60_000L); 

        emitters.put(userId, emitter);

        // 注册回调
        emitter.onCompletion(() -> emitters.remove(userId));
        emitter.onTimeout(() -> emitters.remove(userId));
        emitter.onError((e) -> emitters.remove(userId));

        // 建立连接后立即发送一条消息，防止某些代理或浏览器认为连接已失效
        try {
            emitter.send(SseEmitter.event().name("INIT").data("Connected!"));
        } catch (IOException e) {
            emitter.completeWithError(e);
        }

        return emitter;
    }
}

public void sendMessage(String userId, String content) {
    SseEmitter emitter = emitters.get(userId);
    if (emitter != null) {
        try {
            // 可以发送纯文本，也可以发送封装好的对象（会自动转 JSON）
            emitter.send(SseEmitter.event()
                    .id(UUID.randomUUID().toString())
                    .name("message") // 对应前端的事件名
                    .data(content)
                    .reconnectTime(3000)); // 提示客户端重连间隔
        } catch (IOException e) {
            emitters.remove(userId);
            emitter.completeWithError(e);
        }
    }
}

const eventSource = new EventSource('/api/sse/subscribe/user123');

eventSource.onmessage = (event) => {
    console.log("收到通用消息:", event.data);
};

// 监听特定类型的事件 (对应后端的 .name("message"))
eventSource.addEventListener('message', (event) => {
    const data = JSON.parse(event.data);
    console.log("处理后的数据:", data);
});

eventSource.onerror = (error) => {
    console.error("SSE 错误:", error);
    eventSource.close();
};