From 9d9ee7c5852bbd4af18c2d04881afefd9db624b9 Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Tue, 15 Aug 2023 18:29:53 +0800
Subject: [PATCH] opt: unset timeout for websocket connection

---
 api/handler/chat_handler.go | 12 +++++++-----
 web/src/views/ChatPlus.vue  |  1 +
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/api/handler/chat_handler.go b/api/handler/chat_handler.go
index 2ad6427c..e8d3c18a 100644
--- a/api/handler/chat_handler.go
+++ b/api/handler/chat_handler.go
@@ -50,8 +50,8 @@ func (h *ChatHandler) ChatHandle(c *gin.Context) {
 		return
 	}
 	// 设置读写超时时间
-	_ = ws.SetWriteDeadline(time.Now().Add(300 * time.Second))
-	_ = ws.SetReadDeadline(time.Now().Add(300 * time.Second))
+	//_ = ws.SetWriteDeadline(time.Now().Add(300 * time.Second))
+	//_ = ws.SetReadDeadline(time.Now().Add(300 * time.Second))
 
 	sessionId := c.Query("session_id")
 	roleId := h.GetInt(c, "role_id", 0)
@@ -109,7 +109,7 @@ func (h *ChatHandler) ChatHandle(c *gin.Context) {
 	h.App.ChatClients.Put(sessionId, client)
 	go func() {
 		for {
-			_, message, err := client.Receive()
+			_, msg, err := client.Receive()
 			if err != nil {
 				logger.Error(err)
 				client.Close()
@@ -117,12 +117,14 @@ func (h *ChatHandler) ChatHandle(c *gin.Context) {
 				h.App.ReqCancelFunc.Delete(sessionId)
 				return
 			}
-			logger.Info("Receive a message: ", string(message))
+
+			message := string(msg)
+			logger.Info("Receive a message: ", message)
 			//utils.ReplyMessage(client, "这是一条测试消息！")
 			ctx, cancel := context.WithCancel(context.Background())
 			h.App.ReqCancelFunc.Put(sessionId, cancel)
 			// 回复消息
-			err = h.sendMessage(ctx, session, chatRole, string(message), client)
+			err = h.sendMessage(ctx, session, chatRole, message, client)
 			if err != nil {
 				logger.Error(err)
 			} else {
diff --git a/web/src/views/ChatPlus.vue b/web/src/views/ChatPlus.vue
index fd969427..02cac56a 100644
--- a/web/src/views/ChatPlus.vue
+++ b/web/src/views/ChatPlus.vue
@@ -534,6 +534,7 @@ const connect = function (chat_id, role_id) {
     } else { // 加载聊天记录
       loadChatHistory(chat_id);
     }
+    
   });
 
   _socket.addEventListener('message', event => {