ayo/gocommon
8
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

wshandler config 수정

Browse Source
This commit is contained in:
mountain
2023-08-31 20:57:17 +09:00
parent 57b518562e
commit 58de9a3f0c
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
wshandler/wshandler.go
View File

@ -143,14 +143,10 @@ type WebsocketHandler struct {
} }
type wsConfig struct { type wsConfig struct {
gocommon.StorageAddr gocommon.StorageAddr `json:"storage"`
Maingate string `json:"maingate_service_url"`
} }
var config wsConfig
func init() { func init() {
gocommon.LoadConfig(&config)
gob.Register(UpstreamMessage{}) gob.Register(UpstreamMessage{})
gob.Register(commandMessage{}) gob.Register(commandMessage{})
gob.Register(map[string]any{}) gob.Register(map[string]any{})
@ -161,6 +157,11 @@ func init() {
} }
func NewWebsocketHandler(consumer session.Consumer) (*WebsocketHandler, error) { 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"]) redisSync, err := gocommon.NewRedisClient(config.Redis["wshandler"])
if err != nil { if err != nil {
return nil, err return nil, err