fix: concurrent write to websocket connection

withchao · Nov 21, 2024 · 7859013 · 7859013
1 parent af2f020
commit 7859013
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/msggateway/client.go b/internal/msggateway/client.go
@@ -446,11 +446,11 @@ func (c *Client) handlerTextMessage(b []byte) error {
 		if err != nil {
 			return err
 		}
+		c.w.Lock()
+		defer c.w.Unlock()
 		if err := c.conn.SetWriteDeadline(writeWait); err != nil {
 			return err
 		}
-		c.w.Lock()
-		defer c.w.Unlock()
 		return c.conn.WriteMessage(MessageText, msgData)
 	default:
 		return fmt.Errorf("not support message type %s", msg.Type)