🐛 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], "")
|
response.Fail(context, errcode.ErrWebsocketUpgradeFail, errcode.ErrMsg[errcode.ErrWebsocketUpgradeFail], "")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
defer ws.Close()
|
||||||
|
|
||||||
// 0-2. 测试 Python 微服务是否启动
|
// 0-2. 测试 Python 微服务是否启动
|
||||||
if !micro_service.TestLinkPythonService() {
|
if !micro_service.TestLinkPythonService() {
|
||||||
@ -219,7 +220,7 @@ func (r *Rag) ChatWebSocket(context *gin.Context) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
variable.ZapLog.Error("Failed to send token message via WebSocket", zap.Error(err))
|
variable.ZapLog.Error("Failed to send token message via WebSocket", zap.Error(err))
|
||||||
}
|
}
|
||||||
ws.Close()
|
// ws.Close() // 在上面调用了 defer;// TIP defer 的“栈”性质。
|
||||||
}()
|
}()
|
||||||
|
|
||||||
// 3.
|
// 3.
|
||||||
|
Loading…
x
Reference in New Issue
Block a user