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
}