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

receiver 함수를 region 별로 분리

Browse Source
This commit is contained in:
mountain
2023-07-10 13:04:45 +09:00
parent 1f668586f2
commit 8598d59ab4
1 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
wshandler/wshandler.go
View File

@ -83,7 +83,7 @@ type subhandler struct {
type WebsocketHandler struct {
authCaches map[string]*subhandler
RedisSync *redis.Client
receiverChain []WebSocketMessageReceiver
receiverChain map[string][]WebSocketMessageReceiver
}
type wsConfig struct {
@ -132,20 +132,24 @@ func NewWebsocketHandler(authglobal gocommon.AuthCollectionGlobal) (wsh *Websock
return &WebsocketHandler{
authCaches: authCaches,
RedisSync: redisSync,
receiverChain: make(map[string][]WebSocketMessageReceiver),
}
}
func (ws *WebsocketHandler) RegisterReceiver(receiver WebSocketMessageReceiver) {
ws.receiverChain = append(ws.receiverChain, receiver)
func (ws *WebsocketHandler) RegisterReceiver(region string, receiver WebSocketMessageReceiver) {
ws.receiverChain[region] = append(ws.receiverChain[region], receiver)
}
func (ws *WebsocketHandler) Start(ctx context.Context) {
for _, sh := range ws.authCaches {
if len(ws.receiverChain) == 1 {
sh.callReceiver = ws.receiverChain[0]
for region, sh := range ws.authCaches {
chain := ws.receiverChain[region]
if len(chain) == 0 {
sh.callReceiver = func(accid primitive.ObjectID, alias string, messageType WebSocketMessageType, body io.Reader) {}
} else if len(chain) == 1 {
sh.callReceiver = chain[0]
} else {
sh.callReceiver = func(accid primitive.ObjectID, alias string, messageType WebSocketMessageType, body io.Reader) {
for _, r := range ws.receiverChain {
for _, r := range chain {
r(accid, alias, messageType, body)
}
}