Merge pull request #1940 from jmccann/log_stream_eof

Don't allow "logging" messages after eof is received
This commit is contained in:
Brad Rydzewski 2017-02-15 09:14:01 +11:00 committed by GitHub
commit 6b4fcc9684

View file

@ -67,11 +67,15 @@ func LogStream(c *gin.Context) {
logs := make(chan []byte)
done := make(chan bool)
var eof bool
dest := fmt.Sprintf("/topic/logs.%d", job.ID)
client, _ := stomp.FromContext(c)
sub, err := client.Subscribe(dest, stomp.HandlerFunc(func(m *stomp.Message) {
if m.Header.GetBool("eof") {
eof = true
done <- true
} else if eof {
return
} else {
logs <- m.Body
}