From 58de9a3f0cc26dca03de762ca4724fad3683a263 Mon Sep 17 00:00:00 2001 From: mountain Date: Thu, 31 Aug 2023 20:57:17 +0900 Subject: [PATCH] =?UTF-8?q?wshandler=20config=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 | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/wshandler/wshandler.go b/wshandler/wshandler.go index d623f1c..1897fe5 100644 --- a/wshandler/wshandler.go +++ b/wshandler/wshandler.go @@ -143,14 +143,10 @@ type WebsocketHandler struct { } type wsConfig struct { - gocommon.StorageAddr - Maingate string `json:"maingate_service_url"` + gocommon.StorageAddr `json:"storage"` } -var config wsConfig - func init() { - gocommon.LoadConfig(&config) gob.Register(UpstreamMessage{}) gob.Register(commandMessage{}) gob.Register(map[string]any{}) @@ -161,6 +157,11 @@ func init() { } func NewWebsocketHandler(consumer session.Consumer) (*WebsocketHandler, error) { + var config wsConfig + if err := gocommon.LoadConfig(&config); err != nil { + return nil, err + } + redisSync, err := gocommon.NewRedisClient(config.Redis["wshandler"]) if err != nil { return nil, err