fuzhongyun
/
ai_scheduler
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

结构修改

This commit is contained in:
renzhiyuan 2025-09-24 18:08:21 +08:00
parent 39b5dd2e35
commit 9699bbacbc
2 changed files with 5 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
internal/biz/router.go
View File

@ -142,20 +142,6 @@ func (r *AiRouterBiz) RouteWithSocket(c *websocket.Conn, req *entitys.ChatSockRe
}() }()
defer func() { defer func() {
if err != nil {
_ = entitys.MsgSend(c, entitys.Response{
Content: err.Error(),
Type: entitys.ResponseErr,
})
}
_ = entitys.MsgSend(c, entitys.Response{
Content: "",
Type: entitys.ResponseEnd,
})
//r.channelPool.Put(ch)
close(ch) close(ch)
}() }()

6
internal/services/chat.go
View File

@ -74,6 +74,7 @@ func (h *ChatService) Chat(c *websocket.Conn) {
h.Gw.AddClient(client) h.Gw.AddClient(client)
log.Println("client connected:", clientID) log.Println("client connected:", clientID)
log.Println("客户端已连接") log.Println("客户端已连接")
// 循环读取客户端消息 // 循环读取客户端消息
for { for {
messageType, message, err := c.ReadMessage() messageType, message, err := c.ReadMessage()
@ -108,8 +109,11 @@ func (h *ChatService) Chat(c *websocket.Conn) {
Content: err.Error(), Content: err.Error(),
Type: entitys.ResponseErr, Type: entitys.ResponseErr,
}) })
continue
} }
_ = entitys.MsgSend(c, entitys.Response{
Content: "",
Type: entitys.ResponseEnd,
})
} }
h.Gw.RemoveClient(clientID) h.Gw.RemoveClient(clientID)
_ = c.Close() _ = c.Close()