Open-IM-Server/pkg/common/db/relation/group_member_model_k.go

package relation

import (
	"Open_IM/pkg/common/constant"
	"Open_IM/pkg/common/db/table"
	"Open_IM/pkg/common/tracelog"
	"Open_IM/pkg/utils"
	"context"
	"gorm.io/gorm"
)

type GroupMemberGorm struct {
	DB *gorm.DB
}

func NewGroupMemberDB(db *gorm.DB) *GroupMemberGorm {
	return &GroupMemberGorm{DB: db}
}

func (g *GroupMemberGorm) Create(ctx context.Context, groupMemberList []*table.GroupMemberModel, tx ...*gorm.DB) (err error) {
	defer func() {
		tracelog.SetCtxDebug(ctx, utils.GetFuncName(1), err, "groupMemberList", groupMemberList)
	}()
	return utils.Wrap(getDBConn(g.DB, tx).Create(&groupMemberList).Error, "")
}

func (g *GroupMemberGorm) Delete(ctx context.Context, groupMembers []*table.GroupMemberModel, tx ...*gorm.DB) (err error) {
	defer func() {
		tracelog.SetCtxDebug(ctx, utils.GetFuncName(1), err, "groupMembers", groupMembers)
	}()
	return utils.Wrap(getDBConn(g.DB, tx).Delete(groupMembers).Error, "")
}

func (g *GroupMemberGorm) UpdateByMap(ctx context.Context, groupID string, userID string, args map[string]interface{}, tx ...*gorm.DB) (err error) {
	defer func() {
		tracelog.SetCtxDebug(ctx, utils.GetFuncName(1), err, "groupID", groupID, "userID", userID, "args", args)
	}()
	return utils.Wrap(getDBConn(g.DB, tx).Model(&table.GroupMemberModel{}).Where("group_id = ? and user_id = ?", groupID, userID).Updates(args).Error, "")
}

func (g *GroupMemberGorm) Update(ctx context.Context, groupMembers []*table.GroupMemberModel, tx ...*gorm.DB) (err error) {
	defer func() { tracelog.SetCtxDebug(ctx, utils.GetFuncName(1), err, "groupMembers", groupMembers) }()
	return utils.Wrap(getDBConn(g.DB, tx).Updates(&groupMembers).Error, "")
}

func (g *GroupMemberGorm) Find(ctx context.Context, groupMembers []*table.GroupMemberModel, tx ...*gorm.DB) (groupList []*table.GroupMemberModel, err error) {
	defer func() {
		tracelog.SetCtxDebug(ctx, utils.GetFuncName(1), err, "groupMembers", groupMembers, "groupList", groupList)
	}()
	var where [][]interface{}
	for _, groupMember := range groupMembers {
		where = append(where, []interface{}{groupMember.GroupID, groupMember.UserID})
	}
	return groupList, utils.Wrap(getDBConn(g.DB, tx).Where("(group_id, user_id) in ?", where).Find(&groupList).Error, "")
}

func (g *GroupMemberGorm) Take(ctx context.Context, groupID string, userID string, tx ...*gorm.DB) (groupMember *table.GroupMemberModel, err error) {
	defer func() {
		tracelog.SetCtxDebug(ctx, utils.GetFuncName(1), err, "groupID", groupID, "userID", userID, "groupMember", *groupMember)
	}()
	groupMember = &table.GroupMemberModel{}
	return groupMember, utils.Wrap(getDBConn(g.DB, tx).Where("group_id = ? and user_id = ?", groupID, userID).Take(groupMember).Error, "")
}

func (g *GroupMemberGorm) TakeOwnerInfo(ctx context.Context, groupID string, tx ...*gorm.DB) (groupMember *table.GroupMemberModel, err error) {
	defer func() {
		tracelog.SetCtxDebug(ctx, utils.GetFuncName(1), err, "groupID", groupID, "groupMember", *groupMember)
	}()
	groupMember = &table.GroupMemberModel{}
	return groupMember, utils.Wrap(getDBConn(g.DB, tx).Where("group_id = ? and role_level = ?", groupID, constant.GroupOwner).Take(groupMember).Error, "")
}

