🐛 refactor(web): 优化 WebSocket 连接的关闭逻辑
- 在 ChatWebSocket 方法中添加了 defer ws.Close(),确保在方法结束时关闭 WebSocket 连接 - 移除了冗余的 ws.Close() 调用,避免重复关闭连接 - 优化了代码结构,提高了代码的可读性和维护性
This commit is contained in:
parent
0ff364b677
commit
05383358ff
@ -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.
|
||||
|
Loading…
x
Reference in New Issue
Block a user