From 5dc9d4dca4c39394f7dc4953ed307db90c4e513a Mon Sep 17 00:00:00 2001
From: mountain <mountain@action2quare.com>
Date: Thu, 6 Jul 2023 14:29:53 +0900
Subject: [PATCH] =?UTF-8?q?=EC=86=8C=EC=BC=93=20=EB=A9=94=EC=84=B8?=
 =?UTF-8?q?=EC=A7=80=20=EC=88=98=EC=8B=A0=20=ED=95=A8=EC=88=98=20=EB=93=B1?=
 =?UTF-8?q?=EB=A1=9D=20=ED=83=80=EC=9D=B4=EB=B0=8D=20=EC=88=98=EC=A0=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 wshandler/wshandler.go | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/wshandler/wshandler.go b/wshandler/wshandler.go
index 144abbf..60eb526 100644
--- a/wshandler/wshandler.go
+++ b/wshandler/wshandler.go
@@ -78,8 +78,9 @@ type subhandler struct {
 
 // WebsocketHandler  :
 type WebsocketHandler struct {
-	authCaches map[string]*subhandler
-	RedisSync  *redis.Client
+	authCaches            map[string]*subhandler
+	RedisSync             *redis.Client
+	ClientMessageReceiver WebSocketMessageReceiver
 }
 
 type wsConfig struct {
@@ -125,8 +126,9 @@ func NewWebsocketHandler(authglobal gocommon.AuthCollectionGlobal) (wsh *Websock
 	}
 
 	return &WebsocketHandler{
-		authCaches: authCaches,
-		RedisSync:  redisSync,
+		authCaches:            authCaches,
+		RedisSync:             redisSync,
+		ClientMessageReceiver: func(primitive.ObjectID, string, WebSocketMessageType, io.Reader) {},
 	}
 }
 
@@ -136,7 +138,7 @@ func (ws *WebsocketHandler) Cleanup() {
 	}
 }
 
-func (ws *WebsocketHandler) RegisterHandlers(ctx context.Context, serveMux *http.ServeMux, prefix string, receiver WebSocketMessageReceiver) error {
+func (ws *WebsocketHandler) RegisterHandlers(ctx context.Context, serveMux *http.ServeMux, prefix string) error {
 	for region, sh := range ws.authCaches {
 		if region == "default" {
 			region = ""
@@ -147,8 +149,7 @@ func (ws *WebsocketHandler) RegisterHandlers(ctx context.Context, serveMux *http
 		} else {
 			serveMux.HandleFunc(url, sh.upgrade)
 		}
-		sh.callReceiver = receiver
-
+		sh.callReceiver = ws.ClientMessageReceiver
 		go sh.mainLoop(ctx)
 	}