From 05383358fffb43761f2fca221ef9b349a867aa98 Mon Sep 17 00:00:00 2001
From: Havoc412 <2993167370@qq.com>
Date: Wed, 20 Nov 2024 12:23:48 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20refactor(web):=20=E4=BC=98?=
 =?UTF-8?q?=E5=8C=96=20WebSocket=20=E8=BF=9E=E6=8E=A5=E7=9A=84=E5=85=B3?=
 =?UTF-8?q?=E9=97=AD=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在 ChatWebSocket 方法中添加了 defer ws.Close()，确保在方法结束时关闭 WebSocket 连接
- 移除了冗余的 ws.Close() 调用，避免重复关闭连接
- 优化了代码结构，提高了代码的可读性和维护性
---
 app/http/controller/web/rag_controller.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/app/http/controller/web/rag_controller.go b/app/http/controller/web/rag_controller.go
index 5dad542..8fa20bc 100644
--- a/app/http/controller/web/rag_controller.go
+++ b/app/http/controller/web/rag_controller.go
@@ -156,6 +156,7 @@ func (r *Rag) ChatWebSocket(context *gin.Context) {
 		response.Fail(context, errcode.ErrWebsocketUpgradeFail, errcode.ErrMsg[errcode.ErrWebsocketUpgradeFail], "")
 		return
 	}
+	defer ws.Close()
 
 	// 0-2. 测试 Python 微服务是否启动
 	if !micro_service.TestLinkPythonService() {
@@ -219,7 +220,7 @@ func (r *Rag) ChatWebSocket(context *gin.Context) {
 		if err != nil {
 			variable.ZapLog.Error("Failed to send token message via WebSocket", zap.Error(err))
 		}
-		ws.Close()
+		// ws.Close()  // 在上面调用了 defer；// TIP defer 的“栈”性质。
 	}()
 
 	// 3.