|
|
|
|
@ -14,184 +14,185 @@
|
|
|
|
|
|
|
|
|
|
package relation
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"context"
|
|
|
|
|
|
|
|
|
|
"gorm.io/gorm"
|
|
|
|
|
|
|
|
|
|
"github.com/OpenIMSDK/protocol/constant"
|
|
|
|
|
"github.com/OpenIMSDK/tools/ormutil"
|
|
|
|
|
"github.com/OpenIMSDK/tools/utils"
|
|
|
|
|
|
|
|
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
var _ relation.GroupMemberModelInterface = (*GroupMemberGorm)(nil)
|
|
|
|
|
|
|
|
|
|
type GroupMemberGorm struct {
|
|
|
|
|
*MetaDB
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func NewGroupMemberDB(db *gorm.DB) relation.GroupMemberModelInterface {
|
|
|
|
|
return &GroupMemberGorm{NewMetaDB(db, &relation.GroupMemberModel{})}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) NewTx(tx any) relation.GroupMemberModelInterface {
|
|
|
|
|
return &GroupMemberGorm{NewMetaDB(tx.(*gorm.DB), &relation.GroupMemberModel{})}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) Create(ctx context.Context, groupMemberList []*relation.GroupMemberModel) (err error) {
|
|
|
|
|
return utils.Wrap(g.db(ctx).Create(&groupMemberList).Error, "")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) Delete(ctx context.Context, groupID string, userIDs []string) (err error) {
|
|
|
|
|
return utils.Wrap(
|
|
|
|
|
g.db(ctx).Where("group_id = ? and user_id in (?)", groupID, userIDs).Delete(&relation.GroupMemberModel{}).Error,
|
|
|
|
|
"",
|
|
|
|
|
)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) DeleteGroup(ctx context.Context, groupIDs []string) (err error) {
|
|
|
|
|
return utils.Wrap(g.db(ctx).Where("group_id in (?)", groupIDs).Delete(&relation.GroupMemberModel{}).Error, "")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) Update(ctx context.Context, groupID string, userID string, data map[string]any) (err error) {
|
|
|
|
|
return utils.Wrap(g.db(ctx).Where("group_id = ? and user_id = ?", groupID, userID).Updates(data).Error, "")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) UpdateRoleLevel(
|
|
|
|
|
ctx context.Context,
|
|
|
|
|
groupID string,
|
|
|
|
|
userID string,
|
|
|
|
|
roleLevel int32,
|
|
|
|
|
) (rowsAffected int64, err error) {
|
|
|
|
|
db := g.db(ctx).Where("group_id = ? and user_id = ?", groupID, userID).Updates(map[string]any{
|
|
|
|
|
"role_level": roleLevel,
|
|
|
|
|
})
|
|
|
|
|
return db.RowsAffected, utils.Wrap(db.Error, "")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) Find(
|
|
|
|
|
ctx context.Context,
|
|
|
|
|
groupIDs []string,
|
|
|
|
|
userIDs []string,
|
|
|
|
|
roleLevels []int32,
|
|
|
|
|
) (groupMembers []*relation.GroupMemberModel, err error) {
|
|
|
|
|
db := g.db(ctx)
|
|
|
|
|
if len(groupIDs) > 0 {
|
|
|
|
|
db = db.Where("group_id in (?)", groupIDs)
|
|
|
|
|
}
|
|
|
|
|
if len(userIDs) > 0 {
|
|
|
|
|
db = db.Where("user_id in (?)", userIDs)
|
|
|
|
|
}
|
|
|
|
|
if len(roleLevels) > 0 {
|
|
|
|
|
db = db.Where("role_level in (?)", roleLevels)
|
|
|
|
|
}
|
|
|
|
|
return groupMembers, utils.Wrap(db.Find(&groupMembers).Error, "")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) Take(
|
|
|
|
|
ctx context.Context,
|
|
|
|
|
groupID string,
|
|
|
|
|
userID string,
|
|
|
|
|
) (groupMember *relation.GroupMemberModel, err error) {
|
|
|
|
|
groupMember = &relation.GroupMemberModel{}
|
|
|
|
|
return groupMember, utils.Wrap(
|
|
|
|
|
g.db(ctx).Where("group_id = ? and user_id = ?", groupID, userID).Take(groupMember).Error,
|
|
|
|
|
"",
|
|
|
|
|
)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) TakeOwner(
|
|
|
|
|
ctx context.Context,
|
|
|
|
|
groupID string,
|
|
|
|
|
) (groupMember *relation.GroupMemberModel, err error) {
|
|
|
|
|
groupMember = &relation.GroupMemberModel{}
|
|
|
|
|
return groupMember, utils.Wrap(
|
|
|
|
|
g.db(ctx).Where("group_id = ? and role_level = ?", groupID, constant.GroupOwner).Take(groupMember).Error,
|
|
|
|
|
"",
|
|
|
|
|
)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) SearchMember(
|
|
|
|
|
ctx context.Context,
|
|
|
|
|
keyword string,
|
|
|
|
|
groupIDs []string,
|
|
|
|
|
userIDs []string,
|
|
|
|
|
roleLevels []int32,
|
|
|
|
|
pageNumber, showNumber int32,
|
|
|
|
|
) (total uint32, groupList []*relation.GroupMemberModel, err error) {
|
|
|
|
|
db := g.db(ctx)
|
|
|
|
|
ormutil.GormIn(&db, "group_id", groupIDs)
|
|
|
|
|
ormutil.GormIn(&db, "user_id", userIDs)
|
|
|
|
|
ormutil.GormIn(&db, "role_level", roleLevels)
|
|
|
|
|
return ormutil.GormSearch[relation.GroupMemberModel](db, []string{"nickname"}, keyword, pageNumber, showNumber)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) MapGroupMemberNum(
|
|
|
|
|
ctx context.Context,
|
|
|
|
|
groupIDs []string,
|
|
|
|
|
) (count map[string]uint32, err error) {
|
|
|
|
|
return ormutil.MapCount(g.db(ctx).Where("group_id in (?)", groupIDs), "group_id")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) FindJoinUserID(
|
|
|
|
|
ctx context.Context,
|
|
|
|
|
groupIDs []string,
|
|
|
|
|
) (groupUsers map[string][]string, err error) {
|
|
|
|
|
var groupMembers []*relation.GroupMemberModel
|
|
|
|
|
if err := g.db(ctx).Select("group_id, user_id").Where("group_id in (?)", groupIDs).Find(&groupMembers).Error; err != nil {
|
|
|
|
|
return nil, utils.Wrap(err, "")
|
|
|
|
|
}
|
|
|
|
|
groupUsers = make(map[string][]string)
|
|
|
|
|
for _, item := range groupMembers {
|
|
|
|
|
v, ok := groupUsers[item.GroupID]
|
|
|
|
|
if !ok {
|
|
|
|
|
groupUsers[item.GroupID] = []string{item.UserID}
|
|
|
|
|
} else {
|
|
|
|
|
groupUsers[item.GroupID] = append(v, item.UserID)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return groupUsers, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) FindMemberUserID(ctx context.Context, groupID string) (userIDs []string, err error) {
|
|
|
|
|
return userIDs, utils.Wrap(g.db(ctx).Where("group_id = ?", groupID).Pluck("user_id", &userIDs).Error, "")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) FindUserJoinedGroupID(ctx context.Context, userID string) (groupIDs []string, err error) {
|
|
|
|
|
return groupIDs, utils.Wrap(g.db(ctx).Where("user_id = ?", userID).Pluck("group_id", &groupIDs).Error, "")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) TakeGroupMemberNum(ctx context.Context, groupID string) (count int64, err error) {
|
|
|
|
|
return count, utils.Wrap(g.db(ctx).Where("group_id = ?", groupID).Count(&count).Error, "")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) FindUsersJoinedGroupID(ctx context.Context, userIDs []string) (map[string][]string, error) {
|
|
|
|
|
var groupMembers []*relation.GroupMemberModel
|
|
|
|
|
err := g.db(ctx).Select("group_id, user_id").Where("user_id IN (?)", userIDs).Find(&groupMembers).Error
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
result := make(map[string][]string)
|
|
|
|
|
for _, groupMember := range groupMembers {
|
|
|
|
|
v, ok := result[groupMember.UserID]
|
|
|
|
|
if !ok {
|
|
|
|
|
result[groupMember.UserID] = []string{groupMember.GroupID}
|
|
|
|
|
} else {
|
|
|
|
|
result[groupMember.UserID] = append(v, groupMember.GroupID)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return result, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (g *GroupMemberGorm) FindUserManagedGroupID(ctx context.Context, userID string) (groupIDs []string, err error) {
|
|
|
|
|
return groupIDs, utils.Wrap(
|
|
|
|
|
g.db(ctx).
|
|
|
|
|
Model(&relation.GroupMemberModel{}).
|
|
|
|
|
Where("user_id = ? and (role_level = ? or role_level = ?)", userID, constant.GroupOwner, constant.GroupAdmin).
|
|
|
|
|
Pluck("group_id", &groupIDs).
|
|
|
|
|
Error,
|
|
|
|
|
"",
|
|
|
|
|
)
|
|
|
|
|
}
|
|
|
|
|
//
|
|
|
|
|
//import (
|
|
|
|
|
// "context"
|
|
|
|
|
//
|
|
|
|
|
// "gorm.io/gorm"
|
|
|
|
|
//
|
|
|
|
|
// "github.com/OpenIMSDK/protocol/constant"
|
|
|
|
|
// "github.com/OpenIMSDK/tools/ormutil"
|
|
|
|
|
// "github.com/OpenIMSDK/tools/utils"
|
|
|
|
|
//
|
|
|
|
|
// "github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
|
|
|
//)
|
|
|
|
|
//
|
|
|
|
|
//var _ relation.GroupMemberModelInterface = (*GroupMemberGorm)(nil)
|
|
|
|
|
//
|
|
|
|
|
//type GroupMemberGorm struct {
|
|
|
|
|
// *MetaDB
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func NewGroupMemberDB(db *gorm.DB) relation.GroupMemberModelInterface {
|
|
|
|
|
// return &GroupMemberGorm{NewMetaDB(db, &relation.GroupMemberModel{})}
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) NewTx(tx any) relation.GroupMemberModelInterface {
|
|
|
|
|
// return &GroupMemberGorm{NewMetaDB(tx.(*gorm.DB), &relation.GroupMemberModel{})}
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) Create(ctx context.Context, groupMemberList []*relation.GroupMemberModel) (err error) {
|
|
|
|
|
// return utils.Wrap(g.db(ctx).Create(&groupMemberList).Error, "")
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) Delete(ctx context.Context, groupID string, userIDs []string) (err error) {
|
|
|
|
|
// return utils.Wrap(
|
|
|
|
|
// g.db(ctx).Where("group_id = ? and user_id in (?)", groupID, userIDs).Delete(&relation.GroupMemberModel{}).Error,
|
|
|
|
|
// "",
|
|
|
|
|
// )
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) DeleteGroup(ctx context.Context, groupIDs []string) (err error) {
|
|
|
|
|
// return utils.Wrap(g.db(ctx).Where("group_id in (?)", groupIDs).Delete(&relation.GroupMemberModel{}).Error, "")
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) Update(ctx context.Context, groupID string, userID string, data map[string]any) (err error) {
|
|
|
|
|
// return utils.Wrap(g.db(ctx).Where("group_id = ? and user_id = ?", groupID, userID).Updates(data).Error, "")
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) UpdateRoleLevel(
|
|
|
|
|
// ctx context.Context,
|
|
|
|
|
// groupID string,
|
|
|
|
|
// userID string,
|
|
|
|
|
// roleLevel int32,
|
|
|
|
|
//) (rowsAffected int64, err error) {
|
|
|
|
|
// db := g.db(ctx).Where("group_id = ? and user_id = ?", groupID, userID).Updates(map[string]any{
|
|
|
|
|
// "role_level": roleLevel,
|
|
|
|
|
// })
|
|
|
|
|
// return db.RowsAffected, utils.Wrap(db.Error, "")
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) Find(
|
|
|
|
|
// ctx context.Context,
|
|
|
|
|
// groupIDs []string,
|
|
|
|
|
// userIDs []string,
|
|
|
|
|
// roleLevels []int32,
|
|
|
|
|
//) (groupMembers []*relation.GroupMemberModel, err error) {
|
|
|
|
|
// db := g.db(ctx)
|
|
|
|
|
// if len(groupIDs) > 0 {
|
|
|
|
|
// db = db.Where("group_id in (?)", groupIDs)
|
|
|
|
|
// }
|
|
|
|
|
// if len(userIDs) > 0 {
|
|
|
|
|
// db = db.Where("user_id in (?)", userIDs)
|
|
|
|
|
// }
|
|
|
|
|
// if len(roleLevels) > 0 {
|
|
|
|
|
// db = db.Where("role_level in (?)", roleLevels)
|
|
|
|
|
// }
|
|
|
|
|
// return groupMembers, utils.Wrap(db.Find(&groupMembers).Error, "")
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) Take(
|
|
|
|
|
// ctx context.Context,
|
|
|
|
|
// groupID string,
|
|
|
|
|
// userID string,
|
|
|
|
|
//) (groupMember *relation.GroupMemberModel, err error) {
|
|
|
|
|
// groupMember = &relation.GroupMemberModel{}
|
|
|
|
|
// return groupMember, utils.Wrap(
|
|
|
|
|
// g.db(ctx).Where("group_id = ? and user_id = ?", groupID, userID).Take(groupMember).Error,
|
|
|
|
|
// "",
|
|
|
|
|
// )
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) TakeOwner(
|
|
|
|
|
// ctx context.Context,
|
|
|
|
|
// groupID string,
|
|
|
|
|
//) (groupMember *relation.GroupMemberModel, err error) {
|
|
|
|
|
// groupMember = &relation.GroupMemberModel{}
|
|
|
|
|
// return groupMember, utils.Wrap(
|
|
|
|
|
// g.db(ctx).Where("group_id = ? and role_level = ?", groupID, constant.GroupOwner).Take(groupMember).Error,
|
|
|
|
|
// "",
|
|
|
|
|
// )
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) SearchMember(
|
|
|
|
|
// ctx context.Context,
|
|
|
|
|
// keyword string,
|
|
|
|
|
// groupIDs []string,
|
|
|
|
|
// userIDs []string,
|
|
|
|
|
// roleLevels []int32,
|
|
|
|
|
// pageNumber, showNumber int32,
|
|
|
|
|
//) (total uint32, groupList []*relation.GroupMemberModel, err error) {
|
|
|
|
|
// db := g.db(ctx)
|
|
|
|
|
// ormutil.GormIn(&db, "group_id", groupIDs)
|
|
|
|
|
// ormutil.GormIn(&db, "user_id", userIDs)
|
|
|
|
|
// ormutil.GormIn(&db, "role_level", roleLevels)
|
|
|
|
|
// return ormutil.GormSearch[relation.GroupMemberModel](db, []string{"nickname"}, keyword, pageNumber, showNumber)
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) MapGroupMemberNum(
|
|
|
|
|
// ctx context.Context,
|
|
|
|
|
// groupIDs []string,
|
|
|
|
|
//) (count map[string]uint32, err error) {
|
|
|
|
|
// return ormutil.MapCount(g.db(ctx).Where("group_id in (?)", groupIDs), "group_id")
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) FindJoinUserID(
|
|
|
|
|
// ctx context.Context,
|
|
|
|
|
// groupIDs []string,
|
|
|
|
|
//) (groupUsers map[string][]string, err error) {
|
|
|
|
|
// var groupMembers []*relation.GroupMemberModel
|
|
|
|
|
// if err := g.db(ctx).Select("group_id, user_id").Where("group_id in (?)", groupIDs).Find(&groupMembers).Error; err != nil {
|
|
|
|
|
// return nil, utils.Wrap(err, "")
|
|
|
|
|
// }
|
|
|
|
|
// groupUsers = make(map[string][]string)
|
|
|
|
|
// for _, item := range groupMembers {
|
|
|
|
|
// v, ok := groupUsers[item.GroupID]
|
|
|
|
|
// if !ok {
|
|
|
|
|
// groupUsers[item.GroupID] = []string{item.UserID}
|
|
|
|
|
// } else {
|
|
|
|
|
// groupUsers[item.GroupID] = append(v, item.UserID)
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
// return groupUsers, nil
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) FindMemberUserID(ctx context.Context, groupID string) (userIDs []string, err error) {
|
|
|
|
|
// return userIDs, utils.Wrap(g.db(ctx).Where("group_id = ?", groupID).Pluck("user_id", &userIDs).Error, "")
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) FindUserJoinedGroupID(ctx context.Context, userID string) (groupIDs []string, err error) {
|
|
|
|
|
// return groupIDs, utils.Wrap(g.db(ctx).Where("user_id = ?", userID).Pluck("group_id", &groupIDs).Error, "")
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) TakeGroupMemberNum(ctx context.Context, groupID string) (count int64, err error) {
|
|
|
|
|
// return count, utils.Wrap(g.db(ctx).Where("group_id = ?", groupID).Count(&count).Error, "")
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) FindUsersJoinedGroupID(ctx context.Context, userIDs []string) (map[string][]string, error) {
|
|
|
|
|
// var groupMembers []*relation.GroupMemberModel
|
|
|
|
|
// err := g.db(ctx).Select("group_id, user_id").Where("user_id IN (?)", userIDs).Find(&groupMembers).Error
|
|
|
|
|
// if err != nil {
|
|
|
|
|
// return nil, err
|
|
|
|
|
// }
|
|
|
|
|
// result := make(map[string][]string)
|
|
|
|
|
// for _, groupMember := range groupMembers {
|
|
|
|
|
// v, ok := result[groupMember.UserID]
|
|
|
|
|
// if !ok {
|
|
|
|
|
// result[groupMember.UserID] = []string{groupMember.GroupID}
|
|
|
|
|
// } else {
|
|
|
|
|
// result[groupMember.UserID] = append(v, groupMember.GroupID)
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
// return result, nil
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (g *GroupMemberGorm) FindUserManagedGroupID(ctx context.Context, userID string) (groupIDs []string, err error) {
|
|
|
|
|
// return groupIDs, utils.Wrap(
|
|
|
|
|
// g.db(ctx).
|
|
|
|
|
// Model(&relation.GroupMemberModel{}).
|
|
|
|
|
// Where("user_id = ? and (role_level = ? or role_level = ?)", userID, constant.GroupOwner, constant.GroupAdmin).
|
|
|
|
|
// Pluck("group_id", &groupIDs).
|
|
|
|
|
// Error,
|
|
|
|
|
// "",
|
|
|
|
|
// )
|
|
|
|
|
//}
|
|
|
|
|
|
withchao
2 years ago