livereload: Fix data race in close

Fixes #2625
This commit is contained in:
Bjørn Erik Pedersen 2017-04-30 01:10:03 +02:00
parent 93a447c5dd
commit 355736ec35
2 changed files with 13 additions and 2 deletions

View file

@ -15,6 +15,7 @@ package livereload
import (
"bytes"
"sync"
"github.com/gorilla/websocket"
)
@ -25,6 +26,16 @@ type connection struct {
// Buffered channel of outbound messages.
send chan []byte
// There is a potential data race, especially visible with large files.
// This is protected by synchronisation of the send channel's close.
closer sync.Once
}
func (c *connection) close() {
c.closer.Do(func() {
close(c.send)
})
}
func (c *connection) reader() {

View file

@ -41,14 +41,14 @@ func (h *hub) run() {
h.connections[c] = true
case c := <-h.unregister:
delete(h.connections, c)
close(c.send)
c.close()
case m := <-h.broadcast:
for c := range h.connections {
select {
case c.send <- m:
default:
delete(h.connections, c)
close(c.send)
c.close()
}
}
}