웹 소켓 접속 종료 시그니쳐 변경

wshandler/wshandler.go

@@ -29,7 +29,8 @@ var noAuthFlag = flagx.Bool("noauth", false, "")
 
 type wsconn struct {
 	*websocket.Conn
-	sender *Sender
+	sender       *Sender
+	closeMessage string
 }
 
 type UpstreamMessage struct {
@@ -455,11 +456,16 @@ func upgrade_core(ws *WebsocketHandler, conn *websocket.Conn, accid primitive.Ob
 		for {
 			messageType, r, err := c.NextReader()
 			if err != nil {
+				if ce, ok := err.(*websocket.CloseError); ok {
+					c.closeMessage = ce.Text
+				}
 				c.Close()
 				break
 			}
 
 			if messageType == websocket.CloseMessage {
+				closeMsg, _ := io.ReadAll(r)
+				logger.Println("close message :", string(closeMsg))
 				break
 			}