ljydlv
/
saw-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修改im接收消息存入数据库

This commit is contained in:
junleea 2025-04-11 16:35:36 +08:00
parent a97079747e
commit e0e7f10848
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
handler/im.go
View File

@ -348,7 +348,7 @@ func subscribeAndHandleKBaseTextMessages(ws *websocket.Conn, userId int, kBaseSe
log.Println("Error parsing JSON:", err1) log.Println("Error parsing JSON:", err1)
continue continue
} }
service.CreateAIStreamMsg(userId, data.IMContext.ModelID, data.IMContext.SessionID, message, proto.KnowledgeBaseServerResponseType) //将消息存入数据库 service.CreateAIStreamMsg(userId, data.IMContext.ModelID, data.IMContext.SessionID, data, proto.KnowledgeBaseServerResponseType) //将消息存入数据库
//这里主要将接收到的消息转到对应协程 //这里主要将接收到的消息转到对应协程
//目标chanel //目标chanel
targetChanel := fmt.Sprintf("kbase_%s", data.KnowledgeBase.UUID) targetChanel := fmt.Sprintf("kbase_%s", data.KnowledgeBase.UUID)

2
service/imService.go
View File

@ -330,6 +330,8 @@ func SelectKBaseContentByQuery(imCtx *proto.IMParamContext, data *proto.WSMessag
if err2 != nil { if err2 != nil {
log.Println("Error parsing JSON:", err2) log.Println("Error parsing JSON:", err2)
} }
//讲消息存入数据库
CreateAIStreamMsg()
break break
case <-timer.C: case <-timer.C:
log.Println("kbase query timeout") log.Println("kbase query timeout")