From 0f3001967c37412fc08c3244b6520ec49088ff81 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Sat, 22 Mar 2025 17:42:57 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=8D=E5=9C=A8=E5=BA=94=E7=94=A8=E5=B1=82?=
 =?UTF-8?q?=E5=8F=91=E9=80=81=E5=BF=83=E8=B7=B3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 handler/im.go | 30 ++++++++++++++++--------------
 1 file changed, 16 insertions(+), 14 deletions(-)

diff --git a/handler/im.go b/handler/im.go
index bc634fd..f19b688 100644
--- a/handler/im.go
+++ b/handler/im.go
@@ -135,6 +135,7 @@ func subscribeAndHandleIMMessages(ws *websocket.Conn, userId int) {
 	}()
 
 	go func() {
+		defer close(done)
 		//从redis订阅消息
 		for m := range ch {
 			msg := m.Payload // 获取消息,消息格式为json
@@ -152,18 +153,19 @@ func subscribeAndHandleIMMessages(ws *websocket.Conn, userId int) {
 	}()
 
 	//发送心跳包
-	//for {
-	//	select {
-	//	case <-done:
-	//		//客户端断开连接
-	//		return
-	//	case t := <-ticker.C:
-	//		//发送心跳
-	//		err := ws.WriteMessage(websocket.PingMessage, nil)
-	//		if err != nil {
-	//			log.Println("Ping error:", err, t)
-	//			return
-	//		}
-	//	}
-	//}
+	for {
+		select {
+		case <-done:
+			//客户端断开连接
+			return
+			//case t := <-ticker.C:
+			//	//发送心跳
+			//	err := ws.WriteMessage(websocket.PingMessage, nil)
+			//	if err != nil {
+			//		log.Println("Ping error:", err, t)
+			//		return
+			//	}
+			//}
+		}
+	}
 }