57 lines
1.5 KiB
Go
57 lines
1.5 KiB
Go
|
package dao
|
|||
|
|
|
|||
|
|
import "gorm.io/gorm"
|
|||
|
|
|
|||
|
|
type Message struct {
|
|||
|
|
gorm.Model
|
|||
|
|
FromUserID int `gorm:"column:from_user_id"`
|
|||
|
|
ToUserID int `gorm:"column:to_user_id"`
|
|||
|
|
GroupID int `gorm:"column:group_id"`
|
|||
|
|
Msg string `gorm:"column:msg"`
|
|||
|
|
Status int `gorm:"column:status"` //单聊时才有0,已读，2未读
|
|||
|
|
Type int `gorm:"column:type"` //1为单聊，2为群聊，3为系统消息
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Group struct {
|
|||
|
|
gorm.Model
|
|||
|
|
GroupName string `gorm:"column:group_name"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type GroupUser struct {
|
|||
|
|
gorm.Model
|
|||
|
|
GroupID int `gorm:"column:group_id"`
|
|||
|
|
UserID int `gorm:"column:user_id"`
|
|||
|
|
IsMaster int `gorm:"column:is_master"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Friend struct {
|
|||
|
|
gorm.Model
|
|||
|
|
UserID int `gorm:"column:user_id"`
|
|||
|
|
FriendID int `gorm:"column:friend_id"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建单聊消息
|
|||
|
|
func CreateSimpleMessage(from_user_id, to_user_id int, message string) error {
|
|||
|
|
msg := Message{FromUserID: from_user_id, ToUserID: to_user_id, Msg: message, Type: 1, Status: 0}
|
|||
|
|
res := DB.Debug().Create(msg)
|
|||
|
|
return res.Error
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 每20个消息一组，请求index*20
|
|||
|
|
func GetMsgUserByIndex(from_user_id, to_user_id, index int) ([]Message, error) {
|
|||
|
|
var msgs []Message
|
|||
|
|
res := DB.Debug().Where("from_user_id = ? and to_user_id = ?", from_user_id, to_user_id).Find(&msgs).Order("order by createAt DESC").Limit(20 * index)
|
|||
|
|
return msgs, res.Error
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 修改信息
|
|||
|
|
func UpdateMessage() {
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func FindMessageByID(id uint) []Message {
|
|||
|
|
var msgs []Message
|
|||
|
|
DB.Debug().Where("id = ?", id).Find(msgs)
|
|||
|
|
return msgs
|
|||
|
|
}
|