From 4bdd72152e6e0395c155eb92e645476dfb5e0aab Mon Sep 17 00:00:00 2001
From: mountain <mountain@action2quare.com>
Date: Thu, 6 Jul 2023 14:18:11 +0900
Subject: [PATCH] =?UTF-8?q?receiver=20=EC=9D=B4=EB=8F=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 wshandler/wshandler.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/wshandler/wshandler.go b/wshandler/wshandler.go
index 381c5b4..144abbf 100644
--- a/wshandler/wshandler.go
+++ b/wshandler/wshandler.go
@@ -86,7 +86,7 @@ type wsConfig struct {
 	SyncPipeline string `json:"ws_sync_pipeline"`
 }
 
-func NewWebsocketHandler(authglobal gocommon.AuthCollectionGlobal, receiver WebSocketMessageReceiver) (wsh *WebsocketHandler) {
+func NewWebsocketHandler(authglobal gocommon.AuthCollectionGlobal) (wsh *WebsocketHandler) {
 	var config wsConfig
 	gocommon.LoadConfig(&config)
 
@@ -119,7 +119,6 @@ func NewWebsocketHandler(authglobal gocommon.AuthCollectionGlobal, receiver WebS
 			redisSync:        redisSync,
 			connInOutChan:    make(chan *wsconn),
 			deliveryChan:     make(chan any, 1000),
-			callReceiver:     receiver,
 		}
 
 		authCaches[region] = sh
@@ -137,7 +136,7 @@ func (ws *WebsocketHandler) Cleanup() {
 	}
 }
 
-func (ws *WebsocketHandler) RegisterHandlers(ctx context.Context, serveMux *http.ServeMux, prefix string) error {
+func (ws *WebsocketHandler) RegisterHandlers(ctx context.Context, serveMux *http.ServeMux, prefix string, receiver WebSocketMessageReceiver) error {
 	for region, sh := range ws.authCaches {
 		if region == "default" {
 			region = ""
@@ -148,6 +147,7 @@ func (ws *WebsocketHandler) RegisterHandlers(ctx context.Context, serveMux *http
 		} else {
 			serveMux.HandleFunc(url, sh.upgrade)
 		}
+		sh.callReceiver = receiver
 
 		go sh.mainLoop(ctx)
 	}