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.
71 lines
2.2 KiB
71 lines
2.2 KiB
package group
|
|
|
|
import (
|
|
"Open_IM/pkg/common/constant"
|
|
"Open_IM/pkg/common/db/table/relation"
|
|
pbGroup "Open_IM/pkg/proto/group"
|
|
sdk_ws "Open_IM/pkg/proto/sdk_ws"
|
|
"Open_IM/pkg/utils"
|
|
"context"
|
|
)
|
|
|
|
func (s *groupServer) GetJoinedSuperGroupList(ctx context.Context, req *pbGroup.GetJoinedSuperGroupListReq) (*pbGroup.GetJoinedSuperGroupListResp, error) {
|
|
resp := &pbGroup.GetJoinedSuperGroupListResp{}
|
|
total, groupIDs, err := s.GroupInterface.FindJoinSuperGroup(ctx, req.UserID, req.Pagination.PageNumber, req.Pagination.ShowNumber)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
resp.Total = total
|
|
if len(groupIDs) == 0 {
|
|
return resp, nil
|
|
}
|
|
numMap, err := s.GroupInterface.MapSuperGroupMemberNum(ctx, groupIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
owners, err := s.GroupInterface.FindGroupMember(ctx, groupIDs, nil, []int32{constant.GroupOwner})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ownerMap := utils.SliceToMap(owners, func(e *relation.GroupMemberModel) string {
|
|
return e.GroupID
|
|
})
|
|
groups, err := s.GroupInterface.FindGroup(ctx, groupIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
groupMap := utils.SliceToMap(groups, func(e *relation.GroupModel) string {
|
|
return e.GroupID
|
|
})
|
|
resp.Groups = utils.Slice(groupIDs, func(groupID string) *sdk_ws.GroupInfo {
|
|
return DbToPbGroupInfo(groupMap[groupID], ownerMap[groupID].UserID, numMap[groupID])
|
|
})
|
|
return resp, nil
|
|
}
|
|
|
|
func (s *groupServer) GetSuperGroupsInfo(ctx context.Context, req *pbGroup.GetSuperGroupsInfoReq) (resp *pbGroup.GetSuperGroupsInfoResp, err error) {
|
|
resp = &pbGroup.GetSuperGroupsInfoResp{}
|
|
if len(req.GroupIDs) == 0 {
|
|
return nil, constant.ErrArgs.Wrap("groupIDs empty")
|
|
}
|
|
groups, err := s.GroupInterface.FindGroup(ctx, req.GroupIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
numMap, err := s.GroupInterface.MapSuperGroupMemberNum(ctx, req.GroupIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
owners, err := s.GroupInterface.FindGroupMember(ctx, req.GroupIDs, nil, []int32{constant.GroupOwner})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ownerMap := utils.SliceToMap(owners, func(e *relation.GroupMemberModel) string {
|
|
return e.GroupID
|
|
})
|
|
resp.GroupInfos = utils.Slice(groups, func(e *relation.GroupModel) *sdk_ws.GroupInfo {
|
|
return DbToPbGroupInfo(e, ownerMap[e.GroupID].UserID, numMap[e.GroupID])
|
|
})
|
|
return resp, nil
|
|
}
|