Add optional compression

websocket/client.go

@@ -2,6 +2,7 @@ package websocket
 
 import (
 	"bytes"
+	"compress/gzip"
 	"crypto/tls"
 	"crypto/x509"
 	"encoding/json"
@@ -803,8 +804,7 @@ func (c *Client) readPumpWithDisconnectDetection() {
 		case <-c.done:
 			return
 		default:
-			var msg WSMessage
+			messageType, p, err := c.conn.ReadMessage()
-			err := c.conn.ReadJSON(&msg)
 			if err != nil {
 				// Check if we're shutting down or explicitly disconnected before logging error
 				select {
@@ -829,6 +829,30 @@ func (c *Client) readPumpWithDisconnectDetection() {
 				}
 			}
 
+			// Decompress binary frames (gzip-compressed JSON)
+			var data []byte
+			if messageType == websocket.BinaryMessage {
+				gr, gzErr := gzip.NewReader(bytes.NewReader(p))
+				if gzErr != nil {
+					logger.Error("websocket: failed to create gzip reader: %v", gzErr)
+					continue
+				}
+				data, gzErr = io.ReadAll(gr)
+				gr.Close()
+				if gzErr != nil {
+					logger.Error("websocket: failed to decompress message: %v", gzErr)
+					continue
+				}
+			} else {
+				data = p
+			}
+
+			var msg WSMessage
+			if err = json.Unmarshal(data, &msg); err != nil {
+				logger.Error("websocket: failed to parse message: %v", err)
+				continue
+			}
+
 			// Update config version from incoming message
 			c.setConfigVersion(msg.ConfigVersion)