code cleanup

RabbyHub · Nov 26, 2023 · 93fb843 · 93fb843
1 parent a6cddbc
commit 93fb843
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 10 deletions.
diff --git a/relay/wshandlers.go b/relay/wshandlers.go
@@ -11,14 +11,6 @@ import (
 
 type WsMessageHandler func(*WsServer, SocketMessage)
 
-var (
-	messageHandlers = map[MessageType]WsMessageHandler{
-		Pub:  (*WsServer).pubMessage,
-		Sub:  (*WsServer).subMessage,
-		Ping: (*WsServer).handlePingMessage,
-	}
-)
-
 func (ws *WsServer) pubMessage(message SocketMessage) {
 	topic := message.Topic
 	publisher := message.client

diff --git a/relay/wsserver.go b/relay/wsserver.go
@@ -44,7 +44,7 @@ func NewWSServer(config *config.Config) *WsServer {
 		config: &config.WsServerConfig, // config
 
 		clients:         make(map[*client]struct{}),
-		register:        make(chan *client, 1024),
+		register:        make(chan *client, 4096),
 		unregister:      make(chan ClientUnregisterEvent, 4096),
 		pendingSessions: NewSortedPendingSessions(),
 
@@ -130,7 +130,6 @@ func (ws *WsServer) Run() {
 					log.Info("forward to subscriber", zap.Any("client", subscriber), zap.Any("message", message))
 					subscriber.send(message)
 				}
-
 				continue
 			}