151 lines
3.9 KiB
Go
151 lines
3.9 KiB
Go
package dao
|
||
|
||
import (
|
||
"gorm.io/gorm"
|
||
"videoplayer/proto"
|
||
)
|
||
|
||
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"`
|
||
GroupInfo string `gorm:"column:group_info"`
|
||
GroupType int `gorm:"column:group_type"`
|
||
GroupIcon string `gorm:"column:group_icon"`
|
||
}
|
||
|
||
type GroupUser struct {
|
||
gorm.Model
|
||
GroupID int `gorm:"column:group_id"`
|
||
UserID int `gorm:"column:user_id"`
|
||
IsMaster bool `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: proto.MSG_TYPE_SIMPLE, Status: proto.MSG_STATUS_UNREAD}
|
||
res := DB.Debug().Create(msg)
|
||
return res.Error
|
||
}
|
||
|
||
func CreateGeneralMessage(from_user_id, to_user_id, msg_type, status int, message string) error {
|
||
msg := Message{FromUserID: from_user_id, ToUserID: to_user_id, Msg: message, Type: msg_type, Status: status}
|
||
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 AddFriend(user_id, friend_id int) error {
|
||
friend := Friend{UserID: user_id, FriendID: friend_id}
|
||
friend2 := Friend{UserID: friend_id, FriendID: user_id}
|
||
|
||
//开启事务
|
||
tx := DB.Begin()
|
||
if err := tx.Create(&friend); err.Error != nil {
|
||
tx.Rollback()
|
||
return err.Error
|
||
}
|
||
if err := tx.Create(&friend2); err.Error != nil {
|
||
tx.Rollback()
|
||
return err.Error
|
||
}
|
||
//提交事务
|
||
if err := tx.Commit().Error; err != nil {
|
||
return err
|
||
}
|
||
return error(nil)
|
||
}
|
||
|
||
// 删除好友,删除后双方都不是好友
|
||
func DeleteFriend(user_id, friend_id int) error {
|
||
//开启事务
|
||
tx := DB.Begin()
|
||
if err := tx.Delete(&Friend{}, "user_id = ? and friend_id = ?", user_id, friend_id); err.Error != nil {
|
||
tx.Rollback()
|
||
return err.Error
|
||
}
|
||
if err := tx.Delete(&Friend{}, "user_id = ? and friend_id = ?", friend_id, user_id); err.Error != nil {
|
||
tx.Rollback()
|
||
return err.Error
|
||
}
|
||
//提交事务
|
||
if err := tx.Commit().Error; err != nil {
|
||
return err
|
||
}
|
||
return error(nil)
|
||
}
|
||
|
||
// 通过id查找消息
|
||
func FindMessageByID(id uint) []Message {
|
||
var msgs []Message
|
||
DB.Debug().Where("id = ?", id).Find(msgs)
|
||
return msgs
|
||
}
|
||
|
||
// 更新消息状态
|
||
func UpdateMessageStatus(id uint, status int) error {
|
||
res := DB.Debug().Model(&Message{}).Where("id = ?", id).Update("status", status)
|
||
return res.Error
|
||
}
|
||
|
||
// 创建群聊,需要事务
|
||
func CreateGroup(group_name string, master_user_id int) uint {
|
||
group := Group{GroupName: group_name}
|
||
//开启事务
|
||
tx := DB.Begin()
|
||
if err := tx.Create(&group); err.Error != nil {
|
||
tx.Rollback()
|
||
return 0
|
||
}
|
||
groupUser := GroupUser{GroupID: int(group.ID), UserID: master_user_id, IsMaster: true}
|
||
if err := tx.Create(&groupUser); err.Error != nil {
|
||
tx.Rollback()
|
||
return 0
|
||
}
|
||
//提交事务
|
||
if err := tx.Commit().Error; err != nil {
|
||
return 0
|
||
}
|
||
return group.ID
|
||
}
|
||
|
||
// 加入群聊
|
||
func JoinGroup(group_id, user_id int) error {
|
||
groupUser := GroupUser{GroupID: group_id, UserID: user_id}
|
||
res := DB.Debug().Create(&groupUser)
|
||
return res.Error
|
||
}
|
||
|
||
// 退出群聊
|
||
func QuitGroup(group_id, user_id int) error {
|
||
res := DB.Debug().Delete(&GroupUser{}, "group_id = ? and user_id = ?", group_id, user_id)
|
||
return res.Error
|
||
}
|