From f1ec6a6805036e12b0d8de7e10ee8f2a4de877eb Mon Sep 17 00:00:00 2001 From: FitzLong <2605315944@qq.com> Date: Wed, 17 Nov 2021 11:26:36 +0800 Subject: [PATCH] Group members changed from mysql to mongo for management --- pkg/common/db/mongoModel.go | 298 ++++++++++++++++++++++++++++++++++++ 1 file changed, 298 insertions(+) diff --git a/pkg/common/db/mongoModel.go b/pkg/common/db/mongoModel.go index 93f64ff5f..9a084fd6d 100644 --- a/pkg/common/db/mongoModel.go +++ b/pkg/common/db/mongoModel.go @@ -15,6 +15,7 @@ import ( const cChat = "chat" const cGroup = "group" +const cGroupMemberModel = "groupMemberModel" const singleGocMsgNum = 5000 type MsgInfo struct { @@ -32,6 +33,15 @@ type GroupMember struct { UIDList []string } +type GroupMemberModel struct { + GroupId string `bson:"group_id"` + Uid string `bson:"uid"` + NickName string `bson:"nickname"` + AdministratorLevel int32 `bson:"administrator_level"` + JoinTime time.Time `bson:"join_time"` + UserGroupFaceUrl string `bson:"user_group_face_url"` +} + func (d *DataBases) GetMsgBySeqRange(uid string, seqBegin, seqEnd int64) (SingleMsg []*pbMsg.MsgFormat, GroupMsg []*pbMsg.MsgFormat, MaxSeq int64, MinSeq int64, err error) { var count int64 session := d.mgoSession.Clone() @@ -361,3 +371,291 @@ func isContainInt64(target int64, List []int64) bool { func indexGen(uid string, seqSuffix int64) string { return uid + ":" + strconv.FormatInt(seqSuffix, 10) } + +func (d *DataBases) InsertIntoGroupMember(groupId, uid, nickName, userGroupFaceUrl string, administratorLevel int32) error { + session := d.mgoSession.Clone() + if session == nil { + return errors.New("session == nil") + } + defer session.Close() + + c := session.DB(config.Config.Mongo.DBDatabase).C(cGroupMemberModel) + + n, err := c.Find(bson.M{"group_id": groupId, "uid": uid}).Count() + if err != nil { + return err + } + + if n == 0 { + + groupMsgInfo := GroupMemberModel{} + groupMsgInfo.GroupId = groupId + groupMsgInfo.Uid = uid + groupMsgInfo.NickName = nickName + groupMsgInfo.AdministratorLevel = administratorLevel + groupMsgInfo.JoinTime = time.Now() + groupMsgInfo.UserGroupFaceUrl = userGroupFaceUrl + + err = c.Insert(&groupMsgInfo) + if err != nil { + return err + } + } else { + err = c.Update(bson.M{"group_id": groupId, "uid": uid}, bson.M{"$set": bson.M{"nickname": nickName, + "administrator_level": administratorLevel, "join_time": time.Now(), "user_group_face_url": userGroupFaceUrl}}) + if err != nil { + return err + } + } + + return nil +} + +func (d *DataBases) FindGroupMemberListByUserId(uid string) ([]GroupMemberModel, error) { + session := d.mgoSession.Clone() + if session == nil { + return nil, errors.New("session == nil") + } + defer session.Close() + + c := session.DB(config.Config.Mongo.DBDatabase).C(cGroupMemberModel) + + var sGroupMemberModel []GroupMemberModel + if err := c.Find(bson.M{"uid": uid}).All(&sGroupMemberModel); err != nil { + return nil, err + } + + return sGroupMemberModel, nil + +} + +func (d *DataBases) FindGroupMemberListByGroupId(groupId string) ([]GroupMemberModel, error) { + session := d.mgoSession.Clone() + if session == nil { + return nil, errors.New("session == nil") + } + defer session.Close() + + c := session.DB(config.Config.Mongo.DBDatabase).C(cGroupMemberModel) + + var sGroupMemberModel []GroupMemberModel + if err := c.Find(bson.M{"group_id": groupId}).All(&sGroupMemberModel); err != nil { + return nil, err + } + + return sGroupMemberModel, nil +} + +func (d *DataBases) FindGroupMemberListByGroupIdAndFilterInfo(groupId string, filter int32) ([]GroupMemberModel, error) { + session := d.mgoSession.Clone() + if session == nil { + return nil, errors.New("session == nil") + } + defer session.Close() + + c := session.DB(config.Config.Mongo.DBDatabase).C(cGroupMemberModel) + + var sGroupMemberModel []GroupMemberModel + if err := c.Find(bson.M{"group_id": groupId, "administrator_level": filter}).All(&sGroupMemberModel); err != nil { + return nil, err + } + + return sGroupMemberModel, nil +} + +func (d *DataBases) FindGroupMemberInfoByGroupIdAndUserId(groupId, uid string) (*GroupMemberModel, error) { + session := d.mgoSession.Clone() + if session == nil { + return nil, errors.New("session == nil") + } + defer session.Close() + + c := session.DB(config.Config.Mongo.DBDatabase).C(cGroupMemberModel) + + sGroupMemberModel := GroupMemberModel{} + if err := c.Find(bson.M{"group_id": groupId, "uid": uid}).One(&sGroupMemberModel); err != nil { + return nil, err + } + + return &sGroupMemberModel, nil +} + +func (d *DataBases) DeleteGroupMemberByGroupIdAndUserId(groupId, uid string) error { + session := d.mgoSession.Clone() + if session == nil { + return errors.New("session == nil") + } + defer session.Close() + + c := session.DB(config.Config.Mongo.DBDatabase).C(cGroupMemberModel) + + if err := c.Remove(bson.M{"group_id": groupId, "uid": uid}); err != nil { + return err + } + + return nil +} + +func (d *DataBases) UpdateOwnerGroupNickName(groupId, userId, groupNickName string) error { + session := d.mgoSession.Clone() + if session == nil { + return errors.New("session == nil") + } + defer session.Close() + + c := session.DB(config.Config.Mongo.DBDatabase).C(cGroupMemberModel) + + if err := c.Update(bson.M{"group_id": groupId, "uid": userId}, bson.M{"$set": bson.M{"nickname": groupNickName}}); err != nil { + return err + } + + return nil +} + +func (d *DataBases) SelectGroupList(groupId string) ([]string, error) { + session := d.mgoSession.Clone() + if session == nil { + return nil, errors.New("session == nil") + } + defer session.Close() + + c := session.DB(config.Config.Mongo.DBDatabase).C(cGroupMemberModel) + + var sGroupMemberModel []GroupMemberModel + if err := c.Find(bson.M{"group_id": groupId}).All(&sGroupMemberModel); err != nil { + return nil, err + } + + var groupUserID string + var groupList []string + for _, GroupMemberModel := range sGroupMemberModel { + groupUserID = GroupMemberModel.Uid + groupList = append(groupList, groupUserID) + } + + return groupList, nil +} + +func (d *DataBases) UpdateTheUserAdministratorLevel(groupId, uid string, administratorLevel int64) error { + session := d.mgoSession.Clone() + if session == nil { + return errors.New("session == nil") + } + defer session.Close() + + c := session.DB(config.Config.Mongo.DBDatabase).C(cGroupMemberModel) + + if err := c.Update(bson.M{"group_id": groupId, "uid": uid}, bson.M{"$set": bson.M{"administrator_level": administratorLevel}}); err != nil { + return err + } + + return nil +} + +func (d *DataBases) GetOwnerManagerByGroupId(groupId string) ([]GroupMemberModel, error) { + session := d.mgoSession.Clone() + if session == nil { + return nil, errors.New("session == nil") + } + defer session.Close() + + c := session.DB(config.Config.Mongo.DBDatabase).C(cGroupMemberModel) + + var sGroupMemberModel []GroupMemberModel + if err := c.Find(bson.M{"group_id": groupId, "administrator_level": bson.M{"$gt": 0}}).All(&sGroupMemberModel); err != nil { + return nil, err + } + + return sGroupMemberModel, nil + +} + +func (d *DataBases) IsExistGroupMember(groupId, uid string) bool { + session := d.mgoSession.Clone() + if session == nil { + return false + } + defer session.Close() + + c := session.DB(config.Config.Mongo.DBDatabase).C(cGroupMemberModel) + + number, err := c.Find(bson.M{"group_id": groupId, "uid": uid}).Count() + if err != nil { + return false + } + + if number != 1 { + return false + } + + return true +} + +func (d *DataBases) RemoveGroupMember(groupId string, memberId string) error { + return d.DeleteGroupMemberByGroupIdAndUserId(groupId, memberId) +} + +func (d *DataBases) GetMemberInfoById(groupId string, memberId string) (*GroupMemberModel, error) { + return d.FindGroupMemberInfoByGroupIdAndUserId(groupId, memberId) +} + +func (d *DataBases) GetGroupMemberByGroupId(groupId string, filter int32, begin int32, maxNumber int32) ([]GroupMemberModel, error) { + memberList, err := d.FindGroupMemberListByGroupId(groupId) //sorted by join time + if err != nil { + return nil, err + } + if begin >= int32(len(memberList)) { + return nil, nil + } + + var end int32 + if begin+int32(maxNumber) < int32(len(memberList)) { + end = begin + maxNumber + } else { + end = int32(len(memberList)) + } + return memberList[begin:end], nil +} + +func (d *DataBases) GetJoinedGroupIdListByMemberId(memberId string) ([]GroupMemberModel, error) { + return d.FindGroupMemberListByUserId(memberId) +} + +func (d *DataBases) GetGroupMemberNumByGroupId(groupId string) int32 { + session := d.mgoSession.Clone() + if session == nil { + return 0 + } + defer session.Close() + + c := session.DB(config.Config.Mongo.DBDatabase).C(cGroupMemberModel) + + var number int + number, err := c.Find(bson.M{"group_id": groupId}).Count() + if err != nil { + return 0 + } + + if number != 1 { + return 0 + } + + return int32(number) +} + +func (d *DataBases) GetGroupOwnerByGroupId(groupId string) string { + omList, err := d.GetOwnerManagerByGroupId(groupId) + if err != nil { + return "" + } + for _, v := range omList { + if v.AdministratorLevel == 1 { + return v.Uid + } + } + return "" +} + +func (d *DataBases) InsertGroupMember(groupId, userId, nickName, userFaceUrl string, role int32) error { + return d.InsertIntoGroupMember(groupId, userId, nickName, userFaceUrl, role) +}