From 8400ba9d1b04834eb1e2527e766cda6c042fc098 Mon Sep 17 00:00:00 2001 From: Justin Powell Date: Tue, 25 Oct 2016 11:34:41 -0700 Subject: [PATCH] Add mutex to prevent concurrent connection write --- examples/chat/client.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/examples/chat/client.go b/examples/chat/client.go index c3c518d..fb53f90 100644 --- a/examples/chat/client.go +++ b/examples/chat/client.go @@ -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) }