소켓 메세지 수신 함수 등록 타이밍 수정

wshandler/wshandler.go

@@ -80,6 +80,7 @@ type subhandler struct {
 type WebsocketHandler struct {
 	authCaches            map[string]*subhandler
 	RedisSync             *redis.Client
+	ClientMessageReceiver WebSocketMessageReceiver
 }
 
 type wsConfig struct {
@@ -127,6 +128,7 @@ func NewWebsocketHandler(authglobal gocommon.AuthCollectionGlobal) (wsh *Websock
 	return &WebsocketHandler{
 		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)
 	}