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

587 lines
23 KiB
Go
Raw Normal View History

2021-06-28 15:36:09 +08:00
package group
import (
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"
2021-12-22 15:24:49 +08:00
"Open_IM/pkg/common/db/mysql_model/im_mysql_model"
imdb "Open_IM/pkg/common/db/mysql_model/im_mysql_model"
"Open_IM/pkg/common/log"
2021-12-22 15:24:49 +08:00
"Open_IM/pkg/common/token_verify"
"Open_IM/pkg/grpc-etcdv3/getcdv3"
2021-10-11 22:00:38 +08:00
pbGroup "Open_IM/pkg/proto/group"
2021-12-24 15:02:47 +08:00
open_im_sdk "Open_IM/pkg/proto/sdk_ws"
"Open_IM/pkg/utils"
2021-06-28 15:36:09 +08:00
"context"
2021-12-22 15:24:49 +08:00
"google.golang.org/grpc"
"net"
"strconv"
"strings"
"time"
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 {
log.NewPrivateLog("group")
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() {
2021-12-27 21:08:42 +08:00
log.NewInfo("0", "group rpc start ")
2021-12-22 15:24:49 +08:00
ip := utils.ServerIP
registerAddress := ip + ":" + strconv.Itoa(s.rpcPort)
//listener network
listener, err := net.Listen("tcp", registerAddress)
if err != nil {
2021-12-27 21:08:42 +08:00
log.NewError("0", "Listen failed ", err.Error(), registerAddress)
2021-12-22 15:24:49 +08:00
return
}
2021-12-27 21:08:42 +08:00
log.NewInfo("0", "listen network success, ", registerAddress, listener)
2021-12-22 15:24:49 +08:00
defer listener.Close()
//grpc server
srv := grpc.NewServer()
defer srv.GracefulStop()
//Service registers with etcd
pbGroup.RegisterGroupServer(srv, s)
err = getcdv3.RegisterEtcd(s.etcdSchema, strings.Join(s.etcdAddr, ","), ip, s.rpcPort, s.rpcRegisterName, 10)
if err != nil {
2021-12-27 21:08:42 +08:00
log.NewError("0", "RegisterEtcd failed ", err.Error())
2021-12-22 15:24:49 +08:00
return
}
err = srv.Serve(listener)
if err != nil {
2021-12-27 21:08:42 +08:00
log.NewError("0", "Serve failed ", err.Error())
2021-12-22 15:24:49 +08:00
return
}
2021-12-27 21:08:42 +08:00
log.NewInfo("0", "group rpc success")
2021-12-22 15:24:49 +08:00
}
func (s *groupServer) CreateGroup(ctx context.Context, req *pbGroup.CreateGroupReq) (*pbGroup.CreateGroupResp, error) {
2021-12-23 17:22:49 +08:00
log.NewInfo(req.OperationID, "CreateGroup, args ", req.String())
2021-12-26 18:47:11 +08:00
if !token_verify.CheckAccess(req.OpUserID, req.OwnerUserID) {
log.NewError(req.OperationID, "CheckAccess false ", req.OpUserID, req.OwnerUserID)
return &pbGroup.CreateGroupResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: constant.ErrAccess.ErrMsg}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-28 11:57:12 +08:00
2021-12-22 15:24:49 +08:00
//Time stamp + MD5 to generate group chat id
2021-12-23 17:22:49 +08:00
groupId := utils.Md5(strconv.FormatInt(time.Now().UnixNano(), 10))
2021-12-24 15:02:47 +08:00
//to group
2021-12-28 11:57:12 +08:00
groupInfo := imdb.Group{GroupID: groupId}
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
2021-12-27 21:08:42 +08:00
err := im_mysql_model.InsertIntoGroup(groupInfo)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-28 11:57:12 +08:00
log.NewError(req.OperationID, "InsertIntoGroup failed, ", err.Error(), groupInfo)
2021-12-26 18:47:11 +08:00
return &pbGroup.CreateGroupResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrDB.ErrMsg}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-27 21:08:42 +08:00
us, err := imdb.GetUserByUserID(req.OwnerUserID)
2021-12-23 17:22:49 +08:00
if err != nil {
2021-12-28 11:57:12 +08:00
log.NewError(req.OperationID, "GetUserByUserID failed ", err.Error(), req.OwnerUserID)
2021-12-26 18:47:11 +08:00
return &pbGroup.CreateGroupResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrDB.ErrMsg}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-24 15:02:47 +08:00
//to group member
2021-12-28 21:17:41 +08:00
groupMember := imdb.GroupMember{GroupID: groupId, RoleLevel: constant.GroupOwner}
2021-12-27 21:08:42 +08:00
utils.CopyStructFields(&groupMember, us)
err = im_mysql_model.InsertIntoGroupMember(groupMember)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-28 11:57:12 +08:00
log.NewError(req.OperationID, "InsertIntoGroupMember failed ", err.Error(), groupMember)
2021-12-26 18:47:11 +08:00
return &pbGroup.CreateGroupResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrDB.ErrMsg}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-26 18:47:11 +08:00
err = db.DB.AddGroupMember(groupId, req.OwnerUserID)
2021-12-23 17:22:49 +08:00
if err != nil {
2021-12-26 18:47:11 +08:00
log.NewError(req.OperationID, "AddGroupMember failed ", err.Error(), groupId, req.OwnerUserID)
2021-12-22 15:24:49 +08:00
}
2021-12-28 11:57:12 +08:00
var okUserIDList []string
2021-12-24 15:02:47 +08:00
//to group member
2021-12-23 17:22:49 +08:00
for _, user := range req.InitMemberList {
2021-12-27 21:08:42 +08:00
us, err := im_mysql_model.GetUserByUserID(user.UserID)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "FindUserByUID failed ", err.Error(), user.UserID)
2021-12-22 15:24:49 +08:00
continue
}
2021-12-28 21:17:41 +08:00
if user.RoleLevel == constant.GroupOwner {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "only one owner, failed ", user)
continue
}
2021-12-28 11:57:12 +08:00
groupMember.RoleLevel = user.RoleLevel
2021-12-27 21:08:42 +08:00
utils.CopyStructFields(&groupMember, us)
err = im_mysql_model.InsertIntoGroupMember(groupMember)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-27 21:08:42 +08:00
log.NewError(req.OperationID, "InsertIntoGroupMember failed ", err.Error(), groupMember)
2021-12-28 11:57:12 +08:00
continue
2021-12-22 15:24:49 +08:00
}
2021-12-28 11:57:12 +08:00
2021-12-24 16:02:21 +08:00
okUserIDList = append(okUserIDList, user.UserID)
2021-12-23 17:22:49 +08:00
err = db.DB.AddGroupMember(groupId, user.UserID)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "add mongo group member failed, db.DB.AddGroupMember failed ", err.Error())
2021-12-22 15:24:49 +08:00
}
}
2021-12-24 15:02:47 +08:00
resp := &pbGroup.CreateGroupResp{}
2021-12-27 21:08:42 +08:00
group, err := im_mysql_model.GetGroupInfoByGroupID(groupId)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-27 21:08:42 +08:00
log.NewError(req.OperationID, "GetGroupInfoByGroupID failed ", err.Error(), groupId)
2021-12-26 18:47:11 +08:00
resp.ErrCode = constant.ErrDB.ErrCode
resp.ErrMsg = constant.ErrDB.ErrMsg
2021-12-23 17:22:49 +08:00
return resp, nil
}
2021-12-26 18:47:11 +08:00
chat.GroupCreatedNotification(req.OperationID, req.OpUserID, req.OwnerUserID, groupId, okUserIDList)
2021-12-23 17:22:49 +08:00
utils.CopyStructFields(resp.GroupInfo, group)
2021-12-28 11:57:12 +08:00
resp.GroupInfo.MemberCount = uint32(imdb.GetGroupMemberNumByGroupID(groupId))
resp.GroupInfo.OwnerUserID = req.OwnerUserID
2021-12-23 17:22:49 +08:00
log.NewInfo(req.OperationID, "rpc CreateGroup return ", resp.String())
return resp, nil
2021-12-22 15:24:49 +08:00
}
2021-06-28 15:36:09 +08:00
func (s *groupServer) GetJoinedGroupList(ctx context.Context, req *pbGroup.GetJoinedGroupListReq) (*pbGroup.GetJoinedGroupListResp, error) {
2021-12-23 17:22:49 +08:00
log.NewInfo(req.OperationID, "GetJoinedGroupList, args ", req.String())
if !token_verify.CheckAccess(req.OpUserID, req.FromUserID) {
log.NewError(req.OperationID, "CheckAccess false ", req.OpUserID, req.FromUserID)
2021-12-26 18:47:11 +08:00
return &pbGroup.GetJoinedGroupListResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: constant.ErrAccess.ErrMsg}, nil
2021-06-28 15:36:09 +08:00
}
2021-12-24 15:02:47 +08:00
//group list
2021-12-27 21:08:42 +08:00
joinedGroupList, err := imdb.GetJoinedGroupIDListByUserID(req.FromUserID)
2021-06-28 15:36:09 +08:00
if err != nil {
2021-12-27 21:08:42 +08:00
log.NewError(req.OperationID, "GetJoinedGroupIDListByUserID failed ", err.Error(), req.FromUserID)
2021-12-26 18:47:11 +08:00
return &pbGroup.GetJoinedGroupListResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrDB.ErrMsg}, nil
2021-06-28 15:36:09 +08:00
}
var resp pbGroup.GetJoinedGroupListResp
for _, v := range joinedGroupList {
2021-12-23 17:22:49 +08:00
var groupNode open_im_sdk.GroupInfo
2021-12-27 21:08:42 +08:00
num := imdb.GetGroupMemberNumByGroupID(v)
owner, err2 := imdb.GetGroupOwnerInfoByGroupID(v)
group, err := imdb.GetGroupInfoByGroupID(v)
2021-12-23 17:22:49 +08:00
if num > 0 && owner != nil && err2 == nil && group != nil && err == nil {
utils.CopyStructFields(&groupNode, group)
2021-12-28 15:33:47 +08:00
groupNode.CreateTime = group.CreateTime.Unix()
2021-12-23 17:22:49 +08:00
groupNode.MemberCount = uint32(num)
2021-12-28 11:57:12 +08:00
groupNode.OwnerUserID = owner.UserID
2021-06-28 15:36:09 +08:00
resp.GroupList = append(resp.GroupList, &groupNode)
2021-12-23 17:22:49 +08:00
} else {
log.NewError(req.OperationID, "check nil ", num, owner, err, group)
continue
2021-06-28 15:36:09 +08:00
}
2021-12-23 17:22:49 +08:00
log.NewDebug(req.OperationID, "joinedGroup ", groupNode)
2021-06-28 15:36:09 +08:00
}
2021-12-28 11:57:12 +08:00
log.NewInfo(req.OperationID, "GetJoinedGroupList rpc return ", resp.String())
2021-06-28 15:36:09 +08:00
return &resp, nil
}
func (s *groupServer) InviteUserToGroup(ctx context.Context, req *pbGroup.InviteUserToGroupReq) (*pbGroup.InviteUserToGroupResp, error) {
2021-12-23 17:22:49 +08:00
log.NewInfo(req.OperationID, "InviteUserToGroup args ", req.String())
2021-12-24 15:02:47 +08:00
if !imdb.IsExistGroupMember(req.GroupID, req.OpUserID) && !token_verify.IsMangerUserID(req.OpUserID) {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "no permission InviteUserToGroup ", req.GroupID, req.OpUserID)
2021-12-22 15:24:49 +08:00
return &pbGroup.InviteUserToGroupResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: constant.ErrAccess.ErrMsg}, nil
2021-07-15 11:16:24 +08:00
}
2021-12-28 11:57:12 +08:00
_, err := imdb.GetGroupInfoByGroupID(req.GroupID)
if err != nil {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "FindGroupInfoByGroupId failed ", req.GroupID, err)
2021-12-22 15:24:49 +08:00
return &pbGroup.InviteUserToGroupResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: constant.ErrAccess.ErrMsg}, nil
}
2021-06-28 15:36:09 +08:00
//
//from User: invite: applicant
//to user: invite: invited
var resp pbGroup.InviteUserToGroupResp
2021-12-24 15:02:47 +08:00
var okUserIDList []string
2021-12-23 17:22:49 +08:00
for _, v := range req.InvitedUserIDList {
2021-06-28 15:36:09 +08:00
var resultNode pbGroup.Id2Result
2021-12-23 17:22:49 +08:00
resultNode.UserID = v
2021-06-28 15:36:09 +08:00
resultNode.Result = 0
2021-12-27 21:08:42 +08:00
toUserInfo, err := imdb.GetUserByUserID(v)
2021-06-28 15:36:09 +08:00
if err != nil {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "FindUserByUID failed ", err.Error(), v)
2021-06-28 15:36:09 +08:00
resultNode.Result = -1
2021-12-23 17:22:49 +08:00
resp.Id2ResultList = append(resp.Id2ResultList, &resultNode)
2021-06-28 15:36:09 +08:00
continue
}
2021-07-15 11:16:24 +08:00
if imdb.IsExistGroupMember(req.GroupID, v) {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "IsExistGroupMember ", req.GroupID, v)
2021-06-28 15:36:09 +08:00
resultNode.Result = -1
2021-12-23 17:22:49 +08:00
resp.Id2ResultList = append(resp.Id2ResultList, &resultNode)
2021-06-28 15:36:09 +08:00
continue
}
2021-12-27 21:08:42 +08:00
var toInsertInfo imdb.GroupMember
utils.CopyStructFields(&toInsertInfo, toUserInfo)
toInsertInfo.GroupID = req.GroupID
2021-12-28 21:17:41 +08:00
toInsertInfo.RoleLevel = constant.GroupOrdinaryUsers
2021-12-27 21:08:42 +08:00
err = imdb.InsertIntoGroupMember(toInsertInfo)
2021-06-28 15:36:09 +08:00
if err != nil {
2021-12-27 21:08:42 +08:00
log.NewError(req.OperationID, "InsertIntoGroupMember failed ", req.GroupID, toUserInfo.UserID, toUserInfo.Nickname, toUserInfo.FaceUrl)
2021-06-28 15:36:09 +08:00
resultNode.Result = -1
2021-12-23 17:22:49 +08:00
resp.Id2ResultList = append(resp.Id2ResultList, &resultNode)
2021-06-28 15:36:09 +08:00
continue
}
2021-12-24 15:02:47 +08:00
okUserIDList = append(okUserIDList, v)
2021-12-22 15:24:49 +08:00
err = db.DB.AddGroupMember(req.GroupID, toUserInfo.UserID)
2021-07-15 11:16:24 +08:00
if err != nil {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "AddGroupMember failed ", err.Error(), req.GroupID, toUserInfo.UserID)
2021-07-15 11:16: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
}
2021-12-24 15:02:47 +08:00
chat.MemberInvitedNotification(req.OperationID, req.GroupID, req.OpUserID, req.Reason, okUserIDList)
2021-12-22 15:24:49 +08:00
resp.ErrCode = 0
2021-12-24 15:02:47 +08:00
log.NewInfo(req.OperationID, "InviteUserToGroup rpc return ", resp.String())
2021-06-28 15:36:09 +08:00
return &resp, nil
}
func (s *groupServer) GetGroupAllMember(ctx context.Context, req *pbGroup.GetGroupAllMemberReq) (*pbGroup.GetGroupAllMemberResp, error) {
2021-12-23 17:22:49 +08:00
log.NewInfo(req.OperationID, "GetGroupAllMember, args ", req.String())
2021-06-28 15:36:09 +08:00
var resp pbGroup.GetGroupAllMemberResp
2021-12-27 21:08:42 +08:00
memberList, err := imdb.GetGroupMemberListByGroupID(req.GroupID)
2021-06-28 15:36:09 +08:00
if err != nil {
2021-12-26 18:47:11 +08:00
resp.ErrCode = constant.ErrDB.ErrCode
resp.ErrMsg = constant.ErrDB.ErrMsg
log.NewError(req.OperationID, "FindGroupMemberListByGroupId failed,", err.Error(), req.GroupID)
2021-06-28 15:36:09 +08:00
return &resp, nil
}
2021-12-28 11:57:12 +08:00
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
utils.CopyStructFields(node, v)
2021-06-28 15:36:09 +08:00
resp.MemberList = append(resp.MemberList, &node)
2021-12-23 17:22:49 +08:00
}
2021-12-28 11:57:12 +08:00
log.NewInfo(req.OperationID, "GetGroupAllMember rpc return ", resp.String())
2021-06-28 15:36:09 +08:00
return &resp, nil
}
func (s *groupServer) GetGroupMemberList(ctx context.Context, req *pbGroup.GetGroupMemberListReq) (*pbGroup.GetGroupMemberListResp, error) {
2021-12-23 17:22:49 +08:00
log.NewInfo(req.OperationID, "GetGroupMemberList, args ", req.String())
2021-06-28 15:36:09 +08:00
var resp pbGroup.GetGroupMemberListResp
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 {
2021-12-26 18:47:11 +08:00
resp.ErrCode = constant.ErrDB.ErrCode
2021-12-28 11:57:12 +08:00
resp.ErrMsg = constant.ErrDB.ErrMsg
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "GetGroupMemberByGroupId failed,", req.GroupID, req.Filter, req.NextSeq, 30)
2021-06-28 15:36:09 +08:00
return &resp, nil
}
for _, v := range memberList {
2021-12-23 17:22:49 +08:00
var node open_im_sdk.GroupMemberFullInfo
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))
}
2021-12-22 15:24:49 +08:00
resp.ErrCode = 0
2021-12-24 16:02:21 +08:00
log.NewInfo(req.OperationID, "GetGroupMemberList rpc return ", resp.String())
2021-06-28 15:36:09 +08:00
return &resp, nil
}
func (s *groupServer) KickGroupMember(ctx context.Context, req *pbGroup.KickGroupMemberReq) (*pbGroup.KickGroupMemberResp, error) {
2021-12-23 17:22:49 +08:00
log.NewInfo(req.OperationID, "KickGroupMember args ", req.String())
2021-12-27 21:08:42 +08:00
ownerList, err := imdb.GetOwnerManagerByGroupID(req.GroupID)
2021-06-28 15:36:09 +08:00
if err != nil {
2021-12-22 15:24:49 +08:00
log.NewError(req.OperationID, "GetOwnerManagerByGroupId failed ", err.Error(), req.GroupID)
2021-12-26 18:47:11 +08:00
return &pbGroup.KickGroupMemberResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: constant.ErrAccess.ErrMsg}, nil
2021-06-28 15:36:09 +08:00
}
//op is group owner?
var flag = 0
for _, v := range ownerList {
2021-12-22 15:24:49 +08:00
if v.UserID == req.OpUserID {
2021-06-28 15:36:09 +08:00
flag = 1
2021-12-23 17:22:49 +08:00
log.NewDebug(req.OperationID, "is group owner ", req.OpUserID, req.GroupID)
2021-06-28 15:36:09 +08:00
break
}
}
2021-12-22 15:24:49 +08:00
2021-12-24 15:02:47 +08:00
//op is app manager
2021-09-22 20:10:38 +08:00
if flag != 1 {
2021-12-23 17:22:49 +08:00
if token_verify.IsMangerUserID(req.OpUserID) {
2021-09-22 20:10:38 +08:00
flag = 1
2021-12-23 17:22:49 +08:00
log.NewDebug(req.OperationID, "is app manager ", req.OpUserID)
2021-09-22 20:10:38 +08:00
}
}
2021-06-28 15:36:09 +08:00
if flag != 1 {
2021-12-28 11:57:12 +08:00
log.NewError(req.OperationID, "failed, no access kick ", req.OpUserID)
2021-12-22 15:24:49 +08:00
return &pbGroup.KickGroupMemberResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: constant.ErrAccess.ErrMsg}, nil
2021-06-28 15:36:09 +08:00
}
2021-12-23 17:22:49 +08:00
if len(req.KickedUserIDList) == 0 {
2021-12-22 15:24:49 +08:00
log.NewError(req.OperationID, "failed, kick list 0")
2021-12-26 18:47:11 +08:00
return &pbGroup.KickGroupMemberResp{ErrCode: constant.ErrArgs.ErrCode, ErrMsg: constant.ErrArgs.ErrMsg}, nil
2021-06-28 15:36:09 +08:00
}
2021-12-22 15:24:49 +08:00
2021-12-23 17:22:49 +08:00
groupOwnerUserID := ""
for _, v := range ownerList {
2021-12-28 21:17:41 +08:00
if v.RoleLevel == constant.GroupOwner {
2021-12-23 17:22:49 +08:00
groupOwnerUserID = v.UserID
}
}
2021-12-24 15:02:47 +08:00
var okUserIDList []string
2021-06-28 15:36:09 +08:00
//remove
var resp pbGroup.KickGroupMemberResp
2021-12-23 17:22:49 +08:00
for _, v := range req.KickedUserIDList {
2021-06-28 15:36:09 +08:00
//owner cant kicked
2021-12-23 17:22:49 +08:00
if v == groupOwnerUserID {
log.NewError(req.OperationID, "failed, can't kick owner ", v)
resp.Id2ResultList = append(resp.Id2ResultList, &pbGroup.Id2Result{UserID: v, Result: -1})
2021-06-28 15:36:09 +08:00
continue
}
2021-12-23 17:22:49 +08:00
err := imdb.RemoveGroupMember(req.GroupID, v)
2021-06-28 15:36:09 +08:00
if err != nil {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "RemoveGroupMember failed ", err.Error(), req.GroupID, v)
resp.Id2ResultList = append(resp.Id2ResultList, &pbGroup.Id2Result{UserID: v, Result: -1})
2021-06-28 15:36:09 +08:00
} else {
2021-12-23 17:22:49 +08:00
resp.Id2ResultList = append(resp.Id2ResultList, &pbGroup.Id2Result{UserID: v, Result: 0})
2021-12-24 15:02:47 +08:00
okUserIDList = append(okUserIDList, v)
2021-06-28 15:36:09 +08:00
}
2021-12-24 15:02:47 +08:00
2021-12-23 17:22:49 +08:00
err = db.DB.DelGroupMember(req.GroupID, v)
2021-07-15 11:16:24 +08:00
if err != nil {
2021-12-24 15:02:47 +08:00
log.NewError(req.OperationID, "DelGroupMember failed ", err.Error(), req.GroupID, v)
2021-07-15 11:16:24 +08:00
}
2021-06-28 15:36:09 +08:00
}
2021-12-24 15:02:47 +08:00
chat.MemberKickedNotification(req, okUserIDList)
log.NewInfo(req.OperationID, "GetGroupMemberList rpc return ", resp.String())
2021-06-28 15:36:09 +08:00
return &resp, nil
}
func (s *groupServer) GetGroupMembersInfo(ctx context.Context, req *pbGroup.GetGroupMembersInfoReq) (*pbGroup.GetGroupMembersInfoResp, error) {
2021-12-23 17:22:49 +08:00
log.NewInfo(req.OperationID, "GetGroupMembersInfo args ", req.String())
2021-06-28 15:36:09 +08:00
var resp pbGroup.GetGroupMembersInfoResp
for _, v := range req.MemberList {
2021-12-23 17:22:49 +08:00
var memberNode open_im_sdk.GroupMemberFullInfo
2021-12-27 21:08:42 +08:00
memberInfo, err := imdb.GetMemberInfoByID(req.GroupID, v)
2021-12-23 17:22:49 +08:00
memberNode.UserID = v
2021-06-28 15:36:09 +08:00
if err != nil {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "GetMemberInfoById failed ", err.Error(), req.GroupID, v)
2021-06-28 15:36:09 +08:00
continue
2021-12-23 17:22:49 +08:00
} else {
utils.CopyStructFields(&memberNode, memberInfo)
2021-12-28 15:33:47 +08:00
memberNode.JoinTime = memberInfo.JoinTime.Unix()
2021-12-23 17:22:49 +08:00
resp.MemberList = append(resp.MemberList, &memberNode)
2021-06-28 15:36:09 +08:00
}
}
2021-12-22 15:24:49 +08:00
resp.ErrCode = 0
2021-12-24 15:02:47 +08:00
log.NewInfo(req.OperationID, "GetGroupMembersInfo rpc return ", resp.String())
2021-06-28 15:36:09 +08:00
return &resp, nil
}
2021-12-22 15:24:49 +08:00
2021-12-23 17:22:49 +08:00
func (s *groupServer) GetGroupApplicationList(_ context.Context, req *pbGroup.GetGroupApplicationListReq) (*pbGroup.GetGroupApplicationListResp, error) {
log.NewInfo(req.OperationID, "GetGroupMembersInfo args ", req.String())
reply, err := im_mysql_model.GetGroupApplicationList(req.OpUserID)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "GetGroupApplicationList failed ", err.Error(), req.OpUserID)
2021-12-22 15:24:49 +08:00
return &pbGroup.GetGroupApplicationListResp{ErrCode: 701, ErrMsg: "GetGroupApplicationList failed"}, nil
}
2021-12-27 21:08:42 +08:00
resp := pbGroup.GetGroupApplicationListResp{}
for _, v := range reply {
var node open_im_sdk.GroupRequest
utils.CopyStructFields(&node, v)
}
log.NewInfo(req.OperationID, "GetGroupMembersInfo rpc return ", resp)
return &resp, nil
2021-12-22 15:24:49 +08:00
}
func (s *groupServer) GetGroupsInfo(ctx context.Context, req *pbGroup.GetGroupsInfoReq) (*pbGroup.GetGroupsInfoResp, error) {
2021-12-23 17:22:49 +08:00
log.NewInfo(req.OperationID, "GetGroupsInfo args ", req.String())
groupsInfoList := make([]*open_im_sdk.GroupInfo, 0)
2021-12-22 15:24:49 +08:00
for _, groupID := range req.GroupIDList {
2021-12-27 21:08:42 +08:00
groupInfoFromMysql, err := im_mysql_model.GetGroupInfoByGroupID(groupID)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "FindGroupInfoByGroupId failed ", err.Error(), groupID)
2021-12-22 15:24:49 +08:00
continue
}
2021-12-23 17:22:49 +08:00
var groupInfo open_im_sdk.GroupInfo
utils.CopyStructFields(&groupInfo, groupInfoFromMysql)
2021-12-28 15:33:47 +08:00
groupInfo.CreateTime = groupInfoFromMysql.CreateTime.Unix()
2021-12-22 15:24:49 +08:00
groupsInfoList = append(groupsInfoList, &groupInfo)
}
2021-12-23 17:22:49 +08:00
resp := pbGroup.GetGroupsInfoResp{GroupInfoList: groupsInfoList}
log.NewInfo(req.OperationID, "GetGroupsInfo rpc return ", resp)
return &resp, nil
2021-12-22 15:24:49 +08:00
}
2021-12-26 18:47:11 +08:00
func (s *groupServer) GroupApplicationResponse(_ context.Context, req *pbGroup.GroupApplicationResponseReq) (*pbGroup.GroupApplicationResponseResp, error) {
2021-12-23 17:22:49 +08:00
log.NewInfo(req.OperationID, "GroupApplicationResponse args ", req.String())
2021-12-27 16:48:05 +08:00
groupRequest := imdb.GroupRequest{}
groupRequest.UserID = req.FromUserID
groupRequest.GroupID = req.GroupID
groupRequest.HandledTime = time.Now()
groupRequest.HandleResult = req.HandleResult
groupRequest.HandleUserID = req.OpUserID
groupRequest.HandledMsg = req.HandledMsg
err := imdb.UpdateGroupRequest(groupRequest)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "GroupApplicationResponse failed ", err.Error(), req.String())
2021-12-27 16:48:05 +08:00
return &pbGroup.GroupApplicationResponseResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrDB.ErrMsg}}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-27 16:48:05 +08:00
chat.ApplicationProcessedNotification(req)
2021-12-23 17:22:49 +08:00
if req.HandleResult == 1 {
2021-12-27 16:48:05 +08:00
chat.MemberEnterNotification(req)
2021-12-22 15:24:49 +08:00
}
2021-12-27 16:48:05 +08:00
log.NewInfo(req.OperationID, "rpc GroupApplicationResponse return ")
return &pbGroup.GroupApplicationResponseResp{CommonResp: &pbGroup.CommonResp{}}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-26 18:47:11 +08:00
func (s *groupServer) JoinGroup(ctx context.Context, req *pbGroup.JoinGroupReq) (*pbGroup.JoinGroupResp, error) {
2021-12-23 17:22:49 +08:00
log.NewInfo(req.OperationID, "JoinGroup args ", req.String())
2021-12-28 15:33:47 +08:00
_, err := imdb.GetUserByUserID(req.OpUserID)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-24 15:02:47 +08:00
log.NewError(req.OperationID, "FindUserByUID failed ", err.Error(), req.OpUserID)
2021-12-26 18:47:11 +08:00
return &pbGroup.JoinGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrDB.ErrMsg}}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-27 21:08:42 +08:00
var groupRequest imdb.GroupRequest
utils.CopyStructFields(&groupRequest, req)
err = imdb.UpdateGroupRequest(groupRequest)
if err != nil {
log.NewError(req.OperationID, "UpdateGroupRequest ", err.Error(), groupRequest)
2021-12-26 18:47:11 +08:00
return &pbGroup.JoinGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrDB.ErrMsg}}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-27 21:08:42 +08:00
_, err = imdb.GetGroupMemberListByGroupIDAndRoleLevel(req.GroupID, constant.GroupOwner)
if err != nil {
log.NewError(req.OperationID, "GetGroupMemberListByGroupIDAndRoleLevel failed ", err.Error(), req.GroupID, constant.GroupOwner)
2021-12-26 18:47:11 +08:00
return &pbGroup.JoinGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: 0, ErrMsg: ""}}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-24 15:02:47 +08:00
2021-12-24 16:02:21 +08:00
chat.JoinApplicationNotification(req)
2021-12-22 15:24:49 +08:00
2021-12-24 16:02:21 +08:00
log.NewInfo(req.OperationID, "ReceiveJoinApplicationNotification rpc return ")
2021-12-26 18:47:11 +08:00
return &pbGroup.JoinGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: 0, ErrMsg: ""}}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-26 18:47:11 +08:00
func (s *groupServer) QuitGroup(ctx context.Context, req *pbGroup.QuitGroupReq) (*pbGroup.QuitGroupResp, error) {
2021-12-22 15:24:49 +08:00
log.NewError("QuitGroup args ", req.String())
2021-12-27 21:08:42 +08:00
_, err := imdb.GetGroupMemberInfoByGroupIDAndUserID(req.GroupID, req.OpUserID)
2021-12-22 15:24:49 +08:00
if err != nil {
log.NewError(req.OperationID, "FindGroupMemberInfoByGroupIdAndUserId failed", err.Error(), req.GroupID, req.OpUserID)
2021-12-26 18:47:11 +08:00
return &pbGroup.QuitGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrDB.ErrMsg}}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-27 21:08:42 +08:00
err = imdb.DeleteGroupMemberByGroupIDAndUserID(req.GroupID, req.OpUserID)
2021-12-22 15:24:49 +08:00
if err != nil {
log.NewError(req.OperationID, "DeleteGroupMemberByGroupIdAndUserId failed ", err.Error(), req.GroupID, req.OpUserID)
2021-12-26 18:47:11 +08:00
return &pbGroup.QuitGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrDB.ErrMsg}}, nil
2021-12-22 15:24:49 +08:00
}
err = db.DB.DelGroupMember(req.GroupID, req.OpUserID)
if err != nil {
log.NewError(req.OperationID, "DelGroupMember failed ", req.GroupID, req.OpUserID)
// return &pbGroup.CommonResp{ErrorCode: constant.ErrQuitGroup.ErrCode, ErrorMsg: constant.ErrQuitGroup.ErrMsg}, nil
}
2021-12-24 15:02:47 +08:00
chat.MemberLeaveNotification(req)
2021-12-24 16:02:21 +08:00
log.NewInfo(req.OperationID, "rpc QuitGroup return ")
2021-12-26 18:47:11 +08:00
return &pbGroup.QuitGroupResp{CommonResp: &pbGroup.CommonResp{ErrCode: 0, ErrMsg: ""}}, nil
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
}
2021-12-27 21:08:42 +08:00
groupUserInfo, err := im_mysql_model.GetGroupMemberInfoByGroupIDAndUserID(req.GroupInfo.GroupID, req.OpUserID)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-23 17:22:49 +08:00
log.NewError(req.OperationID, "FindGroupMemberInfoByGroupIdAndUserId failed, ", err.Error(), req.GroupInfo.GroupID, req.OpUserID)
2021-12-22 15:24:49 +08:00
return false
}
2021-12-29 10:57:48 +08:00
if 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
}
2021-12-26 18:47:11 +08:00
func (s *groupServer) SetGroupInfo(ctx context.Context, req *pbGroup.SetGroupInfoReq) (*pbGroup.SetGroupInfoResp, error) {
2021-12-22 15:24:49 +08:00
log.NewInfo(req.OperationID, "SetGroupInfo args ", req.String())
if !hasAccess(req) {
log.NewError(req.OperationID, "no access ")
2021-12-26 18:47:11 +08:00
return &pbGroup.SetGroupInfoResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrAccess.ErrCode, ErrMsg: constant.ErrAccess.ErrMsg}}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-27 21:08:42 +08:00
group, err := im_mysql_model.GetGroupInfoByGroupID(req.GroupInfo.GroupID)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-27 21:08:42 +08:00
log.NewError(req.OperationID, "FindGroupInfoByGroupId failed ", err.Error(), req.GroupInfo.GroupID)
2021-12-26 18:47:11 +08:00
return &pbGroup.SetGroupInfoResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrAccess.ErrMsg}}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-24 16:02:21 +08:00
////bitwise operators: 0001:groupName; 0010:Notification 0100:Introduction; 1000:FaceUrl; 10000:owner
2021-12-22 15:24:49 +08:00
var changedType int32
2021-12-23 17:22:49 +08:00
if group.GroupName != req.GroupInfo.GroupName && req.GroupInfo.GroupName != "" {
2021-12-22 15:24:49 +08:00
changedType = 1
}
2021-12-23 17:22:49 +08:00
if group.Notification != req.GroupInfo.Notification && req.GroupInfo.Notification != "" {
2021-12-22 15:24:49 +08:00
changedType = changedType | (1 << 1)
}
2021-12-23 17:22:49 +08:00
if group.Introduction != req.GroupInfo.Introduction && req.GroupInfo.Introduction != "" {
2021-12-22 15:24:49 +08:00
changedType = changedType | (1 << 2)
}
2021-12-23 17:22:49 +08:00
if group.FaceUrl != req.GroupInfo.FaceUrl && req.GroupInfo.FaceUrl != "" {
2021-12-22 15:24:49 +08:00
changedType = changedType | (1 << 3)
}
//only administrators can set group information
2021-12-27 21:08:42 +08:00
var groupInfo imdb.Group
utils.CopyStructFields(&groupInfo, req.GroupInfo)
err = imdb.SetGroupInfo(groupInfo)
if err != nil {
return &pbGroup.SetGroupInfoResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrAccess.ErrMsg}}, nil
2021-12-22 15:24:49 +08:00
}
if changedType != 0 {
2021-12-24 16:02:21 +08:00
chat.GroupInfoChangedNotification(req.OperationID, req.OpUserID, req.GroupInfo.GroupID, changedType)
2021-12-22 15:24:49 +08:00
}
2021-12-27 21:08:42 +08:00
log.NewInfo("SetGroupInfo rpc return ", pbGroup.SetGroupInfoResp{CommonResp: &pbGroup.CommonResp{}})
return &pbGroup.SetGroupInfoResp{CommonResp: &pbGroup.CommonResp{}}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-26 18:47:11 +08:00
func (s *groupServer) TransferGroupOwner(_ context.Context, req *pbGroup.TransferGroupOwnerReq) (*pbGroup.TransferGroupOwnerResp, error) {
2021-12-24 16:02:21 +08:00
log.NewInfo(req.OperationID, "TransferGroupOwner ", req.String())
2021-12-22 15:24:49 +08:00
2021-12-27 16:48:05 +08:00
if req.OldOwnerUserID == req.NewOwnerUserID {
log.NewError(req.OperationID, "same owner ", req.String())
return &pbGroup.TransferGroupOwnerResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrArgs.ErrCode, ErrMsg: constant.ErrArgs.ErrMsg}}, nil
}
groupMemberInfo := imdb.GroupMember{GroupID: req.GroupID, UserID: req.OldOwnerUserID, RoleLevel: 0}
err := imdb.UpdateGroupMemberInfo(groupMemberInfo)
if err != nil {
return &pbGroup.TransferGroupOwnerResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrDB.ErrMsg}}, nil
}
groupMemberInfo = imdb.GroupMember{GroupID: req.GroupID, UserID: req.NewOwnerUserID, RoleLevel: 1}
err = imdb.UpdateGroupMemberInfo(groupMemberInfo)
2021-12-22 15:24:49 +08:00
if err != nil {
2021-12-26 18:47:11 +08:00
return &pbGroup.TransferGroupOwnerResp{CommonResp: &pbGroup.CommonResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: constant.ErrDB.ErrMsg}}, nil
2021-12-22 15:24:49 +08:00
}
2021-12-24 16:02:21 +08:00
changedType := int32(1) << 4
chat.GroupInfoChangedNotification(req.OperationID, req.OpUserID, req.GroupID, changedType)
2021-12-27 16:48:05 +08:00
return &pbGroup.TransferGroupOwnerResp{CommonResp: &pbGroup.CommonResp{ErrCode: 0, ErrMsg: ""}}, nil
2021-12-22 15:24:49 +08:00
}