From 8ea2a698ce385fbe47e04e19aadf0158d580914f Mon Sep 17 00:00:00 2001
From: mountain <mountain@action2quare.com>
Date: Thu, 27 Jul 2023 21:55:28 +0900
Subject: [PATCH] =?UTF-8?q?=EC=A2=85=EB=A3=8C=EC=8B=9C=20=EC=BB=A8?=
 =?UTF-8?q?=EB=84=A5=EC=85=98=20=EC=A0=95=EB=A6=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 wshandler/wshandler.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/wshandler/wshandler.go b/wshandler/wshandler.go
index 8c57834..531072b 100644
--- a/wshandler/wshandler.go
+++ b/wshandler/wshandler.go
@@ -372,12 +372,26 @@ func (sh *subhandler) mainLoop(ctx context.Context) {
 		return room
 	}
 
+	defer func() {
+		for _, conn := range entireConns {
+			var roomnames []string
+			for _, room := range conn.joinedRooms {
+				roomnames = append(roomnames, room.name)
+			}
+			bt, _ := json.Marshal(roomnames)
+			sh.callReceiver.OnClientMessageReceived(conn.sender, Disconnected, bytes.NewBuffer(bt))
+		}
+	}()
+
 	// 유저에게서 온 메세지, 소켓 연결/해체 처리
 	for {
 		buffer := bytes.NewBuffer(make([]byte, 0, 1024))
 		buffer.Reset()
 
 		select {
+		case <-ctx.Done():
+			return
+
 		case destroyedRoom := <-roomDestroyChan:
 			delete(rooms, destroyedRoom)
 			go sh.callReceiver.OnRoomDestroyed(sh.region, destroyedRoom)