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.