generated from msb_47094/GinChat
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
1.4 KiB
73 lines
1.4 KiB
/**
|
|
* @Auth:ShenZ
|
|
* @Description:
|
|
* @CreateDate:2022/06/15 14:57:55
|
|
*/
|
|
package models
|
|
|
|
import (
|
|
"fmt"
|
|
"ginchat/utils"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type Community struct {
|
|
gorm.Model
|
|
Name string
|
|
OwnerId uint
|
|
Img string
|
|
Desc string
|
|
}
|
|
|
|
func CreateCommunity(community Community) (int, string) {
|
|
tx := utils.DB.Begin()
|
|
//事务一旦开始，不论什么异常最终都会 Rollback
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
tx.Rollback()
|
|
}
|
|
}()
|
|
|
|
if len(community.Name) == 0 {
|
|
return -1, "群名称不能为空"
|
|
}
|
|
if community.OwnerId == 0 {
|
|
return -1, "请先登录"
|
|
}
|
|
if err := utils.DB.Create(&community).Error; err != nil {
|
|
fmt.Println(err)
|
|
tx.Rollback()
|
|
return -1, "建群失败"
|
|
}
|
|
contact := Contact{}
|
|
contact.OwnerId = community.OwnerId
|
|
contact.TargetId = community.ID
|
|
contact.Type = 2 //群关系
|
|
if err := utils.DB.Create(&contact).Error; err != nil {
|
|
tx.Rollback()
|
|
return -1, "添加群关系失败"
|
|
}
|
|
|
|
tx.Commit()
|
|
return 0, "建群成功"
|
|
|
|
}
|
|
|
|
func LoadCommunity(ownerId uint) ([]*Community, string) {
|
|
contacts := make([]Contact, 0)
|
|
objIds := make([]uint64, 0)
|
|
utils.DB.Where("owner_id = ? and type=2", ownerId).Find(&contacts)
|
|
for _, v := range contacts {
|
|
objIds = append(objIds, uint64(v.TargetId))
|
|
}
|
|
|
|
data := make([]*Community, 10)
|
|
utils.DB.Where("id in ?", objIds).Find(&data)
|
|
for _, v := range data {
|
|
fmt.Println(v)
|
|
}
|
|
//utils.DB.Where()
|
|
return data, "查询成功"
|
|
}
|