//func InsertIntoGroupMember(toInsertInfo GroupMemberModel) error {
//	toInsertInfo.JoinTime = time.Now()
//	if toInsertInfo.RoleLevel == 0 {
//		toInsertInfo.RoleLevel = constant.GroupOrdinaryUsers
//	}
//	toInsertInfo.MuteEndTime = time.Unix(int64(time.Now().Second()), 0)
//	err := GroupMemberDB.Table("group_members").Create(toInsertInfo).Error
//	if err != nil {
//		return err
//	}
//	return nil
//}
//
//func BatchInsertIntoGroupMember(toInsertInfoList []*GroupMemberModel) error {
//	for _, toInsertInfo := range toInsertInfoList {
//		toInsertInfo.JoinTime = time.Now()
//		if toInsertInfo.RoleLevel == 0 {
//			toInsertInfo.RoleLevel = constant.GroupOrdinaryUsers
//		}
//		toInsertInfo.MuteEndTime = time.Unix(int64(time.Now().Second()), 0)
//	}
//	return GroupMemberDB.Create(toInsertInfoList).Error
//
//}
//
//func GetGroupMemberListByUserID(userID string) ([]GroupMemberModel, error) {
//	var groupMemberList []GroupMemberModel
//	err := GroupMemberDB.Table("group_members").Where("user_id=?", userID).Find(&groupMemberList).Error
//	if err != nil {
//		return nil, err
//	}
//	return groupMemberList, nil
//}
//
//func GetGroupMemberListByGroupID(groupID string) ([]GroupMemberModel, error) {
//	var groupMemberList []GroupMemberModel
//	err := GroupMemberDB.Table("group_members").Where("group_id=?", groupID).Find(&groupMemberList).Error
//	if err != nil {
//		return nil, err
//	}
//	return groupMemberList, nil
//}
//
//func GetGroupMemberIDListByGroupID(groupID string) ([]string, error) {
//	var groupMemberIDList []string
//	err := GroupMemberDB.Table("group_members").Where("group_id=?", groupID).Pluck("user_id", &groupMemberIDList).Error
//	if err != nil {
//		return nil, err
//	}
//	return groupMemberIDList, nil
//}
//
//func GetGroupMemberListByGroupIDAndRoleLevel(groupID string, roleLevel int32) ([]GroupMemberModel, error) {
//	var groupMemberList []GroupMemberModel
//	err := GroupMemberDB.Table("group_members").Where("group_id=? and role_level=?", groupID, roleLevel).Find(&groupMemberList).Error
//	if err != nil {
//		return nil, err
//	}
//	return groupMemberList, nil
//}
//
//func GetGroupMemberInfoByGroupIDAndUserID(groupID, userID string) (*GroupMemberModel, error) {
//	var groupMember GroupMemberModel
//	err := GroupMemberDB.Table("group_members").Where("group_id=? and user_id=? ", groupID, userID).Limit(1).Take(&groupMember).Error
//	if err != nil {
//		return nil, err
//	}
//	return &groupMember, nil
//}
//
//func DeleteGroupMemberByGroupIDAndUserID(groupID, userID string) error {
//	return GroupMemberDB.Table("group_members").Where("group_id=? and user_id=? ", groupID, userID).Delete(GroupMemberModel{}).Error
//}
//
//func DeleteGroupMemberByGroupID(groupID string) error {
//	return GroupMemberDB.Table("group_members").Where("group_id=?  ", groupID).Delete(GroupMemberModel{}).Error
//}
//
//func UpdateGroupMemberInfo(groupMemberInfo GroupMemberModel) error {
//	return GroupMemberDB.Table("group_members").Where("group_id=? and user_id=?", groupMemberInfo.GroupID, groupMemberInfo.UserID).Updates(&groupMemberInfo).Error
//}
//
//func UpdateGroupMemberInfoByMap(groupMemberInfo GroupMemberModel, m map[string]interface{}) error {
//	return GroupMemberDB.Table("group_members").Where("group_id=? and user_id=?", groupMemberInfo.GroupID, groupMemberInfo.UserID).Updates(m).Error
//}
//
//func GetOwnerManagerByGroupID(groupID string) ([]GroupMemberModel, error) {
//	var groupMemberList []GroupMemberModel
//	err := GroupMemberDB.Table("group_members").Where("group_id=? and role_level>?", groupID, constant.GroupOrdinaryUsers).Find(&groupMemberList).Error
//	if err != nil {
//		return nil, err
//	}
//	return groupMemberList, nil
//}
//
//func GetGroupMemberNumByGroupID(groupID string) (int64, error) {
//	var number int64
//	err := GroupMemberDB.Table("group_members").Where("group_id=?", groupID).Count(&number).Error
//	if err != nil {
//		return 0, utils.Wrap(err, "")
//	}
//	return number, nil
//}
//
//func GetGroupOwnerInfoByGroupID(groupID string) (*GroupMemberModel, error) {
//	omList, err := GetOwnerManagerByGroupID(groupID)
//	if err != nil {
//		return nil, err
//	}
//	for _, v := range omList {
//		if v.RoleLevel == constant.GroupOwner {
//			return &v, nil
//		}
//	}
//	return nil, utils.Wrap(constant.ErrGroupNoOwner, "")
//}
//
//func IsExistGroupMember(groupID, userID string) bool {
//	var number int64
//	err := GroupMemberDB.Table("group_members").Where("group_id = ? and user_id = ?", groupID, userID).Count(&number).Error
//	if err != nil {
//		return false
//	}
//	if number != 1 {
//		return false
//	}
//	return true
//}
//
//func CheckIsExistGroupMember(ctx context.Context, groupID, userID string) error {
//	var number int64
//	err := GroupMemberDB.Table("group_members").Where("group_id = ? and user_id = ?", groupID, userID).Count(&number).Error
//	if err != nil {
//		return constant.ErrDB.Wrap()
//	}
//	if number != 1 {
//		return constant.ErrData.Wrap()
//	}
//	return nil
//}
//
//func GetGroupMemberByGroupID(groupID string, filter int32, begin int32, maxNumber int32) ([]GroupMember, error) {
//	var memberList []GroupMember
//	var err error
//	if filter >= 0 {
//		memberList, err = GetGroupMemberListByGroupIDAndRoleLevel(groupID, filter) //sorted by join time
//	} else {
//		memberList, err = GetGroupMemberListByGroupID(groupID)
//	}
//
//	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 GetJoinedGroupIDListByUserID(userID string) ([]string, error) {
//	memberList, err := GetGroupMemberListByUserID(userID)
//	if err != nil {
//		return nil, err
//	}
//	var groupIDList []string
//	for _, v := range memberList {
//		groupIDList = append(groupIDList, v.GroupID)
//	}
//	return groupIDList, nil
//}
//
//func IsGroupOwnerAdmin(groupID, UserID string) bool {
//	groupMemberList, err := GetOwnerManagerByGroupID(groupID)
//	if err != nil {
//		return false
//	}
//	for _, v := range groupMemberList {
//		if v.UserID == UserID && v.RoleLevel > constant.GroupOrdinaryUsers {
//			return true
//		}
//	}
//	return false
//}
//
//func GetGroupMembersByGroupIdCMS(groupId string, userName string, showNumber, pageNumber int32) ([]GroupMember, error) {
//	var groupMembers []GroupMember
//	err := GroupMemberDB.Table("group_members").Where("group_id=?", groupId).Where(fmt.Sprintf(" nickname like '%%%s%%' ", userName)).Limit(int(showNumber)).Offset(int(showNumber * (pageNumber - 1))).Find(&groupMembers).Error
//	if err != nil {
//		return nil, err
//	}
//	return groupMembers, nil
//}
//
//func GetGroupMembersCount(groupID, userName string) (int64, error) {
//	var count int64
//	if err := GroupMemberDB.Table("group_members").Where("group_id=?", groupID).Where(fmt.Sprintf(" nickname like '%%%s%%' ", userName)).Count(&count).Error; err != nil {
//		return count, err
//	}
//	return count, nil
//}
//
//func UpdateGroupMemberInfoDefaultZero(groupMemberInfo GroupMember, args map[string]interface{}) error {
//	return GroupMemberDB.Model(groupMemberInfo).Updates(args).Error
//}