40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
package dao
|
|
|
|
import "gorm.io/gorm"
|
|
|
|
type AIStreamMsg struct {
|
|
gorm.Model
|
|
Type string `gorm:"column:type"`
|
|
ModelID uint `gorm:"column:model_id"`
|
|
UserID int `gorm:"column:user_id"` //用户id
|
|
SessionID int `gorm:"column:session_id"` //会话id
|
|
MSG string `gorm:"column:msg"` //消息内容
|
|
}
|
|
|
|
func InsertAIStreamMsgToDB(userID, sessionID int, modelID uint, msg, tp string) uint {
|
|
aiStreamMsg := AIStreamMsg{UserID: userID, ModelID: modelID, MSG: msg, Type: tp, SessionID: sessionID}
|
|
DB.Create(&aiStreamMsg)
|
|
if aiStreamMsg.ID == 0 {
|
|
return 0
|
|
}
|
|
return aiStreamMsg.ID
|
|
}
|
|
|
|
func FindAIStreamMsgByUserID(userID int) []AIStreamMsg {
|
|
var aiStreamMsgs []AIStreamMsg
|
|
DB.Where("user_id = ?", userID).Find(&aiStreamMsgs)
|
|
return aiStreamMsgs
|
|
}
|
|
|
|
func FindAIStreamMsgByModelID(modelID uint) []AIStreamMsg {
|
|
var aiStreamMsgs []AIStreamMsg
|
|
DB.Where("model_id = ?", modelID).Find(&aiStreamMsgs)
|
|
return aiStreamMsgs
|
|
}
|
|
|
|
func FindAIStreamMsgByID(id uint) (error, AIStreamMsg) {
|
|
var aiStreamMsg AIStreamMsg
|
|
err := DB.Where("id = ?", id).First(&aiStreamMsg).Error
|
|
return err, aiStreamMsg
|
|
}
|