mirror of https://github.com/gin-gonic/gin.git
32 lines
671 B
Go
32 lines
671 B
Go
package render
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/manucorporat/sse"
|
|
)
|
|
|
|
type sseRender struct{}
|
|
|
|
var SSEvent Render = sseRender{}
|
|
|
|
func (_ sseRender) Render(w http.ResponseWriter, code int, data ...interface{}) error {
|
|
eventName := data[0].(string)
|
|
obj := data[1]
|
|
return WriteSSEvent(w, eventName, obj)
|
|
}
|
|
|
|
func WriteSSEvent(w http.ResponseWriter, eventName string, data interface{}) error {
|
|
header := w.Header()
|
|
if len(header.Get("Content-Type")) == 0 {
|
|
header.Set("Content-Type", sse.ContentType)
|
|
}
|
|
if len(header.Get("Cache-Control")) == 0 {
|
|
header.Set("Cache-Control", "no-cache")
|
|
}
|
|
return sse.Encode(w, sse.Event{
|
|
Event: eventName,
|
|
Data: data,
|
|
})
|
|
}
|