use protobuf for clients with protocol > 1 (#52)

pkg/rtc/types/protocol_version.go

@@ -8,6 +8,10 @@ func (v ProtocolVersion) SupportsPackedStreamId() bool {
 	return v > 0
 }
 
+func (v ProtocolVersion) SupportsProtobuf() bool {
+	return v > 0
+}
+
 func (v ProtocolVersion) HandlesDataPackets() bool {
 	return v > 1
 }

pkg/service/rtcservice.go

@@ -8,6 +8,7 @@ import (
 	"strings"
 
 	"github.com/gorilla/websocket"
+	"github.com/livekit/livekit-server/pkg/rtc/types"
 
 	"github.com/livekit/livekit-server/pkg/config"
 	"github.com/livekit/livekit-server/pkg/logger"
@@ -126,6 +127,9 @@ func (s *RTCService) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 	sigConn := NewWSSignalConnection(conn)
+	if types.ProtocolVersion(pi.ProtocolVersion).SupportsProtobuf() {
+		sigConn.useJSON = false
+	}
 
 	logger.Infow("new client WS connected",
 		"connectionId", connId,

pkg/service/wsprotocol.go

@@ -45,10 +45,12 @@ func (c *WSSignalConnection) ReadRequest() (*livekit.SignalRequest, error) {
 		msg := &livekit.SignalRequest{}
 		switch messageType {
 		case websocket.BinaryMessage:
-			c.mu.Lock()
-			// switch to protobuf if client supports it
-			c.useJSON = false
-			c.mu.Unlock()
+			if c.useJSON {
+				c.mu.Lock()
+				// switch to protobuf if client supports it
+				c.useJSON = false
+				c.mu.Unlock()
+			}
 			// protobuf encoded
 			err := proto.Unmarshal(payload, msg)
 			return msg, err