Files
open-im-server/internal/rpc/group/group.go
T

1693 lines
63 KiB
Go
Raw Normal View History

2021-06-28 15:36:09 +08:00
package group
import (
2023-01-05 11:33:17 +08:00
"Open_IM/internal/rpc/fault_tolerant"
2021-12-23 18:29:39 +08:00
chat "Open_IM/internal/rpc/msg"
"Open_IM/pkg/common/config"
"Open_IM/pkg/common/constant"
"Open_IM/pkg/common/db"
imdb "Open_IM/pkg/common/db/mysql_model/im_mysql_model"
2022-08-30 01:38:23 +08:00
rocksCache "Open_IM/pkg/common/db/rocks_cache"
"Open_IM/pkg/common/log"
2022-09-15 01:22:20 +08:00
promePkg "Open_IM/pkg/common/prometheus"
2021-12-22 15:24:49 +08:00
"Open_IM/pkg/common/token_verify"
2023-01-04 11:06:28 +08:00
"Open_IM/pkg/common/trace_log"
2021-12-30 10:56:17 +08:00
cp "Open_IM/pkg/common/utils"
2023-01-04 10:12:20 +08:00
open_im_sdk "Open_IM/pkg/proto/sdk_ws"
"github.com/OpenIMSDK/getcdv3"
2022-05-07 11:14:29 +08:00
pbCache "Open_IM/pkg/proto/cache"
2022-06-07 17:45:07 +08:00
pbConversation "Open_IM/pkg/proto/conversation"
2021-10-11 22:00:38 +08:00
pbGroup "Open_IM/pkg/proto/group"
2022-04-20 18:47:21 +08:00
pbUser "Open_IM/pkg/proto/user"
"Open_IM/pkg/utils"
2021-06-28 15:36:09 +08:00
"context"
2022-08-30 14:17:40 +08:00
"errors"
2022-06-10 13:32:40 +08:00
"math/big"
2021-12-22 15:24:49 +08:00
"net"
"strconv"
"strings"
"time"
2022-08-30 01:38:23 +08:00
2022-09-15 16:27:36 +08:00
grpcPrometheus "github.com/grpc-ecosystem/go-grpc-prometheus"
2022-08-30 01:38:23 +08:00
"google.golang.org/grpc"
2022-12-16 16:54:18 +08:00
"google.golang.org/protobuf/types/known/wrapperspb"
2022-08-30 14:17:40 +08:00
"gorm.io/gorm"
2021-06-28 15:36:09 +08:00
)
2021-12-22 15:24:49 +08:00
type groupServer struct {
rpcPort int
rpcRegisterName string
etcdSchema string
etcdAddr []string
}
func NewGroupServer(port int) *groupServer {
2022-03-31 11:15:06 +08:00
log.NewPrivateLog(constant.LogFileName)
2021-12-22 15:24:49 +08:00
return &groupServer{
rpcPort: port,
rpcRegisterName: config.Config.RpcRegisterName.OpenImGroupName,
etcdSchema: config.Config.Etcd.EtcdSchema,
etcdAddr: config.Config.Etcd.EtcdAddr,
}
}
2021-12-24 15:02:47 +08:00
2021-12-22 15:24:49 +08:00
func (s *groupServer) Run() {
2022-03-29 11:48:17 +08:00
log.NewInfo("", "group rpc start ")
2022-05-07 17:05:05 +08:00
listenIP := ""
if config.Config.ListenIP == "" {
listenIP = "0.0.0.0"
} else {
listenIP = config.Config.ListenIP
}
address := listenIP + ":" + strconv.Itoa(s.rpcPort)
2021-12-22 15:24:49 +08:00
//listener network
2022-05-07 17:05:05 +08:00
listener, err := net.Listen("tcp", address)
2021-12-22 15:24:49 +08:00
if err != nil {
2022-05-10 09:09:37 +08:00
panic("listening err:" + err.Error() + s.rpcRegisterName)
2021-12-22 15:24:49 +08:00
}
2022-05-07 17:05:05 +08:00
log.NewInfo("", "listen network success, ", address, listener)
2021-12-22 15:24:49 +08:00
defer listener.Close()
//grpc server
2022-08-01 17:44:40 +08:00
recvSize := 1024 * 1024 * constant.GroupRPCRecvSize
sendSize := 1024 * 1024 * constant.GroupRPCSendSize
2022-09-15 19:55:24 +08:00
var grpcOpts = []grpc.ServerOption{
2022-08-01 11:52:43 +08:00
grpc.MaxRecvMsgSize(recvSize),
grpc.MaxSendMsgSize(sendSize),
}
2022-09-15 01:22:20 +08:00
if config.Config.Prometheus.Enable {
2022-09-15 08:45:10 +08:00
promePkg.NewGrpcRequestCounter()
promePkg.NewGrpcRequestFailedCounter()
promePkg.NewGrpcRequestSuccessCounter()
2022-09-15 16:27:36 +08:00
grpcOpts = append(grpcOpts, []grpc.ServerOption{
2022-09-15 16:39:49 +08:00
// grpc.UnaryInterceptor(promePkg.UnaryServerInterceptorProme),
2022-09-15 16:27:36 +08:00
grpc.StreamInterceptor(grpcPrometheus.StreamServerInterceptor),
grpc.UnaryInterceptor(grpcPrometheus.UnaryServerInterceptor),
}...)
2022-09-15 01:22:20 +08:00
}
2022-09-15 19:55:24 +08:00
srv := grpc.NewServer(grpcOpts...)
2021-12-22 15:24:49 +08:00
defer srv.GracefulStop()
//Service registers with etcd
pbGroup.RegisterGroupServer(srv, s)
2022-05-07 17:05:05 +08:00
2022-05-11 12:20:28 +08:00
rpcRegisterIP := config.Config.RpcRegisterIP
2022-06-23 09:24:05 +08:00
if config.Config.RpcRegisterIP == "" {
2022-05-07 17:05:05 +08:00
rpcRegisterIP, err = utils.GetLocalIP()
if err != nil {
log.Error("", "GetLocalIP failed ", err.Error())
}
}
2022-05-29 19:44:22 +08:00
log.NewInfo("", "rpcRegisterIP", rpcRegisterIP)
2023-01-04 15:51:02 +08:00
err = getcdv3.RegisterEtcd(s.etcdSchema, strings.Join(s.etcdAddr, ","), rpcRegisterIP, s.rpcPort, s.rpcRegisterName, 10, "")
2021-12-22 15:24:49 +08:00
if err != nil {
2022-03-29 11:48:17 +08:00
log.NewError("", "RegisterEtcd failed ", err.Error())
2022-08-26 17:41:58 +08:00
panic(utils.Wrap(err, "register group module rpc to etcd err"))
2021-12-22 15:24:49 +08:00
}
2022-05-11 12:13:17 +08:00
log.Info("", "RegisterEtcd ", s.etcdSchema, strings.Join(s.etcdAddr, ","), rpcRegisterIP, s.rpcPort, s.rpcRegisterName)
2021-12-22 15:24:49 +08:00
err = srv.Serve(listener)
if err != nil {
2022-03-29 11:48:17 +08:00
log.NewError("", "Serve failed ", err.Error())
2021-12-22 15:24:49 +08:00
return
}
2022-03-29 11:48:17 +08:00
log.NewInfo("", "group rpc success")
2021-12-22 15:24:49 +08:00
}
2023-01-06 14:11:12 +08:00
func (s *groupServer) CreateGroup(ctx context.Context, req *pbGroup.CreateGroupReq) (resp *pbGroup.CreateGroupResp, _ error) {
2023-01-06 14:06:18 +08:00
resp = &pbGroup.CreateGroupResp{CommonResp: &open_im_sdk.CommonResp{}, GroupInfo: &open_im_sdk.GroupInfo{}}
2023-01-06 11:19:31 +08:00
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
2023-01-06 14:19:03 +08:00
defer func() {
2023-01-06 14:21:00 +08:00
trace_log.SetContextInfo(ctx, utils.GetFuncName(1), nil, "req", req.String(), "resp", resp.String())
2023-01-06 14:19:03 +08:00
trace_log.ShowLog(ctx)
}()
2023-01-06 11:49:11 +08:00
if err := token_verify.CheckAccessV2(ctx, req.OpUserID, req.OwnerUserID); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 11:19:31 +08:00
return
2022-10-21 15:44:55 +08:00
}
2023-01-06 16:19:57 +08:00
var groupOwnerNum int
var userIDs []string
for _, info := range req.InitMemberList {
if info.RoleLevel == constant.GroupOwner {
groupOwnerNum++
}
userIDs = append(userIDs, info.UserID)
}
if req.OwnerUserID != "" {
groupOwnerNum++
userIDs = append(userIDs, req.OwnerUserID)
}
if groupOwnerNum != 1 {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrArgs, resp.CommonResp)
2023-01-06 16:19:57 +08:00
return
}
if utils.IsRepeatStringSlice(userIDs) {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrArgs, resp.CommonResp)
2023-01-06 16:19:57 +08:00
return
}
2023-01-09 14:27:08 +08:00
users, err := rocksCache.GetUserInfoFromCacheBatch(ctx, userIDs)
2023-01-06 16:19:57 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 16:19:57 +08:00
return
}
if len(users) != len(userIDs) {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrArgs, resp.CommonResp)
2023-01-06 16:19:57 +08:00
return
}
userMap := make(map[string]*imdb.User)
for i, user := range users {
userMap[user.UserID] = users[i]
}
2023-01-09 16:37:33 +08:00
if err := s.DelGroupAndUserCache(ctx, "", userIDs); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 16:19:57 +08:00
return
}
2023-01-06 11:19:31 +08:00
if err := callbackBeforeCreateGroup(ctx, req); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 11:19:31 +08:00
return
2022-10-21 15:44:55 +08:00
}
groupId := req.GroupInfo.GroupID
if groupId == "" {
2022-06-10 13:32:40 +08:00
groupId = utils.Md5(req.OperationID + strconv.FormatInt(time.Now().UnixNano(), 10))
bi := big.NewInt(0)
bi.SetString(groupId[0:8], 16)
groupId = bi.String()
}
2023-01-04 17:22:55 +08:00
groupInfo := imdb.Group{}
2021-12-27 21:08:42 +08:00
utils.CopyStructFields(&groupInfo, req.GroupInfo)
2021-12-28 11:57:12 +08:00
groupInfo.CreatorUserID = req.OpUserID
2022-01-18 21:42:27 +08:00
groupInfo.GroupID = groupId
2022-09-13 15:27:00 +08:00
groupInfo.CreateTime = time.Now()
2022-07-14 13:44:36 +08:00
if groupInfo.NotificationUpdateTime.Unix() < 0 {
groupInfo.NotificationUpdateTime = utils.UnixSecondToTime(0)
}
2022-05-27 18:40:24 +08:00
if req.GroupInfo.GroupType != constant.SuperGroup {
2023-01-06 11:19:31 +08:00
var groupMembers []*imdb.GroupMember
2023-01-06 16:19:57 +08:00
joinGroup := func(userID string, roleLevel int32) error {
groupMember := &imdb.GroupMember{GroupID: groupId, RoleLevel: roleLevel, OperatorUserID: req.OpUserID, JoinSource: constant.JoinByInvitation, InviterUserID: req.OpUserID}
user := userMap[userID]
utils.CopyStructFields(&groupMember, user)
if err := CallbackBeforeMemberJoinGroup(ctx, req.OperationID, groupMember, groupInfo.Ex); err != nil {
return err
2022-05-27 16:13:44 +08:00
}
2023-01-06 16:19:57 +08:00
groupMembers = append(groupMembers, groupMember)
return nil
}
if req.OwnerUserID == "" {
if err := joinGroup(req.OwnerUserID, constant.GroupOwner); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 16:19:57 +08:00
return
2022-05-27 16:13:44 +08:00
}
2023-01-06 16:19:57 +08:00
}
for _, info := range req.InitMemberList {
if err := joinGroup(info.UserID, info.RoleLevel); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 11:19:31 +08:00
return
2022-05-27 16:13:44 +08:00
}
}
2023-01-06 11:19:31 +08:00
if err := (*imdb.GroupMember)(nil).Create(ctx, groupMembers); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 11:19:31 +08:00
return
}
2022-05-27 18:40:24 +08:00
} else {
2023-01-06 16:19:57 +08:00
if err := db.DB.CreateSuperGroup(groupId, userIDs, len(userIDs)); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 11:19:31 +08:00
return
}
2022-07-20 18:12:39 +08:00
}
2023-01-06 16:19:57 +08:00
if err := (*imdb.Group)(nil).Create(ctx, []*imdb.Group{&groupInfo}); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 11:19:31 +08:00
return
2023-01-06 16:19:57 +08:00
}
utils.CopyStructFields(resp.GroupInfo, groupInfo)
resp.GroupInfo.MemberCount = uint32(len(userIDs))
if req.GroupInfo.GroupType != constant.SuperGroup {
chat.GroupCreatedNotification(req.OperationID, req.OpUserID, groupId, userIDs)
} else {
2023-01-06 16:19:57 +08:00
for _, userID := range userIDs {
2023-01-09 16:37:33 +08:00
if err := rocksCache.DelJoinedSuperGroupIDListFromCache(ctx, userID); err != nil {
2023-01-06 16:19:57 +08:00
trace_log.SetContextInfo(ctx, "DelJoinedSuperGroupIDListFromCache", err, "userID", userID)
}
}
go func() {
for _, v := range userIDs {
chat.SuperGroupNotification(req.OperationID, v, v)
}
}()
}
2023-01-06 16:19:57 +08:00
return
2021-12-22 15:24:49 +08:00
}
2023-01-06 16:19:57 +08:00
func (s *groupServer) GetJoinedGroupList(ctx context.Context, req *pbGroup.GetJoinedGroupListReq) (resp *pbGroup.GetJoinedGroupListResp, _ error) {
2023-01-06 14:06:18 +08:00
resp = &pbGroup.GetJoinedGroupListResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
2023-01-06 17:00:14 +08:00
defer func() {
2023-01-06 16:31:40 +08:00
trace_log.SetContextInfo(ctx, utils.GetFuncName(1), nil, "req", req.String(), "resp", resp.String())
trace_log.ShowLog(ctx)
}()
2023-01-06 16:19:57 +08:00
if err := token_verify.CheckAccessV2(ctx, req.OpUserID, req.FromUserID); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 14:06:18 +08:00
return
2021-06-28 15:36:09 +08:00
}
2023-01-09 14:27:08 +08:00
joinedGroupList, err := rocksCache.GetJoinedGroupIDListFromCache(ctx, req.FromUserID)
2021-06-28 15:36:09 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 14:06:18 +08:00
return
2021-06-28 15:36:09 +08:00
}
2023-01-09 14:28:19 +08:00
for _, groupID := range joinedGroupList {
2021-12-23 17:22:49 +08:00
var groupNode open_im_sdk.GroupInfo
2023-01-09 16:51:26 +08:00
num, err := rocksCache.GetGroupMemberNumFromCache(ctx, groupID)
2022-07-20 15:49:21 +08:00
if err != nil {
2023-01-09 14:28:19 +08:00
log.NewError(req.OperationID, utils.GetSelfFuncName(), err.Error(), groupID)
2022-07-20 15:49:21 +08:00
continue
}
2023-01-09 14:28:19 +08:00
owner, err := (*imdb.GroupMember)(nil).TakeOwnerInfo(ctx, groupID)
//owner, err2 := imdb.GetGroupOwnerInfoByGroupID(groupID)
if err != nil {
trace_log.SetContextInfo(ctx, "TakeOwnerInfo", err, "groupID", groupID)
2021-12-23 17:22:49 +08:00
continue
2021-06-28 15:36:09 +08:00
}
2023-01-09 14:28:19 +08:00
group, err := rocksCache.GetGroupInfoFromCache(ctx, groupID)
2022-07-20 15:49:21 +08:00
if err != nil {
2021-12-23 17:22:49 +08:00
continue
2021-06-28 15:36:09 +08:00
}
2022-08-16 19:54:04 +08:00
if group.GroupType == constant.SuperGroup {
continue
}
2022-07-20 15:49:21 +08:00
if group.Status == constant.GroupStatusDismissed {
2023-01-09 14:28:19 +08:00
trace_log.SetContextInfo(ctx, "GetGroupInfoFromCache", err, "groupID", groupID)
2022-07-20 15:49:21 +08:00
continue
}
utils.CopyStructFields(&groupNode, group)
groupNode.CreateTime = uint32(group.CreateTime.Unix())
groupNode.NotificationUpdateTime = uint32(group.NotificationUpdateTime.Unix())
if group.NotificationUpdateTime.Unix() < 0 {
groupNode.NotificationUpdateTime = 0
}
groupNode.MemberCount = uint32(num)
groupNode.OwnerUserID = owner.UserID
resp.GroupList = append(resp.GroupList, &groupNode)
2021-06-28 15:36:09 +08:00
}
2023-01-09 14:28:19 +08:00
return
2021-06-28 15:36:09 +08:00
}
2023-01-06 17:00:14 +08:00
func (s *groupServer) InviteUserToGroup(ctx context.Context, req *pbGroup.InviteUserToGroupReq) (resp *pbGroup.InviteUserToGroupResp, _ error) {
resp = &pbGroup.InviteUserToGroupResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetFuncName(1), nil, "req", req.String(), "resp", resp.String())
trace_log.ShowLog(ctx)
}()
2022-04-16 20:10:10 +08:00
if !imdb.IsExistGroupMember(req.GroupID, req.OpUserID) && !token_verify.IsManagerUserID(req.OpUserID) {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrIdentity, resp.CommonResp)
2023-01-06 17:00:14 +08:00
return
2021-07-15 11:16:24 +08:00
}
2023-01-06 17:00:14 +08:00
groupInfo, err := (*imdb.Group)(nil).Take(ctx, req.GroupID)
2021-12-28 11:57:12 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:00:14 +08:00
return
2022-04-18 10:53:23 +08:00
}
if groupInfo.Status == constant.GroupStatusDismissed {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrDismissedAlready, resp.CommonResp)
2023-01-06 17:00:14 +08:00
return
}
2022-06-23 18:56:52 +08:00
if groupInfo.NeedVerification == constant.AllNeedVerification &&
!imdb.IsGroupOwnerAdmin(req.GroupID, req.OpUserID) && !token_verify.IsManagerUserID(req.OpUserID) {
joinReq := pbGroup.JoinGroupReq{}
for _, v := range req.InvitedUserIDList {
2023-01-04 17:22:55 +08:00
var groupRequest imdb.GroupRequest
2022-06-23 18:56:52 +08:00
groupRequest.UserID = v
groupRequest.GroupID = req.GroupID
2022-07-12 19:24:06 +08:00
groupRequest.JoinSource = constant.JoinByInvitation
groupRequest.InviterUserID = req.OpUserID
2022-06-23 18:56:52 +08:00
err = imdb.InsertIntoGroupRequest(groupRequest)
if err != nil {
var resultNode pbGroup.Id2Result
resultNode.Result = -1
resultNode.UserID = v
resp.Id2ResultList = append(resp.Id2ResultList, &resultNode)
continue
} else {
var resultNode pbGroup.Id2Result
resultNode.Result = 0
resultNode.UserID = v
resp.Id2ResultList = append(resp.Id2ResultList, &resultNode)
joinReq.GroupID = req.GroupID
joinReq.OperationID = req.OperationID
joinReq.OpUserID = v
resp.Id2ResultList = append(resp.Id2ResultList, &resultNode)
chat.JoinGroupApplicationNotification(&joinReq)
}
}
2023-01-06 17:00:14 +08:00
return
2022-06-23 18:56:52 +08:00
}
2023-01-09 16:37:33 +08:00
if err := s.DelGroupAndUserCache(ctx, req.GroupID, req.InvitedUserIDList); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:00:14 +08:00
return
2022-11-22 11:07:20 +08:00
}
2021-06-28 15:36:09 +08:00
//from User: invite: applicant
//to user: invite: invited
2021-12-24 15:02:47 +08:00
var okUserIDList []string
2022-05-27 18:40:24 +08:00
if groupInfo.GroupType != constant.SuperGroup {
for _, v := range req.InvitedUserIDList {
var resultNode pbGroup.Id2Result
resultNode.UserID = v
resultNode.Result = 0
toUserInfo, err := imdb.GetUserByUserID(v)
if err != nil {
2023-01-06 17:00:14 +08:00
trace_log.SetContextInfo(ctx, "GetUserByUserID", err, "userID", v)
2022-05-27 18:40:24 +08:00
resultNode.Result = -1
resp.Id2ResultList = append(resp.Id2ResultList, &resultNode)
continue
}
2021-07-15 11:16:24 +08:00
2022-05-27 18:40:24 +08:00
if imdb.IsExistGroupMember(req.GroupID, v) {
2023-01-06 17:00:14 +08:00
trace_log.SetContextInfo(ctx, "IsExistGroupMember", err, "groupID", req.GroupID, "userID", v)
2022-05-27 18:40:24 +08:00
resultNode.Result = -1
resp.Id2ResultList = append(resp.Id2ResultList, &resultNode)
continue
}
2023-01-04 17:22:55 +08:00
var toInsertInfo imdb.GroupMember
2022-05-27 18:40:24 +08:00
utils.CopyStructFields(&toInsertInfo, toUserInfo)
toInsertInfo.GroupID = req.GroupID
toInsertInfo.RoleLevel = constant.GroupOrdinaryUsers
toInsertInfo.OperatorUserID = req.OpUserID
2022-07-12 19:24:06 +08:00
toInsertInfo.InviterUserID = req.OpUserID
toInsertInfo.JoinSource = constant.JoinByInvitation
2023-01-06 17:00:14 +08:00
if err := CallbackBeforeMemberJoinGroup(ctx, req.OperationID, &toInsertInfo, groupInfo.Ex); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:00:14 +08:00
return
2022-11-22 11:07:20 +08:00
}
2022-05-27 18:40:24 +08:00
err = imdb.InsertIntoGroupMember(toInsertInfo)
if err != nil {
2023-01-06 17:00:14 +08:00
trace_log.SetContextInfo(ctx, "InsertIntoGroupMember", err, "args", toInsertInfo)
2022-05-27 18:40:24 +08:00
resultNode.Result = -1
resp.Id2ResultList = append(resp.Id2ResultList, &resultNode)
continue
}
okUserIDList = append(okUserIDList, v)
err = db.DB.AddGroupMember(req.GroupID, toUserInfo.UserID)
if err != nil {
2023-01-06 17:00:14 +08:00
trace_log.SetContextInfo(ctx, "AddGroupMember", err, "groupID", req.GroupID, "userID", toUserInfo.UserID)
2022-05-27 18:40:24 +08:00
}
2021-12-23 17:22:49 +08:00
resp.Id2ResultList = append(resp.Id2ResultList, &resultNode)
2021-06-28 15:36:09 +08:00
}
2022-05-27 18:40:24 +08:00
} else {
okUserIDList = req.InvitedUserIDList
if err := db.DB.AddUserToSuperGroup(req.GroupID, req.InvitedUserIDList); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:00:14 +08:00
return
2022-04-21 16:46:48 +08:00
}
2022-05-27 18:40:24 +08:00
}
2022-12-02 18:39:15 +08:00
2022-05-27 18:40:24 +08:00
if groupInfo.GroupType != constant.SuperGroup {
chat.MemberInvitedNotification(req.OperationID, req.GroupID, req.OpUserID, req.Reason, okUserIDList)
2022-05-30 15:10:26 +08:00
} else {
2023-01-09 16:37:33 +08:00
for _, userID := range req.InvitedUserIDList {
if err := rocksCache.DelJoinedSuperGroupIDListFromCache(ctx, userID); err != nil {
trace_log.SetContextInfo(ctx, "DelJoinedSuperGroupIDListFromCache", err, "userID", userID)
2022-07-29 14:36:07 +08:00
}
}
2022-08-26 11:39:01 +08:00
for _, v := range req.InvitedUserIDList {
chat.SuperGroupNotification(req.OperationID, v, v)
}
2022-05-27 18:40:24 +08:00
}
2023-01-06 17:00:14 +08:00
return
2021-06-28 15:36:09 +08:00
}
2023-01-06 17:25:22 +08:00
func (s *groupServer) GetGroupAllMember(ctx context.Context, req *pbGroup.GetGroupAllMemberReq) (resp *pbGroup.GetGroupAllMemberResp, err error) {
resp = &pbGroup.GetGroupAllMemberResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetFuncName(1), nil, "req", req.String(), "resp", resp.String())
trace_log.ShowLog(ctx)
}()
2023-01-06 17:00:14 +08:00
groupInfo, err := rocksCache.GetGroupInfoFromCache(ctx, req.GroupID)
2021-06-28 15:36:09 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:25:22 +08:00
return
2021-06-28 15:36:09 +08:00
}
2022-05-27 18:40:24 +08:00
if groupInfo.GroupType != constant.SuperGroup {
2023-01-09 16:37:33 +08:00
memberList, err := rocksCache.GetGroupMembersInfoFromCache(ctx, req.Count, req.Offset, req.GroupID)
2022-05-27 18:40:24 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:25:22 +08:00
return
2022-05-27 18:40:24 +08:00
}
for _, v := range memberList {
var node open_im_sdk.GroupMemberFullInfo
2022-07-14 12:08:28 +08:00
cp.GroupMemberDBCopyOpenIM(&node, v)
2022-05-27 18:40:24 +08:00
resp.MemberList = append(resp.MemberList, &node)
}
2021-12-23 17:22:49 +08:00
}
2023-01-06 17:25:22 +08:00
return
2021-06-28 15:36:09 +08:00
}
2023-01-06 17:25:22 +08:00
func (s *groupServer) GetGroupMemberList(ctx context.Context, req *pbGroup.GetGroupMemberListReq) (resp *pbGroup.GetGroupMemberListResp, err error) {
resp = &pbGroup.GetGroupMemberListResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetFuncName(1), nil, "req", req.String(), "resp", resp.String())
trace_log.ShowLog(ctx)
}()
2021-12-27 21:08:42 +08:00
memberList, err := imdb.GetGroupMemberByGroupID(req.GroupID, req.Filter, req.NextSeq, 30)
2021-06-28 15:36:09 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:25:22 +08:00
return
2021-06-28 15:36:09 +08:00
}
for _, v := range memberList {
2021-12-23 17:22:49 +08:00
var node open_im_sdk.GroupMemberFullInfo
2021-12-30 16:30:13 +08:00
utils.CopyStructFields(&node, &v)
2021-06-28 15:36:09 +08:00
resp.MemberList = append(resp.MemberList, &node)
}
//db operate get db sorted by join time
if int32(len(memberList)) < 30 {
resp.NextSeq = 0
} else {
resp.NextSeq = req.NextSeq + int32(len(memberList))
}
2023-01-06 17:25:22 +08:00
return
2021-06-28 15:36:09 +08:00
}
2022-06-06 23:19:07 +08:00
func (s *groupServer) getGroupUserLevel(groupID, userID string) (int, error) {
opFlag := 0
if !token_verify.IsManagerUserID(userID) {
opInfo, err := imdb.GetGroupMemberInfoByGroupIDAndUserID(groupID, userID)
if err != nil {
return opFlag, utils.Wrap(err, "")
}
if opInfo.RoleLevel == constant.GroupOrdinaryUsers {
opFlag = 0
} else if opInfo.RoleLevel == constant.GroupOwner {
opFlag = 2 //owner
} else {
opFlag = 3 //admin
}
} else {
opFlag = 1 //app manager
}
return opFlag, nil
}
2023-01-06 17:25:22 +08:00
func (s *groupServer) KickGroupMember(ctx context.Context, req *pbGroup.KickGroupMemberReq) (resp *pbGroup.KickGroupMemberResp, _ error) {
resp = &pbGroup.KickGroupMemberResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetFuncName(1), nil, "req", req.String(), "resp", resp.String())
trace_log.ShowLog(ctx)
}()
groupInfo, err := rocksCache.GetGroupInfoFromCache(ctx, req.GroupID)
2021-06-28 15:36:09 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:25:22 +08:00
return
2021-06-28 15:36:09 +08:00
}
2022-05-27 18:40:24 +08:00
var okUserIDList []string
if groupInfo.GroupType != constant.SuperGroup {
2022-06-06 23:19:07 +08:00
opFlag := 0
if !token_verify.IsManagerUserID(req.OpUserID) {
2023-01-09 16:37:33 +08:00
opInfo, err := rocksCache.GetGroupMemberInfoFromCache(ctx, req.GroupID, req.OpUserID)
2022-06-06 23:19:07 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:25:22 +08:00
return
2022-05-27 18:40:24 +08:00
}
2022-06-06 23:19:07 +08:00
if opInfo.RoleLevel == constant.GroupOrdinaryUsers {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrNoPermission, resp.CommonResp)
2023-01-06 17:25:22 +08:00
return
2022-06-06 23:19:07 +08:00
} else if opInfo.RoleLevel == constant.GroupOwner {
opFlag = 2 //owner
} else {
opFlag = 3 //admin
2022-05-27 18:40:24 +08:00
}
2022-06-06 23:19:07 +08:00
} else {
opFlag = 1 //app manager
2021-09-22 20:10:38 +08:00
}
2022-06-06 23:19:07 +08:00
//op is group owner?
2022-05-27 18:40:24 +08:00
if len(req.KickedUserIDList) == 0 {
2023-01-06 17:25:22 +08:00
//log.NewError(req.OperationID, "failed, kick list 0")
//return &pbGroup.KickGroupMemberResp{ErrCode: constant.ErrArgs.ErrCode, ErrMsg: constant.ErrArgs.ErrMsg}, nil
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrArgs, resp.CommonResp)
2023-01-06 17:25:22 +08:00
return
2022-05-27 18:40:24 +08:00
}
2023-01-09 16:37:33 +08:00
if err := s.DelGroupAndUserCache(ctx, req.GroupID, req.KickedUserIDList); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:25:22 +08:00
return
2022-11-22 11:07:20 +08:00
}
2022-05-27 18:40:24 +08:00
//remove
for _, v := range req.KickedUserIDList {
2023-01-09 16:37:33 +08:00
kickedInfo, err := rocksCache.GetGroupMemberInfoFromCache(ctx, req.GroupID, v)
2022-06-06 23:19:07 +08:00
if err != nil {
resp.Id2ResultList = append(resp.Id2ResultList, &pbGroup.Id2Result{UserID: v, Result: -1})
2023-01-06 17:25:22 +08:00
trace_log.SetContextInfo(ctx, "GetGroupMemberInfoFromCache", err, "groupID", req.GroupID, "userID", v)
2022-06-06 23:19:07 +08:00
continue
}
if kickedInfo.RoleLevel == constant.GroupAdmin && opFlag == 3 {
2022-05-27 18:40:24 +08:00
resp.Id2ResultList = append(resp.Id2ResultList, &pbGroup.Id2Result{UserID: v, Result: -1})
2023-01-06 17:25:22 +08:00
trace_log.SetContextInfo(ctx, "", nil, "msg", "is constant.GroupAdmin, can't kicked", "groupID", req.GroupID, "userID", v)
2022-05-27 18:40:24 +08:00
continue
}
2022-06-06 23:19:07 +08:00
if kickedInfo.RoleLevel == constant.GroupOwner && opFlag != 1 {
resp.Id2ResultList = append(resp.Id2ResultList, &pbGroup.Id2Result{UserID: v, Result: -1})
2023-01-06 17:25:22 +08:00
trace_log.SetContextInfo(ctx, "", nil, "msg", "is constant.GroupOwner, can't kicked", "groupID", req.GroupID, "userID", v)
2022-06-06 23:19:07 +08:00
continue
}
2022-08-12 18:37:51 +08:00
err = imdb.DeleteGroupMemberByGroupIDAndUserID(req.GroupID, v)
2023-01-06 17:25:22 +08:00
trace_log.SetContextInfo(ctx, "RemoveGroupMember", err, "groupID", req.GroupID, "userID", v)
2022-05-27 18:40:24 +08:00
if err != nil {
log.NewError(req.OperationID, "RemoveGroupMember failed ", err.Error(), req.GroupID, v)
resp.Id2ResultList = append(resp.Id2ResultList, &pbGroup.Id2Result{UserID: v, Result: -1})
} else {
resp.Id2ResultList = append(resp.Id2ResultList, &pbGroup.Id2Result{UserID: v, Result: 0})
okUserIDList = append(okUserIDList, v)
}
2021-06-28 15:36:09 +08:00
}
2022-05-27 18:40:24 +08:00
var reqPb pbUser.SetConversationReq
2022-08-21 23:37:43 +08:00
var c pbConversation.Conversation
2022-05-27 18:40:24 +08:00
for _, v := range okUserIDList {
reqPb.OperationID = req.OperationID
c.OwnerUserID = v
c.ConversationID = utils.GetConversationIDBySessionType(req.GroupID, constant.GroupChatType)
c.ConversationType = constant.GroupChatType
c.GroupID = req.GroupID
c.IsNotInGroup = true
reqPb.Conversation = &c
2023-01-09 16:37:33 +08:00
etcdConn, err := utils.GetConn(ctx, config.Config.RpcRegisterName.OpenImUserName)
2023-01-06 17:25:22 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:25:22 +08:00
return
2022-06-16 14:51:36 +08:00
}
2022-05-27 18:40:24 +08:00
client := pbUser.NewUserClient(etcdConn)
respPb, err := client.SetConversation(context.Background(), &reqPb)
2023-01-06 17:25:22 +08:00
trace_log.SetContextInfo(ctx, "SetConversation", err, "req", &reqPb, "resp", respPb)
2021-06-28 15:36:09 +08:00
}
2022-05-27 18:40:24 +08:00
} else {
okUserIDList = req.KickedUserIDList
if err := db.DB.RemoverUserFromSuperGroup(req.GroupID, okUserIDList); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:25:22 +08:00
return
2021-07-15 11:16:24 +08:00
}
2021-06-28 15:36:09 +08:00
}
2022-05-07 11:14:29 +08:00
2022-05-27 18:40:24 +08:00
if groupInfo.GroupType != constant.SuperGroup {
2022-07-29 14:36:07 +08:00
for _, userID := range okUserIDList {
2023-01-09 16:37:33 +08:00
if err := rocksCache.DelGroupMemberInfoFromCache(ctx, req.GroupID, userID); err != nil {
2023-01-06 17:25:22 +08:00
trace_log.SetContextInfo(ctx, "DelGroupMemberInfoFromCache", err, "groupID", req.GroupID, "userID", userID)
2022-08-04 17:20:33 +08:00
}
2022-07-29 14:36:07 +08:00
}
2022-05-27 18:40:24 +08:00
chat.MemberKickedNotification(req, okUserIDList)
2022-05-30 15:10:26 +08:00
} else {
2022-07-29 14:36:07 +08:00
for _, userID := range okUserIDList {
2023-01-09 16:37:33 +08:00
if err = rocksCache.DelJoinedSuperGroupIDListFromCache(ctx, userID); err != nil {
2023-01-06 17:25:22 +08:00
trace_log.SetContextInfo(ctx, "DelGroupMemberInfoFromCache", err, "userID", userID)
2022-07-29 14:36:07 +08:00
}
}
2022-06-21 14:54:42 +08:00
go func() {
for _, v := range req.KickedUserIDList {
chat.SuperGroupNotification(req.OperationID, v, v)
}
}()
2022-05-27 18:40:24 +08:00
}
2023-01-06 17:25:22 +08:00
return
2021-06-28 15:36:09 +08:00
}
2023-01-05 20:23:33 +08:00
func (s *groupServer) GetGroupMembersInfo(ctx context.Context, req *pbGroup.GetGroupMembersInfoReq) (resp *pbGroup.GetGroupMembersInfoResp, err error) {
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), err, "rpc req", req.String(), "rpc resp", resp.String())
trace_log.ShowLog(ctx)
}()
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
resp = &pbGroup.GetGroupMembersInfoResp{CommonResp: &open_im_sdk.CommonResp{}}
2022-07-14 12:08:28 +08:00
resp.MemberList = []*open_im_sdk.GroupMemberFullInfo{}
2022-08-04 17:20:33 +08:00
for _, userID := range req.MemberList {
2023-01-09 16:37:33 +08:00
groupMember, err := rocksCache.GetGroupMemberInfoFromCache(ctx, req.GroupID, userID)
2022-08-04 17:20:33 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
return
2021-06-28 15:36:09 +08:00
}
2022-08-04 17:20:33 +08:00
var memberNode open_im_sdk.GroupMemberFullInfo
utils.CopyStructFields(&memberNode, groupMember)
memberNode.JoinTime = int32(groupMember.JoinTime.Unix())
resp.MemberList = append(resp.MemberList, &memberNode)
2021-06-28 15:36:09 +08:00
}
2023-01-05 20:23:33 +08:00
return resp, nil
2021-06-28 15:36:09 +08:00
}
2021-12-22 15:24:49 +08:00
2023-01-04 10:12:20 +08:00
func FillGroupInfoByGroupID(operationID, groupID string, groupInfo *open_im_sdk.GroupInfo) error {
group, err := imdb.TakeGroupInfoByGroupID(groupID)
if err != nil {
log.Error(operationID, "TakeGroupInfoByGroupID failed ", err.Error(), groupID)
return utils.Wrap(err, "")
}
if group.Status == constant.GroupStatusDismissed {
log.Debug(operationID, " group constant.GroupStatusDismissed ", group.GroupID)
2023-01-04 11:06:28 +08:00
return utils.Wrap(constant.ErrDismissedAlready, "")
2023-01-04 10:12:20 +08:00
}
return utils.Wrap(cp.GroupDBCopyOpenIM(groupInfo, group), "")
}
func FillPublicUserInfoByUserID(operationID, userID string, userInfo *open_im_sdk.PublicUserInfo) error {
user, err := imdb.TakeUserByUserID(userID)
2021-12-22 15:24:49 +08:00
if err != nil {
2023-01-04 10:12:20 +08:00
log.Error(operationID, "TakeUserByUserID failed ", err.Error(), userID)
return utils.Wrap(err, "")
2021-12-22 15:24:49 +08:00
}
2023-01-04 10:12:20 +08:00
cp.UserDBCopyOpenIMPublicUser(userInfo, user)
return nil
}
2021-12-27 21:08:42 +08:00
2023-01-05 20:23:33 +08:00
func (s *groupServer) GetGroupApplicationList(ctx context.Context, req *pbGroup.GetGroupApplicationListReq) (resp *pbGroup.GetGroupApplicationListResp, err error) {
2023-01-05 17:09:35 +08:00
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
2023-01-06 13:41:50 +08:00
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), err, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
2023-01-04 11:06:28 +08:00
2023-01-05 20:23:33 +08:00
resp = &pbGroup.GetGroupApplicationListResp{CommonResp: &open_im_sdk.CommonResp{}}
2023-01-04 10:12:20 +08:00
reply, err := imdb.GetRecvGroupApplicationList(req.FromUserID)
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-05 20:23:33 +08:00
return resp, nil
2023-01-04 10:12:20 +08:00
}
var errResult error
2023-01-05 17:09:35 +08:00
trace_log.SetContextInfo(ctx, "GetRecvGroupApplicationList", nil, " FromUserID: ", req.FromUserID, "GroupApplicationList: ", reply)
2021-12-27 21:08:42 +08:00
for _, v := range reply {
2022-01-15 15:25:12 +08:00
node := open_im_sdk.GroupRequest{UserInfo: &open_im_sdk.PublicUserInfo{}, GroupInfo: &open_im_sdk.GroupInfo{}}
2023-01-04 10:12:20 +08:00
err := FillGroupInfoByGroupID(req.OperationID, v.GroupID, node.GroupInfo)
2022-01-15 15:25:12 +08:00
if err != nil {
2023-01-04 10:12:20 +08:00
if !errors.Is(errors.Unwrap(err), constant.ErrDismissedAlready) {
errResult = err
}
2022-07-20 10:18:43 +08:00
continue
}
2023-01-05 17:09:35 +08:00
trace_log.SetContextInfo(ctx, "FillGroupInfoByGroupID ", nil, " groupID: ", v.GroupID, " groupInfo: ", node.GroupInfo)
2023-01-04 10:12:20 +08:00
err = FillPublicUserInfoByUserID(req.OperationID, v.UserID, node.UserInfo)
2022-01-15 15:25:12 +08:00
if err != nil {
2023-01-04 10:12:20 +08:00
errResult = err
2022-01-15 15:25:12 +08:00
continue
}
2021-12-30 10:56:17 +08:00
cp.GroupRequestDBCopyOpenIM(&node, &v)
2021-12-30 11:13:19 +08:00
resp.GroupRequestList = append(resp.GroupRequestList, &node)
2021-12-27 21:08:42 +08:00
}
2023-01-04 10:12:20 +08:00
if errResult != nil && len(resp.GroupRequestList) == 0 {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-05 20:23:33 +08:00
return resp, nil
2023-01-04 10:12:20 +08:00
}
2023-01-05 17:09:35 +08:00
trace_log.SetRpcRespInfo(ctx, utils.GetSelfFuncName(), resp.String())
2023-01-05 20:23:33 +08:00
return resp, nil
2021-12-22 15:24:49 +08:00
}
2023-01-06 17:58:35 +08:00
func (s *groupServer) GetGroupsInfo(ctx context.Context, req *pbGroup.GetGroupsInfoReq) (resp *pbGroup.GetGroupsInfoResp, _ error) {
resp = &pbGroup.GetGroupsInfoResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
2023-01-05 17:14:28 +08:00
defer func() {
2023-01-06 17:58:35 +08:00
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
2023-01-05 17:14:28 +08:00
trace_log.ShowLog(ctx)
}()
2021-12-23 17:22:49 +08:00
groupsInfoList := make([]*open_im_sdk.GroupInfo, 0)
2021-12-22 15:24:49 +08:00
for _, groupID := range req.GroupIDList {
2023-01-05 17:09:35 +08:00
groupInfoFromRedis, err := rocksCache.GetGroupInfoFromCache(ctx, groupID)
2021-12-22 15:24:49 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2021-12-22 15:24:49 +08:00
continue
}
2021-12-23 17:22:49 +08:00
var groupInfo open_im_sdk.GroupInfo
2022-07-14 12:08:28 +08:00
cp.GroupDBCopyOpenIM(&groupInfo, groupInfoFromRedis)
groupInfo.NeedVerification = groupInfoFromRedis.NeedVerification
2021-12-22 15:24:49 +08:00
groupsInfoList = append(groupsInfoList, &groupInfo)
}
2023-01-04 15:51:02 +08:00
resp.GroupInfoList = groupsInfoList
2023-01-06 17:58:35 +08:00
return
2021-12-22 15:24:49 +08:00
}
2023-01-05 15:52:48 +08:00
func CheckPermission(ctx context.Context, groupID string, userID string) (err error) {
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), err, "groupID", groupID, "userID", userID)
}()
if !token_verify.IsManagerUserID(userID) && !imdb.IsGroupOwnerAdmin(groupID, userID) {
2023-01-05 17:09:35 +08:00
return utils.Wrap(constant.ErrNoPermission, utils.GetSelfFuncName())
2023-01-05 15:52:48 +08:00
}
2023-01-05 11:33:17 +08:00
return nil
2023-01-04 17:22:55 +08:00
}
2023-01-05 11:33:17 +08:00
2023-01-06 17:58:35 +08:00
func (s *groupServer) GroupApplicationResponse(ctx context.Context, req *pbGroup.GroupApplicationResponseReq) (resp *pbGroup.GroupApplicationResponseResp, _ error) {
resp = &pbGroup.GroupApplicationResponseResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2023-01-05 13:36:49 +08:00
if err := CheckPermission(ctx, req.GroupID, req.OpUserID); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:58:35 +08:00
return
2021-12-30 12:05:10 +08:00
}
2023-01-04 17:22:55 +08:00
groupRequest := imdb.GroupRequest{}
utils.CopyStructFields(&groupRequest, req)
groupRequest.UserID = req.FromUserID
groupRequest.HandleUserID = req.OpUserID
groupRequest.HandledTime = time.Now()
2023-01-05 13:36:49 +08:00
if err := (&imdb.GroupRequest{}).Update(ctx, []*imdb.GroupRequest{&groupRequest}); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:58:35 +08:00
return
2022-07-12 19:24:06 +08:00
}
2023-01-05 13:36:49 +08:00
groupInfo, err := rocksCache.GetGroupInfoFromCache(ctx, req.GroupID)
2022-11-22 11:07:20 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:58:35 +08:00
return
2022-11-22 11:07:20 +08:00
}
2022-01-16 16:31:28 +08:00
if req.HandleResult == constant.GroupResponseAgree {
user, err := imdb.GetUserByUserID(req.FromUserID)
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:58:35 +08:00
return
2023-01-05 11:33:17 +08:00
}
2023-01-05 13:36:49 +08:00
request, err := (&imdb.GroupRequest{}).Take(ctx, req.GroupID, req.FromUserID)
2023-01-05 11:33:17 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:58:35 +08:00
return
2021-12-30 12:27:26 +08:00
}
2023-01-04 17:22:55 +08:00
member := imdb.GroupMember{}
2022-01-16 16:31:28 +08:00
member.GroupID = req.GroupID
member.UserID = req.FromUserID
member.RoleLevel = constant.GroupOrdinaryUsers
member.OperatorUserID = req.OpUserID
2022-02-09 15:07:31 +08:00
member.FaceURL = user.FaceURL
2022-01-16 16:31:28 +08:00
member.Nickname = user.Nickname
2022-07-12 19:24:06 +08:00
member.JoinSource = request.JoinSource
member.InviterUserID = request.InviterUserID
2023-01-05 11:33:17 +08:00
member.MuteEndTime = time.Unix(int64(time.Now().Second()), 0)
2023-01-05 20:23:33 +08:00
err = CallbackBeforeMemberJoinGroup(ctx, req.OperationID, &member, groupInfo.Ex)
2023-01-06 11:19:31 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:58:35 +08:00
return
2022-11-22 11:07:20 +08:00
}
2023-01-05 20:23:33 +08:00
2023-01-05 15:04:48 +08:00
err = (&imdb.GroupMember{}).Create(ctx, []*imdb.GroupMember{&member})
2022-01-16 16:31:28 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:58:35 +08:00
return
2022-04-21 16:46:48 +08:00
}
2023-01-05 11:33:17 +08:00
etcdCacheConn, err := fault_tolerant.GetDefaultConn(config.Config.RpcRegisterName.OpenImCacheName, req.OperationID)
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:58:35 +08:00
return
2022-06-16 14:51:36 +08:00
}
2022-05-07 15:01:48 +08:00
cacheClient := pbCache.NewCacheClient(etcdCacheConn)
2022-07-14 12:08:28 +08:00
cacheResp, err := cacheClient.DelGroupMemberIDListFromCache(context.Background(), &pbCache.DelGroupMemberIDListFromCacheReq{OperationID: req.OperationID, GroupID: req.GroupID})
2022-05-07 15:01:48 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:58:35 +08:00
return
2022-05-07 15:01:48 +08:00
}
if cacheResp.CommonResp.ErrCode != 0 {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 17:58:35 +08:00
return
2022-05-07 15:01:48 +08:00
}
2023-01-09 16:37:33 +08:00
_ = rocksCache.DelGroupMemberListHashFromCache(ctx, req.GroupID)
_ = rocksCache.DelJoinedGroupIDListFromCache(ctx, req.FromUserID)
_ = rocksCache.DelGroupMemberNumFromCache(ctx, req.GroupID)
2022-01-16 16:31:28 +08:00
chat.GroupApplicationAcceptedNotification(req)
2021-12-27 16:48:05 +08:00
chat.MemberEnterNotification(req)
2022-01-16 16:31:28 +08:00
} else if req.HandleResult == constant.GroupResponseRefuse {
chat.GroupApplicationRejectedNotification(req)
} else {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrArgs, resp.CommonResp)
2023-01-06 17:58:35 +08:00
return
2021-12-22 15:24:49 +08:00
}
2023-01-06 17:58:35 +08:00
return
2021-12-22 15:24:49 +08:00
}
2023-01-06 18:25:41 +08:00
func (s *groupServer) JoinGroup(ctx context.Context, req *pbGroup.JoinGroupReq) (resp *pbGroup.JoinGroupResp, _ error) {
resp = &pbGroup.JoinGroupResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
if _, err := imdb.GetUserByUserID(req.OpUserID); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2021-12-22 15:24:49 +08:00
}
2023-01-06 18:25:41 +08:00
groupInfo, err := rocksCache.GetGroupInfoFromCache(ctx, req.GroupID)
2022-04-18 10:53:23 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-04-18 10:53:23 +08:00
}
if groupInfo.Status == constant.GroupStatusDismissed {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrDismissedAlready, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-04-18 10:53:23 +08:00
}
2022-06-23 18:56:52 +08:00
if groupInfo.NeedVerification == constant.Directly {
if groupInfo.GroupType != constant.SuperGroup {
2022-07-14 12:08:28 +08:00
us, err := imdb.GetUserByUserID(req.OpUserID)
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-07-14 12:08:28 +08:00
}
//to group member
2023-01-04 17:22:55 +08:00
groupMember := imdb.GroupMember{GroupID: req.GroupID, RoleLevel: constant.GroupOrdinaryUsers, OperatorUserID: req.OpUserID}
2022-07-14 12:08:28 +08:00
utils.CopyStructFields(&groupMember, us)
2023-01-06 18:25:41 +08:00
if err := CallbackBeforeMemberJoinGroup(ctx, req.OperationID, &groupMember, groupInfo.Ex); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-06-23 18:56:52 +08:00
}
2023-01-09 16:37:33 +08:00
if err := s.DelGroupAndUserCache(ctx, req.GroupID, []string{req.OpUserID}); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-06-23 18:56:52 +08:00
}
2022-11-22 11:07:20 +08:00
err = imdb.InsertIntoGroupMember(groupMember)
2022-07-15 18:47:04 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-08-08 11:30:10 +08:00
}
2022-12-05 15:49:08 +08:00
var sessionType int
if groupInfo.GroupType == constant.NormalGroup {
sessionType = constant.GroupChatType
} else {
sessionType = constant.SuperGroupChatType
}
2022-12-05 11:30:54 +08:00
var reqPb pbUser.SetConversationReq
var c pbConversation.Conversation
reqPb.OperationID = req.OperationID
c.OwnerUserID = req.OpUserID
2022-12-05 15:49:08 +08:00
c.ConversationID = utils.GetConversationIDBySessionType(req.GroupID, sessionType)
c.ConversationType = int32(sessionType)
2022-12-05 11:30:54 +08:00
c.GroupID = req.GroupID
c.IsNotInGroup = false
c.UpdateUnreadCountTime = utils.GetCurrentTimestampByMill()
reqPb.Conversation = &c
2023-01-09 16:37:33 +08:00
etcdConn, err := utils.GetConn(ctx, config.Config.RpcRegisterName.OpenImUserName)
2023-01-06 18:25:41 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-12-05 11:30:54 +08:00
}
client := pbUser.NewUserClient(etcdConn)
respPb, err := client.SetConversation(context.Background(), &reqPb)
2023-01-06 18:25:41 +08:00
trace_log.SetContextInfo(ctx, "SetConversation", err, "req", reqPb, "resp", respPb)
2022-06-23 18:56:52 +08:00
chat.MemberEnterDirectlyNotification(req.GroupID, req.OpUserID, req.OperationID)
2023-01-06 18:25:41 +08:00
return
2022-06-23 18:56:52 +08:00
} else {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrGroupTypeNotSupport, resp.CommonResp)
log.Error(req.OperationID, "JoinGroup rpc failed, group type: ", groupInfo.GroupType, "not support directly")
2023-01-06 18:25:41 +08:00
return
2022-06-23 18:56:52 +08:00
}
}
2023-01-04 17:22:55 +08:00
var groupRequest imdb.GroupRequest
2021-12-30 10:14:30 +08:00
groupRequest.UserID = req.OpUserID
groupRequest.ReqMsg = req.ReqMessage
groupRequest.GroupID = req.GroupID
2022-07-12 19:24:06 +08:00
groupRequest.JoinSource = req.JoinSource
2021-12-30 13:53:05 +08:00
err = imdb.InsertIntoGroupRequest(groupRequest)
2021-12-27 21:08:42 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2021-12-22 15:24:49 +08:00
}
2022-01-16 16:31:28 +08:00
chat.JoinGroupApplicationNotification(req)
2023-01-06 18:25:41 +08:00
return
2021-12-22 15:24:49 +08:00
}
2023-01-06 18:25:41 +08:00
func (s *groupServer) QuitGroup(ctx context.Context, req *pbGroup.QuitGroupReq) (resp *pbGroup.QuitGroupResp, _ error) {
resp = &pbGroup.QuitGroupResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2022-06-20 11:57:53 +08:00
groupInfo, err := imdb.GetGroupInfoByGroupID(req.GroupID)
2021-12-22 15:24:49 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2021-12-22 15:24:49 +08:00
}
2022-06-20 11:57:53 +08:00
if groupInfo.GroupType != constant.SuperGroup {
2023-01-09 16:37:33 +08:00
_, err = rocksCache.GetGroupMemberInfoFromCache(ctx, req.GroupID, req.OpUserID)
2022-06-20 11:57:53 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-06-20 11:57:53 +08:00
}
2023-01-09 16:37:33 +08:00
if err := s.DelGroupAndUserCache(ctx, req.GroupID, []string{req.OpUserID}); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-11-22 11:07:20 +08:00
}
2022-06-20 11:57:53 +08:00
err = imdb.DeleteGroupMemberByGroupIDAndUserID(req.GroupID, req.OpUserID)
if err != nil {
2023-01-09 16:37:33 +08:00
log.Error(req.OperationID, "JoinGroup rpc failed, group type: ", groupInfo.GroupType, "not support directly")
2023-01-06 18:25:41 +08:00
return
2022-06-20 11:57:53 +08:00
}
2022-04-20 18:47:21 +08:00
} else {
2022-06-20 11:57:53 +08:00
okUserIDList := []string{req.OpUserID}
if err := db.DB.RemoverUserFromSuperGroup(req.GroupID, okUserIDList); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-06-20 11:57:53 +08:00
}
2022-04-20 18:47:21 +08:00
}
2022-05-07 11:14:29 +08:00
2022-06-20 11:57:53 +08:00
if groupInfo.GroupType != constant.SuperGroup {
2023-01-09 16:37:33 +08:00
_ = rocksCache.DelGroupMemberInfoFromCache(ctx, req.GroupID, req.OpUserID)
2022-06-20 11:57:53 +08:00
chat.MemberQuitNotification(req)
} else {
2023-01-09 16:37:33 +08:00
_ = rocksCache.DelJoinedSuperGroupIDListFromCache(ctx, req.OpUserID)
_ = rocksCache.DelGroupMemberListHashFromCache(ctx, req.GroupID)
2022-06-21 14:54:42 +08:00
chat.SuperGroupNotification(req.OperationID, req.OpUserID, req.OpUserID)
2022-06-20 11:57:53 +08:00
}
2023-01-06 18:25:41 +08:00
return
2021-12-22 15:24:49 +08:00
}
func hasAccess(req *pbGroup.SetGroupInfoReq) bool {
if utils.IsContain(req.OpUserID, config.Config.Manager.AppManagerUid) {
return true
}
2022-06-28 20:39:27 +08:00
groupUserInfo, err := imdb.GetGroupMemberInfoByGroupIDAndUserID(req.GroupInfoForSet.GroupID, req.OpUserID)
2021-12-22 15:24:49 +08:00
if err != nil {
2022-06-28 20:39:27 +08:00
log.NewError(req.OperationID, "GetGroupMemberInfoByGroupIDAndUserID failed, ", err.Error(), req.GroupInfoForSet.GroupID, req.OpUserID)
2021-12-22 15:24:49 +08:00
return false
}
2021-12-29 22:06:27 +08:00
if groupUserInfo.RoleLevel == constant.GroupOwner || groupUserInfo.RoleLevel == constant.GroupAdmin {
2021-12-22 15:24:49 +08:00
return true
}
2021-12-28 15:33:47 +08:00
return false
2021-12-22 15:24:49 +08:00
}
2023-01-06 18:25:41 +08:00
func (s *groupServer) SetGroupInfo(ctx context.Context, req *pbGroup.SetGroupInfoReq) (resp *pbGroup.SetGroupInfoResp, err error) {
resp = &pbGroup.SetGroupInfoResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2021-12-22 15:24:49 +08:00
if !hasAccess(req) {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrIdentity, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2021-12-22 15:24:49 +08:00
}
2022-06-28 20:39:27 +08:00
group, err := imdb.GetGroupInfoByGroupID(req.GroupInfoForSet.GroupID)
2021-12-22 15:24:49 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2021-12-22 15:24:49 +08:00
}
2022-04-18 10:53:23 +08:00
if group.Status == constant.GroupStatusDismissed {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrDismissedAlready, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-04-18 10:53:23 +08:00
}
2021-12-22 15:24:49 +08:00
var changedType int32
2022-06-06 20:39:45 +08:00
groupName := ""
notification := ""
introduction := ""
faceURL := ""
2022-06-28 20:39:27 +08:00
if group.GroupName != req.GroupInfoForSet.GroupName && req.GroupInfoForSet.GroupName != "" {
2021-12-22 15:24:49 +08:00
changedType = 1
2022-06-28 20:39:27 +08:00
groupName = req.GroupInfoForSet.GroupName
2021-12-22 15:24:49 +08:00
}
2022-06-28 20:39:27 +08:00
if group.Notification != req.GroupInfoForSet.Notification && req.GroupInfoForSet.Notification != "" {
2021-12-22 15:24:49 +08:00
changedType = changedType | (1 << 1)
2022-06-28 20:39:27 +08:00
notification = req.GroupInfoForSet.Notification
2021-12-22 15:24:49 +08:00
}
2022-06-28 20:39:27 +08:00
if group.Introduction != req.GroupInfoForSet.Introduction && req.GroupInfoForSet.Introduction != "" {
2021-12-22 15:24:49 +08:00
changedType = changedType | (1 << 2)
2022-06-28 20:39:27 +08:00
introduction = req.GroupInfoForSet.Introduction
2021-12-22 15:24:49 +08:00
}
2022-06-28 20:39:27 +08:00
if group.FaceURL != req.GroupInfoForSet.FaceURL && req.GroupInfoForSet.FaceURL != "" {
2021-12-22 15:24:49 +08:00
changedType = changedType | (1 << 3)
2022-06-28 20:39:27 +08:00
faceURL = req.GroupInfoForSet.FaceURL
2021-12-22 15:24:49 +08:00
}
2022-06-23 18:56:52 +08:00
2022-06-28 20:39:27 +08:00
if req.GroupInfoForSet.NeedVerification != nil {
2022-06-23 18:56:52 +08:00
changedType = changedType | (1 << 4)
m := make(map[string]interface{})
2022-06-28 20:39:27 +08:00
m["need_verification"] = req.GroupInfoForSet.NeedVerification.Value
if err := imdb.UpdateGroupInfoDefaultZero(req.GroupInfoForSet.GroupID, m); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-06-23 18:56:52 +08:00
}
}
2022-07-12 17:22:45 +08:00
if req.GroupInfoForSet.LookMemberInfo != nil {
changedType = changedType | (1 << 5)
m := make(map[string]interface{})
m["look_member_info"] = req.GroupInfoForSet.LookMemberInfo.Value
if err := imdb.UpdateGroupInfoDefaultZero(req.GroupInfoForSet.GroupID, m); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-07-12 17:22:45 +08:00
}
}
if req.GroupInfoForSet.ApplyMemberFriend != nil {
changedType = changedType | (1 << 6)
m := make(map[string]interface{})
m["apply_member_friend"] = req.GroupInfoForSet.ApplyMemberFriend.Value
if err := imdb.UpdateGroupInfoDefaultZero(req.GroupInfoForSet.GroupID, m); err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrIdentity, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-07-12 17:22:45 +08:00
}
}
2021-12-22 15:24:49 +08:00
//only administrators can set group information
2023-01-04 17:22:55 +08:00
var groupInfo imdb.Group
2022-06-28 20:39:27 +08:00
utils.CopyStructFields(&groupInfo, req.GroupInfoForSet)
2022-07-13 11:09:29 +08:00
if req.GroupInfoForSet.Notification != "" {
groupInfo.NotificationUserID = req.OpUserID
groupInfo.NotificationUpdateTime = time.Now()
}
2023-01-09 16:37:33 +08:00
if err := rocksCache.DelGroupInfoFromCache(ctx, req.GroupInfoForSet.GroupID); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-11-22 11:07:20 +08:00
}
2021-12-27 21:08:42 +08:00
err = imdb.SetGroupInfo(groupInfo)
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2021-12-22 15:24:49 +08:00
}
if changedType != 0 {
2022-06-28 20:39:27 +08:00
chat.GroupInfoSetNotification(req.OperationID, req.OpUserID, req.GroupInfoForSet.GroupID, groupName, notification, introduction, faceURL, req.GroupInfoForSet.NeedVerification)
2021-12-22 15:24:49 +08:00
}
2022-06-28 20:39:27 +08:00
if req.GroupInfoForSet.Notification != "" {
2022-06-07 17:45:07 +08:00
//get group member user id
2022-06-28 20:39:27 +08:00
getGroupMemberIDListFromCacheReq := &pbCache.GetGroupMemberIDListFromCacheReq{OperationID: req.OperationID, GroupID: req.GroupInfoForSet.GroupID}
2023-01-09 16:37:33 +08:00
etcdConn, err := utils.GetConn(ctx, config.Config.RpcRegisterName.OpenImCacheName)
2023-01-06 18:25:41 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-06-16 14:51:36 +08:00
}
2022-06-07 17:45:07 +08:00
client := pbCache.NewCacheClient(etcdConn)
2023-01-06 18:25:41 +08:00
cacheResp, err := client.GetGroupMemberIDListFromCache(ctx, getGroupMemberIDListFromCacheReq)
2022-06-07 17:45:07 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-06-07 17:45:07 +08:00
}
2023-01-09 16:37:33 +08:00
if err = constant.CommonResp2Err(cacheResp.CommonResp); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-06-07 17:45:07 +08:00
}
var conversationReq pbConversation.ModifyConversationFieldReq
conversation := pbConversation.Conversation{
OwnerUserID: req.OpUserID,
2022-06-28 20:39:27 +08:00
ConversationID: utils.GetConversationIDBySessionType(req.GroupInfoForSet.GroupID, constant.GroupChatType),
2022-06-07 17:45:07 +08:00
ConversationType: constant.GroupChatType,
2022-06-28 20:39:27 +08:00
GroupID: req.GroupInfoForSet.GroupID,
2022-06-07 17:45:07 +08:00
}
conversationReq.Conversation = &conversation
conversationReq.OperationID = req.OperationID
conversationReq.FieldType = constant.FieldGroupAtType
conversation.GroupAtType = constant.GroupNotification
conversationReq.UserIDList = cacheResp.UserIDList
2023-01-09 16:37:33 +08:00
nEtcdConn, err := utils.GetConn(ctx, config.Config.RpcRegisterName.OpenImConversationName)
2023-01-06 18:25:41 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:25:41 +08:00
return
2022-06-16 14:51:36 +08:00
}
2022-06-07 17:45:07 +08:00
nClient := pbConversation.NewConversationClient(nEtcdConn)
conversationReply, err := nClient.ModifyConversationField(context.Background(), &conversationReq)
2023-01-06 18:25:41 +08:00
trace_log.SetContextInfo(ctx, "ModifyConversationField", err, "req", &conversationReq, "resp", conversationReply)
2022-06-07 17:45:07 +08:00
}
2023-01-06 18:25:41 +08:00
return
2021-12-22 15:24:49 +08:00
}
2023-01-06 18:39:48 +08:00
func (s *groupServer) TransferGroupOwner(ctx context.Context, req *pbGroup.TransferGroupOwnerReq) (resp *pbGroup.TransferGroupOwnerResp, _ error) {
resp = &pbGroup.TransferGroupOwnerResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2021-12-22 15:24:49 +08:00
2022-04-18 10:53:23 +08:00
groupInfo, err := imdb.GetGroupInfoByGroupID(req.GroupID)
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-04-18 10:53:23 +08:00
}
if groupInfo.Status == constant.GroupStatusDismissed {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrDismissedAlready, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-04-18 10:53:23 +08:00
}
2021-12-27 16:48:05 +08:00
if req.OldOwnerUserID == req.NewOwnerUserID {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2021-12-27 16:48:05 +08:00
}
2023-01-09 16:37:33 +08:00
err = rocksCache.DelGroupMemberInfoFromCache(ctx, req.GroupID, req.NewOwnerUserID)
2022-11-22 11:07:20 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-11-22 11:07:20 +08:00
}
2023-01-09 16:37:33 +08:00
err = rocksCache.DelGroupMemberInfoFromCache(ctx, req.GroupID, req.OldOwnerUserID)
2022-11-22 11:07:20 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-11-22 11:07:20 +08:00
}
2023-01-04 17:22:55 +08:00
groupMemberInfo := imdb.GroupMember{GroupID: req.GroupID, UserID: req.OldOwnerUserID, RoleLevel: constant.GroupOrdinaryUsers}
2022-04-18 13:01:16 +08:00
err = imdb.UpdateGroupMemberInfo(groupMemberInfo)
2021-12-27 16:48:05 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2021-12-27 16:48:05 +08:00
}
2023-01-04 17:22:55 +08:00
groupMemberInfo = imdb.GroupMember{GroupID: req.GroupID, UserID: req.NewOwnerUserID, RoleLevel: constant.GroupOwner}
2021-12-27 16:48:05 +08:00
err = imdb.UpdateGroupMemberInfo(groupMemberInfo)
2021-12-22 15:24:49 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2021-12-22 15:24:49 +08:00
}
2022-01-16 16:31:28 +08:00
chat.GroupOwnerTransferredNotification(req)
2023-01-06 18:39:48 +08:00
return
2021-12-22 15:24:49 +08:00
}
2022-01-25 19:18:04 +08:00
2023-01-06 18:39:48 +08:00
func (s *groupServer) GetGroups(ctx context.Context, req *pbGroup.GetGroupsReq) (resp *pbGroup.GetGroupsResp, err error) {
resp = &pbGroup.GetGroupsResp{
CommonResp: &open_im_sdk.CommonResp{},
2023-01-09 16:37:33 +08:00
Groups: []*pbGroup.CMSGroup{},
2022-08-30 16:53:15 +08:00
Pagination: &open_im_sdk.ResponsePagination{CurrentPage: req.Pagination.PageNumber, ShowNumber: req.Pagination.ShowNumber},
2022-01-25 19:18:04 +08:00
}
2023-01-06 18:39:48 +08:00
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2022-08-30 01:38:23 +08:00
if req.GroupID != "" {
groupInfoDB, err := imdb.GetGroupInfoByGroupID(req.GroupID)
2022-02-07 19:42:09 +08:00
if err != nil {
2022-08-30 16:53:15 +08:00
if errors.Is(err, gorm.ErrRecordNotFound) {
return resp, nil
}
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-02-07 19:42:09 +08:00
}
2022-08-30 01:38:23 +08:00
resp.GroupNum = 1
groupInfo := &open_im_sdk.GroupInfo{}
utils.CopyStructFields(groupInfo, groupInfoDB)
2022-08-30 14:17:40 +08:00
groupMember, err := imdb.GetGroupOwnerInfoByGroupID(req.GroupID)
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-08-30 14:17:40 +08:00
}
2022-09-06 16:08:28 +08:00
memberNum, err := imdb.GetGroupMembersCount(req.GroupID, "")
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-09-06 16:08:28 +08:00
}
groupInfo.MemberCount = uint32(memberNum)
2022-09-13 16:19:01 +08:00
groupInfo.CreateTime = uint32(groupInfoDB.CreateTime.Unix())
2023-01-09 16:37:33 +08:00
resp.Groups = append(resp.Groups, &pbGroup.CMSGroup{GroupInfo: groupInfo, GroupOwnerUserName: groupMember.Nickname, GroupOwnerUserID: groupMember.UserID})
2022-08-30 01:38:23 +08:00
} else {
2022-09-13 15:27:00 +08:00
groups, count, err := imdb.GetGroupsByName(req.GroupName, req.Pagination.PageNumber, req.Pagination.ShowNumber)
2022-02-21 21:02:37 +08:00
if err != nil {
2023-01-06 18:39:48 +08:00
trace_log.SetContextInfo(ctx, "GetGroupsByName", err, "GroupName", req.GroupName, "PageNumber", req.Pagination.PageNumber, "ShowNumber", req.Pagination.ShowNumber)
2022-08-30 01:38:23 +08:00
}
for _, v := range groups {
group := &pbGroup.CMSGroup{GroupInfo: &open_im_sdk.GroupInfo{}}
utils.CopyStructFields(group.GroupInfo, v)
groupMember, err := imdb.GetGroupOwnerInfoByGroupID(v.GroupID)
if err != nil {
2023-01-06 18:39:48 +08:00
trace_log.SetContextInfo(ctx, "GetGroupOwnerInfoByGroupID", err, "GroupID", v.GroupID)
2022-08-30 01:38:23 +08:00
continue
}
group.GroupInfo.CreateTime = uint32(v.CreateTime.Unix())
group.GroupOwnerUserID = groupMember.UserID
group.GroupOwnerUserName = groupMember.Nickname
2023-01-09 16:37:33 +08:00
resp.Groups = append(resp.Groups, group)
2022-08-30 01:38:23 +08:00
}
2022-09-13 15:27:00 +08:00
resp.GroupNum = int32(count)
2022-02-21 21:02:37 +08:00
}
2023-01-06 18:39:48 +08:00
return
2022-02-07 19:42:09 +08:00
}
2023-01-06 18:39:48 +08:00
func (s *groupServer) GetGroupMembersCMS(ctx context.Context, req *pbGroup.GetGroupMembersCMSReq) (resp *pbGroup.GetGroupMembersCMSResp, _ error) {
resp = &pbGroup.GetGroupMembersCMSResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2022-08-11 19:47:59 +08:00
groupMembers, err := imdb.GetGroupMembersByGroupIdCMS(req.GroupID, req.UserName, req.Pagination.ShowNumber, req.Pagination.PageNumber)
2022-02-07 19:42:09 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-02-08 20:24:59 +08:00
}
2022-08-11 19:47:59 +08:00
groupMembersCount, err := imdb.GetGroupMembersCount(req.GroupID, req.UserName)
2022-02-08 20:24:59 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-01-26 18:43:01 +08:00
}
2022-02-08 20:24:59 +08:00
log.NewInfo(req.OperationID, groupMembersCount)
2022-07-14 12:08:28 +08:00
resp.MemberNums = int32(groupMembersCount)
2022-02-07 19:42:09 +08:00
for _, groupMember := range groupMembers {
2022-08-11 19:47:59 +08:00
member := open_im_sdk.GroupMemberFullInfo{}
utils.CopyStructFields(&member, groupMember)
2022-08-12 11:24:21 +08:00
member.JoinTime = int32(groupMember.JoinTime.Unix())
member.MuteEndTime = uint32(groupMember.MuteEndTime.Unix())
2022-08-11 19:47:59 +08:00
resp.Members = append(resp.Members, &member)
2022-02-07 19:42:09 +08:00
}
2022-02-08 20:24:59 +08:00
resp.Pagination = &open_im_sdk.ResponsePagination{
CurrentPage: req.Pagination.PageNumber,
ShowNumber: req.Pagination.ShowNumber,
}
2023-01-06 18:39:48 +08:00
return
2022-02-07 19:42:09 +08:00
}
2023-01-06 18:39:48 +08:00
func (s *groupServer) GetUserReqApplicationList(ctx context.Context, req *pbGroup.GetUserReqApplicationListReq) (resp *pbGroup.GetUserReqApplicationListResp, _ error) {
resp = &pbGroup.GetUserReqApplicationListResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2022-02-09 19:20:46 +08:00
groupRequests, err := imdb.GetUserReqGroupByUserID(req.UserID)
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-02-09 19:20:46 +08:00
}
for _, groupReq := range groupRequests {
node := open_im_sdk.GroupRequest{UserInfo: &open_im_sdk.PublicUserInfo{}, GroupInfo: &open_im_sdk.GroupInfo{}}
group, err := imdb.GetGroupInfoByGroupID(groupReq.GroupID)
if err != nil {
2023-01-06 18:39:48 +08:00
trace_log.SetContextInfo(ctx, "GetGroupInfoByGroupID", err, "GroupID", groupReq.GroupID)
2022-02-09 19:20:46 +08:00
continue
}
user, err := imdb.GetUserByUserID(groupReq.UserID)
if err != nil {
2023-01-06 18:39:48 +08:00
trace_log.SetContextInfo(ctx, "GetUserByUserID", err, "UserID", groupReq.UserID)
2022-02-09 19:20:46 +08:00
continue
}
cp.GroupRequestDBCopyOpenIM(&node, &groupReq)
cp.UserDBCopyOpenIMPublicUser(node.UserInfo, user)
cp.GroupDBCopyOpenIM(node.GroupInfo, group)
resp.GroupRequestList = append(resp.GroupRequestList, &node)
}
2023-01-06 18:39:48 +08:00
return
2022-02-14 10:46:32 +08:00
}
2022-03-24 15:33:30 +08:00
2023-01-09 16:37:33 +08:00
func (s *groupServer) DismissGroup(ctx context.Context, req *pbGroup.DismissGroupReq) (resp *pbGroup.DismissGroupResp, _ error) {
2023-01-06 18:39:48 +08:00
resp = &pbGroup.DismissGroupResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2022-04-16 20:10:10 +08:00
if !token_verify.IsManagerUserID(req.OpUserID) && !imdb.IsGroupOwnerAdmin(req.GroupID, req.OpUserID) {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrIdentity, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-03-24 15:33:30 +08:00
}
2023-01-09 16:37:33 +08:00
if err := rocksCache.DelGroupInfoFromCache(ctx, req.GroupID); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-11-22 11:07:20 +08:00
}
2023-01-09 16:37:33 +08:00
if err := s.DelGroupAndUserCache(ctx, req.GroupID, nil); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-11-22 11:07:20 +08:00
}
2022-03-24 15:33:30 +08:00
err := imdb.OperateGroupStatus(req.GroupID, constant.GroupStatusDismissed)
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-03-24 15:33:30 +08:00
}
2022-05-27 18:40:24 +08:00
groupInfo, err := imdb.GetGroupInfoByGroupID(req.GroupID)
2022-04-20 18:47:21 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-04-20 18:47:21 +08:00
}
2022-05-27 18:40:24 +08:00
if groupInfo.GroupType != constant.SuperGroup {
memberList, err := imdb.GetGroupMemberListByGroupID(req.GroupID)
2022-04-20 18:47:21 +08:00
if err != nil {
2023-01-06 18:39:48 +08:00
trace_log.SetContextInfo(ctx, "GetGroupMemberListByGroupID", err, "groupID", req.GroupID)
2022-04-20 18:47:21 +08:00
}
2022-05-27 18:40:24 +08:00
//modify quitter conversation info
var reqPb pbUser.SetConversationReq
2022-08-21 23:37:43 +08:00
var c pbConversation.Conversation
2022-05-27 18:40:24 +08:00
for _, v := range memberList {
reqPb.OperationID = req.OperationID
c.OwnerUserID = v.UserID
c.ConversationID = utils.GetConversationIDBySessionType(req.GroupID, constant.GroupChatType)
c.ConversationType = constant.GroupChatType
c.GroupID = req.GroupID
c.IsNotInGroup = true
reqPb.Conversation = &c
2023-01-09 16:37:33 +08:00
etcdConn, err := utils.GetConn(ctx, config.Config.RpcRegisterName.OpenImUserName)
2022-05-27 18:40:24 +08:00
client := pbUser.NewUserClient(etcdConn)
respPb, err := client.SetConversation(context.Background(), &reqPb)
2023-01-06 18:39:48 +08:00
trace_log.SetContextInfo(ctx, "SetConversation", err, "req", &reqPb, "resp", respPb)
2022-05-27 18:40:24 +08:00
}
err = imdb.DeleteGroupMemberByGroupID(req.GroupID)
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-05-27 18:40:24 +08:00
}
chat.GroupDismissedNotification(req)
} else {
err = db.DB.DeleteSuperGroup(req.GroupID)
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-05-27 18:40:24 +08:00
}
2022-03-24 15:33:30 +08:00
}
2023-01-06 18:39:48 +08:00
return
2022-03-24 15:33:30 +08:00
}
2022-03-29 09:34:45 +08:00
2023-01-09 16:37:33 +08:00
func (s *groupServer) MuteGroupMember(ctx context.Context, req *pbGroup.MuteGroupMemberReq) (resp *pbGroup.MuteGroupMemberResp, _ error) {
2023-01-06 18:39:48 +08:00
resp = &pbGroup.MuteGroupMemberResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2022-06-06 23:19:07 +08:00
opFlag, err := s.getGroupUserLevel(req.GroupID, req.OpUserID)
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-03-29 09:34:45 +08:00
}
2022-06-06 23:19:07 +08:00
if opFlag == 0 {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(constant.ErrNoPermission, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-06-06 23:19:07 +08:00
}
2023-01-09 16:37:33 +08:00
mutedInfo, err := rocksCache.GetGroupMemberInfoFromCache(ctx, req.GroupID, req.UserID)
2022-06-06 23:19:07 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-06-06 23:19:07 +08:00
}
if mutedInfo.RoleLevel == constant.GroupOwner && opFlag != 1 {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-06-06 23:19:07 +08:00
}
if mutedInfo.RoleLevel == constant.GroupAdmin && opFlag == 3 {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-06-06 23:19:07 +08:00
}
2023-01-09 16:37:33 +08:00
if err := rocksCache.DelGroupMemberInfoFromCache(ctx, req.GroupID, req.UserID); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-11-22 11:07:20 +08:00
}
2023-01-04 17:22:55 +08:00
groupMemberInfo := imdb.GroupMember{GroupID: req.GroupID, UserID: req.UserID}
2022-03-29 17:03:06 +08:00
groupMemberInfo.MuteEndTime = time.Unix(int64(time.Now().Second())+int64(req.MutedSeconds), time.Now().UnixNano())
2022-06-06 23:19:07 +08:00
err = imdb.UpdateGroupMemberInfo(groupMemberInfo)
2022-03-29 09:34:45 +08:00
if err != nil {
2023-01-09 16:37:33 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-06 18:39:48 +08:00
return
2022-03-29 09:34:45 +08:00
}
2022-03-29 16:04:15 +08:00
chat.GroupMemberMutedNotification(req.OperationID, req.OpUserID, req.GroupID, req.UserID, req.MutedSeconds)
2023-01-06 18:39:48 +08:00
return
2022-03-29 09:34:45 +08:00
}
2023-01-09 10:01:17 +08:00
func (s *groupServer) CancelMuteGroupMember(ctx context.Context, req *pbGroup.CancelMuteGroupMemberReq) (resp *pbGroup.CancelMuteGroupMemberResp, _ error) {
resp = &pbGroup.CancelMuteGroupMemberResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2022-06-06 23:19:07 +08:00
opFlag, err := s.getGroupUserLevel(req.GroupID, req.OpUserID)
if err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 10:01:17 +08:00
return
2022-06-06 23:19:07 +08:00
}
if opFlag == 0 {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(constant.ErrNoPermission, resp.CommonResp)
2023-01-09 10:01:17 +08:00
return
2022-03-29 09:34:45 +08:00
}
2022-06-06 23:19:07 +08:00
mutedInfo, err := imdb.GetGroupMemberInfoByGroupIDAndUserID(req.GroupID, req.UserID)
if err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 10:01:17 +08:00
return
2022-06-06 23:19:07 +08:00
}
if mutedInfo.RoleLevel == constant.GroupOwner && opFlag != 1 {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 10:01:17 +08:00
return
2022-06-06 23:19:07 +08:00
}
if mutedInfo.RoleLevel == constant.GroupAdmin && opFlag == 3 {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 10:01:17 +08:00
return
2022-06-06 23:19:07 +08:00
}
2023-01-09 16:51:26 +08:00
if err := rocksCache.DelGroupMemberInfoFromCache(ctx, req.GroupID, req.UserID); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 10:01:17 +08:00
return
2022-11-22 11:07:20 +08:00
}
2022-06-06 23:19:07 +08:00
2023-01-04 17:22:55 +08:00
groupMemberInfo := imdb.GroupMember{GroupID: req.GroupID, UserID: req.UserID}
2022-03-29 09:34:45 +08:00
groupMemberInfo.MuteEndTime = time.Unix(0, 0)
2022-06-06 23:19:07 +08:00
err = imdb.UpdateGroupMemberInfo(groupMemberInfo)
2022-03-29 09:34:45 +08:00
if err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 10:01:17 +08:00
return
2022-03-29 09:34:45 +08:00
}
2022-03-29 16:04:15 +08:00
chat.GroupMemberCancelMutedNotification(req.OperationID, req.OpUserID, req.GroupID, req.UserID)
2023-01-09 10:01:17 +08:00
return
2022-03-29 09:34:45 +08:00
}
2023-01-09 10:01:17 +08:00
func (s *groupServer) MuteGroup(ctx context.Context, req *pbGroup.MuteGroupReq) (resp *pbGroup.MuteGroupResp, _ error) {
2023-01-09 13:57:30 +08:00
resp = &pbGroup.MuteGroupResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2022-06-06 23:19:07 +08:00
opFlag, err := s.getGroupUserLevel(req.GroupID, req.OpUserID)
if err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 13:57:30 +08:00
return
2022-06-06 23:19:07 +08:00
}
if opFlag == 0 {
2023-01-09 13:57:30 +08:00
//errMsg := req.OperationID + "opFlag == 0 " + req.GroupID + req.OpUserID
//log.Error(req.OperationID, errMsg)
//return &pbGroup.MuteGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: errMsg}}, nil
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(constant.ErrNoPermission, resp.CommonResp)
2023-01-09 13:57:30 +08:00
return
2022-06-06 23:19:07 +08:00
}
2022-06-07 10:20:56 +08:00
//mutedInfo, err := imdb.GetGroupMemberInfoByGroupIDAndUserID(req.GroupID, req.UserID)
//if err != nil {
// errMsg := req.OperationID + " GetGroupMemberInfoByGroupIDAndUserID failed " + req.GroupID + req.OpUserID + err.Error()
// return &pbGroup.MuteGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: errMsg}}, nil
//}
//if mutedInfo.RoleLevel == constant.GroupOwner && opFlag != 1 {
// errMsg := req.OperationID + " mutedInfo.RoleLevel == constant.GroupOwner " + req.GroupID + req.OpUserID + err.Error()
// return &pbGroup.MuteGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: errMsg}}, nil
//}
//if mutedInfo.RoleLevel == constant.GroupAdmin && opFlag == 3 {
// errMsg := req.OperationID + " mutedInfo.RoleLevel == constant.GroupAdmin " + req.GroupID + req.OpUserID + err.Error()
// return &pbGroup.MuteGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: errMsg}}, nil
//}
2023-01-09 16:51:26 +08:00
if err := rocksCache.DelGroupInfoFromCache(ctx, req.GroupID); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 13:57:30 +08:00
return
2022-11-22 11:07:20 +08:00
}
2022-06-06 23:19:07 +08:00
err = imdb.OperateGroupStatus(req.GroupID, constant.GroupStatusMuted)
2022-03-29 11:48:17 +08:00
if err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 13:57:30 +08:00
return
2022-03-29 11:48:17 +08:00
}
2022-11-22 11:07:20 +08:00
2022-03-29 16:04:15 +08:00
chat.GroupMutedNotification(req.OperationID, req.OpUserID, req.GroupID)
2023-01-09 13:57:30 +08:00
return
2022-03-29 09:34:45 +08:00
}
2023-01-09 14:09:07 +08:00
func (s *groupServer) CancelMuteGroup(ctx context.Context, req *pbGroup.CancelMuteGroupReq) (resp *pbGroup.CancelMuteGroupResp, _ error) {
resp = &pbGroup.CancelMuteGroupResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2022-06-06 23:19:07 +08:00
opFlag, err := s.getGroupUserLevel(req.GroupID, req.OpUserID)
if err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:09:07 +08:00
return
2022-06-06 23:19:07 +08:00
}
if opFlag == 0 {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:09:07 +08:00
return
2022-06-06 23:19:07 +08:00
}
2022-06-07 10:13:19 +08:00
//mutedInfo, err := imdb.GetGroupMemberInfoByGroupIDAndUserID(req.GroupID, req.)
//if err != nil {
// errMsg := req.OperationID + " GetGroupMemberInfoByGroupIDAndUserID failed " + req.GroupID + req.OpUserID + err.Error()
// return &pbGroup.CancelMuteGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: errMsg}}, nil
//}
//if mutedInfo.RoleLevel == constant.GroupOwner && opFlag != 1 {
// errMsg := req.OperationID + " mutedInfo.RoleLevel == constant.GroupOwner " + req.GroupID + req.OpUserID + err.Error()
// return &pbGroup.CancelMuteGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: errMsg}}, nil
//}
//if mutedInfo.RoleLevel == constant.GroupAdmin && opFlag == 3 {
// errMsg := req.OperationID + " mutedInfo.RoleLevel == constant.GroupAdmin " + req.GroupID + req.OpUserID + err.Error()
// return &pbGroup.CancelMuteGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: errMsg}}, nil
//}
2022-06-07 18:09:11 +08:00
log.Debug(req.OperationID, "UpdateGroupInfoDefaultZero ", req.GroupID, map[string]interface{}{"status": constant.GroupOk})
2023-01-09 16:51:26 +08:00
if err := rocksCache.DelGroupInfoFromCache(ctx, req.GroupID); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:09:07 +08:00
return
2022-11-22 11:07:20 +08:00
}
2022-06-06 23:19:07 +08:00
err = imdb.UpdateGroupInfoDefaultZero(req.GroupID, map[string]interface{}{"status": constant.GroupOk})
2022-03-29 11:48:17 +08:00
if err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:09:07 +08:00
return
2022-03-29 11:48:17 +08:00
}
2022-03-29 16:04:15 +08:00
chat.GroupCancelMutedNotification(req.OperationID, req.OpUserID, req.GroupID)
2023-01-09 14:09:07 +08:00
return
2022-03-29 09:34:45 +08:00
}
2022-04-18 11:52:17 +08:00
2023-01-09 14:09:07 +08:00
func (s *groupServer) SetGroupMemberNickname(ctx context.Context, req *pbGroup.SetGroupMemberNicknameReq) (resp *pbGroup.SetGroupMemberNicknameResp, _ error) {
resp = &pbGroup.SetGroupMemberNicknameResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2022-04-18 11:52:17 +08:00
if req.OpUserID != req.UserID && !token_verify.IsManagerUserID(req.OpUserID) {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(constant.ErrIdentity, resp.CommonResp)
2023-01-09 14:09:07 +08:00
return
2022-04-18 11:52:17 +08:00
}
2022-12-16 16:54:18 +08:00
cbReq := &pbGroup.SetGroupMemberInfoReq{
GroupID: req.GroupID,
UserID: req.UserID,
OperationID: req.OperationID,
OpUserID: req.OpUserID,
Nickname: &wrapperspb.StringValue{Value: req.Nickname},
}
2023-01-09 14:09:07 +08:00
if err := CallbackBeforeSetGroupMemberInfo(ctx, cbReq); err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:09:07 +08:00
return
2022-12-16 16:54:18 +08:00
}
nickName := cbReq.Nickname.Value
2023-01-04 17:22:55 +08:00
groupMemberInfo := imdb.GroupMember{}
2022-04-18 11:52:17 +08:00
groupMemberInfo.UserID = req.UserID
groupMemberInfo.GroupID = req.GroupID
2022-12-16 16:54:18 +08:00
if nickName == "" {
2022-04-18 11:52:17 +08:00
userNickname, err := imdb.GetUserNameByUserID(groupMemberInfo.UserID)
if err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:09:07 +08:00
return
2022-04-18 11:52:17 +08:00
}
groupMemberInfo.Nickname = userNickname
} else {
2022-12-16 16:54:18 +08:00
groupMemberInfo.Nickname = nickName
2022-04-18 11:52:17 +08:00
}
2022-11-22 11:07:20 +08:00
2023-01-09 16:51:26 +08:00
if err := rocksCache.DelGroupMemberInfoFromCache(ctx, req.GroupID, req.UserID); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:09:07 +08:00
return
2022-11-22 11:07:20 +08:00
}
2023-01-09 14:09:07 +08:00
if err := imdb.UpdateGroupMemberInfo(groupMemberInfo); err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:09:07 +08:00
return
2022-04-18 11:52:17 +08:00
}
chat.GroupMemberInfoSetNotification(req.OperationID, req.OpUserID, req.GroupID, req.UserID)
2023-01-09 14:09:07 +08:00
return
2022-04-18 11:52:17 +08:00
}
2022-06-02 18:17:11 +08:00
2023-01-09 14:09:07 +08:00
func (s *groupServer) SetGroupMemberInfo(ctx context.Context, req *pbGroup.SetGroupMemberInfoReq) (resp *pbGroup.SetGroupMemberInfoResp, _ error) {
resp = &pbGroup.SetGroupMemberInfoResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2023-01-09 16:51:26 +08:00
if err := rocksCache.DelGroupMemberInfoFromCache(ctx, req.GroupID, req.UserID); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:09:07 +08:00
return
2022-12-16 16:54:18 +08:00
}
2023-01-09 14:09:07 +08:00
if err := CallbackBeforeSetGroupMemberInfo(ctx, req); err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:09:07 +08:00
return
2022-12-16 16:54:18 +08:00
}
2023-01-04 17:22:55 +08:00
groupMember := imdb.GroupMember{
2022-06-07 16:18:18 +08:00
GroupID: req.GroupID,
UserID: req.UserID,
}
m := make(map[string]interface{})
if req.RoleLevel != nil {
m["role_level"] = req.RoleLevel.Value
}
if req.FaceURL != nil {
m["user_group_face_url"] = req.FaceURL.Value
}
if req.Nickname != nil {
m["nickname"] = req.Nickname.Value
}
if req.Ex != nil {
m["ex"] = req.Ex.Value
2022-12-09 13:18:36 +08:00
} else {
2022-12-12 14:46:49 +08:00
m["ex"] = nil
2022-06-07 16:18:18 +08:00
}
2023-01-09 16:51:26 +08:00
if err := imdb.UpdateGroupMemberInfoByMap(groupMember, m); err != nil {
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:09:07 +08:00
return
2022-06-07 16:18:18 +08:00
}
2022-06-08 15:05:57 +08:00
if req.RoleLevel != nil {
switch req.RoleLevel.Value {
case constant.GroupOrdinaryUsers:
2022-07-20 21:26:52 +08:00
//msg.GroupMemberRoleLevelChangeNotification(req.OperationID, req.OpUserID, req.GroupID, req.UserID, constant.GroupMemberSetToOrdinaryUserNotification)
2022-06-08 15:40:33 +08:00
chat.GroupMemberInfoSetNotification(req.OperationID, req.OpUserID, req.GroupID, req.UserID)
2022-06-08 15:05:57 +08:00
case constant.GroupAdmin, constant.GroupOwner:
2022-07-20 21:26:52 +08:00
//msg.GroupMemberRoleLevelChangeNotification(req.OperationID, req.OpUserID, req.GroupID, req.UserID, constant.GroupMemberSetToAdminNotification)
2022-06-08 15:40:33 +08:00
chat.GroupMemberInfoSetNotification(req.OperationID, req.OpUserID, req.GroupID, req.UserID)
2022-06-08 15:05:57 +08:00
}
} else {
chat.GroupMemberInfoSetNotification(req.OperationID, req.OpUserID, req.GroupID, req.UserID)
}
2023-01-09 14:09:07 +08:00
return
2022-06-02 18:17:11 +08:00
}
2022-08-08 11:30:10 +08:00
2023-01-09 16:51:26 +08:00
func (s *groupServer) GetGroupAbstractInfo(ctx context.Context, req *pbGroup.GetGroupAbstractInfoReq) (resp *pbGroup.GetGroupAbstractInfoResp, _ error) {
2023-01-09 14:28:19 +08:00
resp = &pbGroup.GetGroupAbstractInfoResp{CommonResp: &open_im_sdk.CommonResp{}}
ctx = trace_log.NewRpcCtx(ctx, utils.GetSelfFuncName(), req.OperationID)
defer func() {
trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), nil, "rpc req ", req.String(), "rpc resp ", resp.String())
trace_log.ShowLog(ctx)
}()
2023-01-09 16:51:26 +08:00
hashCode, err := rocksCache.GetGroupMemberListHashFromCache(ctx, req.GroupID)
2022-08-08 11:30:10 +08:00
if err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:28:19 +08:00
return
2022-08-08 11:30:10 +08:00
}
resp.GroupMemberListHash = hashCode
2023-01-09 14:27:08 +08:00
num, err := rocksCache.GetGroupMemberNumFromCache(ctx, req.GroupID)
2022-08-08 11:30:10 +08:00
if err != nil {
2023-01-09 16:51:26 +08:00
constant.SetErrorForResp(err, resp.CommonResp)
2023-01-09 14:28:19 +08:00
return
2022-08-08 11:30:10 +08:00
}
resp.GroupMemberNumber = int32(num)
return resp, nil
}
2022-11-22 11:07:20 +08:00
2023-01-09 16:37:33 +08:00
func (s *groupServer) DelGroupAndUserCache(ctx context.Context, groupID string, userIDList []string) error {
operationID := trace_log.GetOperationID(ctx)
2022-11-22 11:07:20 +08:00
if groupID != "" {
2023-01-09 16:37:33 +08:00
etcdConn, err := utils.GetConn(ctx, config.Config.RpcRegisterName.OpenImCacheName)
if err != nil {
return err
2022-11-22 11:07:20 +08:00
}
cacheClient := pbCache.NewCacheClient(etcdConn)
cacheResp, err := cacheClient.DelGroupMemberIDListFromCache(context.Background(), &pbCache.DelGroupMemberIDListFromCacheReq{
GroupID: groupID,
OperationID: operationID,
})
if err != nil {
log.NewError(operationID, "DelGroupMemberIDListFromCache rpc call failed ", err.Error())
return utils.Wrap(err, "")
}
2023-01-09 16:37:33 +08:00
err = constant.CommonResp2Err(cacheResp.CommonResp)
err = rocksCache.DelGroupMemberListHashFromCache(ctx, groupID)
2022-11-22 11:07:20 +08:00
if err != nil {
log.NewError(operationID, utils.GetSelfFuncName(), groupID, err.Error())
return utils.Wrap(err, "")
}
2023-01-09 16:37:33 +08:00
err = rocksCache.DelGroupMemberNumFromCache(ctx, groupID)
2022-11-22 11:07:20 +08:00
if err != nil {
log.NewError(operationID, utils.GetSelfFuncName(), err.Error(), groupID)
return utils.Wrap(err, "")
}
}
if userIDList != nil {
for _, userID := range userIDList {
2023-01-09 16:37:33 +08:00
err := rocksCache.DelJoinedGroupIDListFromCache(ctx, userID)
2022-11-22 11:07:20 +08:00
if err != nil {
log.NewError(operationID, utils.GetSelfFuncName(), err.Error())
return utils.Wrap(err, "")
}
}
}
return nil
}