xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-07 01:33:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed bug for websocket close tip message

Browse Source
This commit is contained in:
RockYang
2024-04-15 18:15:15 +08:00
parent 3a8a69ac2e
commit cd562ab8ed
6 changed files with 3 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/handler/markmap_handler.go
View File

@@ -42,14 +42,8 @@ func (h *MarkMapHandler) Client(c *gin.Context) {
modelId := h.GetInt(c, "model_id", 0)
userId := h.GetInt(c, "user_id", 0)
logger.Info(modelId)
client := types.NewWsClient(ws)
if cli := h.clients.Get(userId); cli != nil {
cli.Close()
}
// 保存会话连接
h.clients.Put(userId, client)
go func() {
for {

4
api/handler/mj_handler.go
View File

@@ -78,10 +78,6 @@ func (h *MidJourneyHandler) Client(c *gin.Context) {
}
client := types.NewWsClient(ws)
// close the existed connections
if cli := h.pool.Clients.Get(uint(userId)); cli != nil {
cli.Close()
}
h.pool.Clients.Put(uint(userId), client)
logger.Infof("New websocket connected, IP: %s", c.RemoteIP())
}

4
api/handler/sd_handler.go
View File

@@ -61,10 +61,6 @@ func (h *SdJobHandler) Client(c *gin.Context) {
}
client := types.NewWsClient(ws)
// close the existed connections
if cli := h.pool.Clients.Get(uint(userId)); cli != nil {
cli.Close()
}
h.pool.Clients.Put(uint(userId), client)
logger.Infof("New websocket connected, IP: %s", c.RemoteIP())
}