From 4b35e0e6386b1a27e4dec854d1fc2c755f20aeef Mon Sep 17 00:00:00 2001
From: mountain <mountain@action2quare.com>
Date: Tue, 18 Jul 2023 09:45:27 +0900
Subject: [PATCH] =?UTF-8?q?EventReceiver=20=EC=9D=B8=ED=84=B0=ED=8E=98?=
 =?UTF-8?q?=EC=9D=B4=EC=8A=A4=20=EC=B6=94=EA=B0=80?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 wshandler/wshandler.go | 60 +++++++++++++++++++++++++++++++-----------
 1 file changed, 45 insertions(+), 15 deletions(-)

diff --git a/wshandler/wshandler.go b/wshandler/wshandler.go
index b6fd1b7..62fef08 100644
--- a/wshandler/wshandler.go
+++ b/wshandler/wshandler.go
@@ -87,7 +87,11 @@ type Sender struct {
 	Alias  string
 }
 
-type WebSocketMessageReceiver func(sender *Sender, messageType WebSocketMessageType, body io.Reader)
+type EventReceiver interface {
+	OnClientMessageReceived(sender *Sender, messageType WebSocketMessageType, body io.Reader)
+	OnRoomCreated(region, name string)
+	OnRoomDestroyed(region, name string)
+}
 
 type subhandler struct {
 	redisMsgChanName  string
@@ -96,10 +100,10 @@ type subhandler struct {
 	connInOutChan     chan *wsconn
 	deliveryChan      chan any
 	localDeliveryChan chan any
-	callReceiver      WebSocketMessageReceiver
+	callReceiver      EventReceiver
 	connWaitGroup     sync.WaitGroup
 	region            string
-	receiverChain     []WebSocketMessageReceiver
+	receiverChain     []EventReceiver
 }
 
 // WebsocketHandler  :
@@ -165,25 +169,50 @@ func NewWebsocketHandler() (*WebsocketHandler, error) {
 	}, nil
 }
 
-func (ws *WebsocketHandler) RegisterReceiver(region string, receiver WebSocketMessageReceiver) {
+func (ws *WebsocketHandler) RegisterReceiver(region string, receiver EventReceiver) {
 	if sh := ws.subhandlers[region]; sh != nil {
 		sh.receiverChain = append(sh.receiverChain, receiver)
 	}
 }
 
+type nilReceiver struct{}
+
+func (r *nilReceiver) OnClientMessageReceived(sender *Sender, messageType WebSocketMessageType, body io.Reader) {
+}
+func (r *nilReceiver) OnRoomCreated(region, name string)   {}
+func (r *nilReceiver) OnRoomDestroyed(region, name string) {}
+
+type chainReceiver struct {
+	chain []EventReceiver
+}
+
+func (r *chainReceiver) OnClientMessageReceived(sender *Sender, messageType WebSocketMessageType, body io.Reader) {
+	for _, cr := range r.chain {
+		cr.OnClientMessageReceived(sender, messageType, body)
+	}
+}
+
+func (r *chainReceiver) OnRoomCreated(region, name string) {
+	for _, cr := range r.chain {
+		cr.OnRoomCreated(region, name)
+	}
+}
+
+func (r *chainReceiver) OnRoomDestroyed(region, name string) {
+	for _, cr := range r.chain {
+		cr.OnRoomDestroyed(region, name)
+	}
+}
+
 func (ws *WebsocketHandler) Start(ctx context.Context) {
 	for _, sh := range ws.subhandlers {
 		chain := sh.receiverChain
 		if len(chain) == 0 {
-			sh.callReceiver = func(sender *Sender, messageType WebSocketMessageType, body io.Reader) {}
+			sh.callReceiver = &nilReceiver{}
 		} else if len(chain) == 1 {
 			sh.callReceiver = chain[0]
 		} else {
-			sh.callReceiver = func(sender *Sender, messageType WebSocketMessageType, body io.Reader) {
-				for _, r := range chain {
-					r(sender, messageType, body)
-				}
-			}
+			sh.callReceiver = &chainReceiver{chain: sh.receiverChain}
 		}
 
 		go sh.mainLoop(ctx)
@@ -322,6 +351,7 @@ func (sh *subhandler) mainLoop(ctx context.Context) {
 		select {
 		case destroyedRoom := <-roomDestroyChan:
 			delete(rooms, destroyedRoom)
+			go sh.callReceiver.OnRoomDestroyed(sh.region, destroyedRoom)
 
 		case usermsg := <-sh.localDeliveryChan:
 			// 로컬에 connection이 있는지 먼저 확인해 보기 위한 채널
@@ -444,10 +474,10 @@ func (sh *subhandler) mainLoop(ctx context.Context) {
 					room.out(c)
 				}
 				c.joinedRooms = nil
-				sh.callReceiver(c.sender, Disconnected, nil)
+				go sh.callReceiver.OnClientMessageReceived(c.sender, Disconnected, nil)
 			} else {
 				entireConns[c.sender.Accid.Hex()] = c
-				sh.callReceiver(c.sender, Connected, nil)
+				go sh.callReceiver.OnClientMessageReceived(c.sender, Connected, nil)
 			}
 		}
 	}
@@ -475,15 +505,15 @@ func upgrade_core(sh *subhandler, conn *websocket.Conn, accid primitive.ObjectID
 			}
 
 			if messageType == websocket.CloseMessage {
-				sh.callReceiver(c.sender, CloseMessage, r)
+				sh.callReceiver.OnClientMessageReceived(c.sender, CloseMessage, r)
 				break
 			}
 
 			if messageType == websocket.TextMessage {
 				// 유저가 직접 보낸 메시지
-				sh.callReceiver(c.sender, TextMessage, r)
+				sh.callReceiver.OnClientMessageReceived(c.sender, TextMessage, r)
 			} else if messageType == websocket.BinaryMessage {
-				sh.callReceiver(c.sender, BinaryMessage, r)
+				sh.callReceiver.OnClientMessageReceived(c.sender, BinaryMessage, r)
 			}
 		}
 		sh.redisSync.Del(context.Background(), accid.Hex())