From 93fb84323ebcc7616c680712d67d729d317ab054 Mon Sep 17 00:00:00 2001
From: cryptocifer <mantianyu@gmail.com>
Date: Sun, 26 Nov 2023 08:49:34 +0300
Subject: [PATCH] code cleanup

---
 relay/wshandlers.go | 8 --------
 relay/wsserver.go   | 3 +--
 2 files changed, 1 insertion(+), 10 deletions(-)

diff --git a/relay/wshandlers.go b/relay/wshandlers.go
index 50eeef6..5727c12 100644
--- 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
index ae81fb5..44126f2 100644
--- 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
 			}