🐛 refactor(web): 优化 WebSocket 连接的关闭逻辑

- 在 ChatWebSocket 方法中添加了 defer ws.Close(),确保在方法结束时关闭 WebSocket 连接
- 移除了冗余的 ws.Close() 调用,避免重复关闭连接
- 优化了代码结构,提高了代码的可读性和维护性
This commit is contained in:
Havoc412 2024-11-20 12:23:48 +08:00
parent 0ff364b677
commit 05383358ff

View File

@ -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.