c5_labsci/zciyon/sse.go
2026-01-27 00:52:00 +08:00

52 lines
1.0 KiB
Go

package zciyon
import (
"net/http"
"strings"
)
func SSEInit(w http.ResponseWriter) bool {
_, ok := w.(http.Flusher)
if !ok {
return false
}
w.Header().Set("Content-Type", "text/event-stream")
w.Header().Set("Cache-Control", "no-cache")
w.Header().Set("Connection", "keep-alive")
w.Header().Set("X-Accel-Buffering", "no")
return true
}
func SSESend_data(w http.ResponseWriter, data string, argid ...any) bool {
id := ""
if len(argid) > 0 {
id = Tostr(argid[0])
}
msg := ""
msg += "data: " + strings.Replace(data, "\n", "<br/>", -1) + "\n"
if id != "" {
msg += "id: " + id + "\n"
}
msg += "\n"
_, err := w.Write([]byte(msg))
if err != nil {
return false
}
w.(http.Flusher).Flush()
return true
}
func SSESend_event(w http.ResponseWriter, data string, argevent ...string) bool {
event := "_t_"
if len(argevent) > 0 {
event = argevent[0]
}
msg := "data: " + strings.Replace(data, "\n", "<br/>", -1) + "\nevent: " + event + "\n\n"
_, err := w.Write([]byte(msg))
if err != nil {
return false
}
w.(http.Flusher).Flush()
return true
}