package com.backendsys.modules.sse.utils;

import cn.hutool.core.convert.Convert;
import com.backendsys.modules.sse.emitter.SseEmitterManager;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;

import java.io.IOException;

@Component
public class SseUtil {

    // [SSE] 发送消息
    public void send(String data) {

        Long userId = 1L;

        SseEmitterManager manager = SseEmitterManager.getInstance();
        SseEmitter emitter = manager.getEmitter(Convert.toStr(userId));
        if (emitter != null) {
            try {
                emitter.send(SseEmitter.event().data(data));
            } catch (IOException e) {
                System.out.println(e.getMessage());
                manager.removeEmitter(emitter);
            }
        }
    }

}