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

receiver 이동

Browse Source
This commit is contained in:
mountain
2023-07-06 14:18:11 +09:00
parent 8c94fc6e29
commit 4bdd72152e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
wshandler/wshandler.go
View File

@ -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)
}