mirrors/woodpecker
1
0
Fork 0
mirror of https://github.com/woodpecker-ci/woodpecker.git synced 2024-11-20 08:51:01 +00:00
Code Issues Projects Releases Wiki Activity

Don't allow "logging" messages after eof is received

Browse source
This commit is contained in:
Jacob McCann 2017-02-14 13:33:29 -06:00
parent 89607232bc
commit c0fcc2dfed
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
server/stream.go
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
}