mirror of https://github.com/gorilla/websocket.git
Add mutex to prevent concurrent connection write
This commit is contained in:
parent
0b847f2fac
commit
8400ba9d1b
|
@ -46,6 +46,9 @@ type Client struct {
|
|||
|
||||
// Buffered channel of outbound messages.
|
||||
send chan []byte
|
||||
|
||||
// Mutex to prevent concurrent writing to connection.
|
||||
mux sync.Mutex
|
||||
}
|
||||
|
||||
// readPump pumps messages from the websocket connection to the hub.
|
||||
|
@ -72,6 +75,8 @@ func (c *Client) readPump() {
|
|||
|
||||
// write writes a message with the given message type and payload.
|
||||
func (c *Client) write(mt int, payload []byte) error {
|
||||
c.mux.Lock()
|
||||
defer c.mux.Unlock()
|
||||
c.conn.SetWriteDeadline(time.Now().Add(writeWait))
|
||||
return c.conn.WriteMessage(mt, payload)
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue