From 81e0c2a5bf40b5fa8a08906b2f9100288dc49719 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Mon, 28 Oct 2024 20:56:26 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=AE=9E=E6=97=B6=E6=9F=A5?=
 =?UTF-8?q?=E7=9C=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 handler/device.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/handler/device.go b/handler/device.go
index 27955c2..a0030aa 100644
--- a/handler/device.go
+++ b/handler/device.go
@@ -221,11 +221,11 @@ func GetRealTimeImage(c *gin.Context) {
 	ws, err := upgrader.Upgrade(c.Writer, c.Request, nil)
 	clients[ws] = true
 	if err != nil {
-		fmt.Println(err)
+		fmt.Println("connect wss err:", err)
 		return
 	}
-	defer ws.Close()
 	worker.SetRedisWithExpire(strconv.Itoa(int(device.ID))+"_is_play", "1", time.Minute*5)
+	fmt.Println("device_id:", device_id_int, " has set is_play to 1")
 	go subscribeAndHandleMessages(ws, device_id_int)
 }
 
@@ -255,11 +255,13 @@ func subscribeAndHandleMessages(ws *websocket.Conn, device_id int) {
 			msgObj.From_user_id = -1
 			res3, _ = json.Marshal(msgObj)
 		}
+		fmt.Println("send message to client length:", len(res3))
 		err2 := ws.WriteMessage(websocket.TextMessage, res3)
 		if err2 != nil {
 			clientsMux.Lock()
 			clients[ws] = false
 			clientsMux.Unlock()
+			ws.Close()
 			worker.SetRedisWithExpire(strconv.Itoa(device_id)+"_is_play", "0", time.Minute*5)
 			break
 		}