Evan Reichard
b2c56ac286
All checks were successful
continuous-integration/drone/push Build is passing
77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
package api
|
|
|
|
import (
|
|
"bytes"
|
|
"html/template"
|
|
"net/http"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type streamer struct {
|
|
templates map[string]*template.Template
|
|
writer gin.ResponseWriter
|
|
mutex sync.Mutex
|
|
completeCh chan struct{}
|
|
}
|
|
|
|
func (api *API) newStreamer(c *gin.Context, data string) *streamer {
|
|
stream := &streamer{
|
|
templates: api.Templates,
|
|
writer: c.Writer,
|
|
completeCh: make(chan struct{}),
|
|
}
|
|
|
|
// Set Headers
|
|
header := stream.writer.Header()
|
|
header.Set("Transfer-Encoding", "chunked")
|
|
header.Set("Content-Type", "text/html; charset=utf-8")
|
|
header.Set("X-Content-Type-Options", "nosniff")
|
|
stream.writer.WriteHeader(http.StatusOK)
|
|
|
|
// Send Open Element Tags
|
|
stream.write(data)
|
|
|
|
// Keep Alive
|
|
go func() {
|
|
closeCh := stream.writer.CloseNotify()
|
|
for {
|
|
select {
|
|
case <-stream.completeCh:
|
|
return
|
|
case <-closeCh:
|
|
return
|
|
default:
|
|
stream.write("<!-- ping -->")
|
|
time.Sleep(2 * time.Second)
|
|
}
|
|
}
|
|
}()
|
|
|
|
return stream
|
|
}
|
|
|
|
func (stream *streamer) write(str string) {
|
|
stream.mutex.Lock()
|
|
stream.writer.WriteString(str)
|
|
stream.writer.(http.Flusher).Flush()
|
|
stream.mutex.Unlock()
|
|
}
|
|
|
|
func (stream *streamer) send(templateName string, templateVars gin.H) {
|
|
t := stream.templates[templateName]
|
|
buf := &bytes.Buffer{}
|
|
_ = t.ExecuteTemplate(buf, templateName, templateVars)
|
|
stream.write(buf.String())
|
|
}
|
|
|
|
func (stream *streamer) close(data string) {
|
|
// Send Close Element Tags
|
|
stream.write(data)
|
|
|
|
// Close
|
|
close(stream.completeCh)
|
|
}
|