Files
open-im-server/internal/rpc/msg/send_msg.go
T

1173 lines
53 KiB
Go
Raw Normal View History

2021-12-23 17:19:57 +08:00
package msg
2021-05-26 19:40:38 +08:00
import (
"Open_IM/pkg/common/config"
"Open_IM/pkg/common/constant"
2021-12-06 20:03:59 +08:00
"Open_IM/pkg/common/db"
2022-08-01 19:13:26 +08:00
rocksCache "Open_IM/pkg/common/db/rocks_cache"
"Open_IM/pkg/common/log"
2023-02-09 14:40:49 +08:00
"Open_IM/pkg/common/tokenverify"
2023-01-11 16:23:16 +08:00
"Open_IM/pkg/getcdv3"
2022-04-28 15:10:42 +08:00
cacheRpc "Open_IM/pkg/proto/cache"
2022-04-24 15:55:19 +08:00
pbConversation "Open_IM/pkg/proto/conversation"
2022-07-20 21:15:30 +08:00
pbChat "Open_IM/pkg/proto/msg"
2022-09-19 22:16:45 +08:00
pbPush "Open_IM/pkg/proto/push"
2022-05-11 18:33:48 +08:00
pbRelay "Open_IM/pkg/proto/relay"
2021-12-23 17:19:57 +08:00
sdk_ws "Open_IM/pkg/proto/sdk_ws"
"Open_IM/pkg/utils"
2021-05-26 19:40:38 +08:00
"context"
2022-05-11 18:33:48 +08:00
"errors"
2021-05-26 19:40:38 +08:00
"math/rand"
"strconv"
2021-06-28 15:35:18 +08:00
"strings"
2022-04-29 18:47:55 +08:00
"sync"
2021-05-26 19:40:38 +08:00
"time"
2022-08-17 14:57:57 +08:00
2022-09-15 12:07:28 +08:00
promePkg "Open_IM/pkg/common/prometheus"
2022-08-17 14:57:57 +08:00
go_redis "github.com/go-redis/redis/v8"
"github.com/golang/protobuf/proto"
2021-05-26 19:40:38 +08:00
)
2022-05-13 21:05:39 +08:00
//When the number of group members is greater than this valueOnline users will be sent firstGuaranteed service availability
const GroupMemberNum = 500
2022-09-14 17:45:38 +08:00
var (
ExcludeContentType = []int{constant.HasReadReceipt, constant.GroupHasReadReceipt}
)
2022-08-05 19:10:08 +08:00
2022-09-29 17:57:26 +08:00
type Validator interface {
validate(pb *pbChat.SendMsgReq) (bool, int32, string)
}
//type MessageValidator struct {
//
//}
type MessageRevoked struct {
RevokerID string `json:"revokerID"`
RevokerRole int32 `json:"revokerRole"`
ClientMsgID string `json:"clientMsgID"`
RevokerNickname string `json:"revokerNickname"`
RevokeTime int64 `json:"revokeTime"`
SourceMessageSendTime int64 `json:"sourceMessageSendTime"`
SourceMessageSendID string `json:"sourceMessageSendID"`
SourceMessageSenderNickname string `json:"sourceMessageSenderNickname"`
SessionType int32 `json:"sessionType"`
Seq uint32 `json:"seq"`
}
2021-06-28 15:35:18 +08:00
type MsgCallBackReq struct {
SendID string `json:"sendID"`
RecvID string `json:"recvID"`
Content string `json:"content"`
SendTime int64 `json:"sendTime"`
MsgFrom int32 `json:"msgFrom"`
ContentType int32 `json:"contentType"`
SessionType int32 `json:"sessionType"`
PlatformID int32 `json:"senderPlatformID"`
MsgID string `json:"msgID"`
IsOnlineOnly bool `json:"isOnlineOnly"`
2021-06-28 15:35:18 +08:00
}
type MsgCallBackResp struct {
ErrCode int32 `json:"errCode"`
ErrMsg string `json:"errMsg"`
ResponseErrCode int32 `json:"responseErrCode"`
ResponseResult struct {
ModifiedMsg string `json:"modifiedMsg"`
Ext string `json:"ext"`
}
}
2021-05-26 19:40:38 +08:00
2022-06-09 12:39:28 +08:00
func isMessageHasReadEnabled(pb *pbChat.SendMsgReq) (bool, int32, string) {
switch pb.MsgData.ContentType {
case constant.HasReadReceipt:
if config.Config.SingleMessageHasReadReceiptEnable {
return true, 0, ""
} else {
return false, constant.ErrMessageHasReadDisable.ErrCode, constant.ErrMessageHasReadDisable.ErrMsg
}
case constant.GroupHasReadReceipt:
if config.Config.GroupMessageHasReadReceiptEnable {
return true, 0, ""
} else {
return false, constant.ErrMessageHasReadDisable.ErrCode, constant.ErrMessageHasReadDisable.ErrMsg
}
}
return true, 0, ""
}
2023-01-11 11:15:46 +08:00
func userIsMuteAndIsAdminInGroup(ctx context.Context, groupID, userID string) (isMute bool, isAdmin bool, err error) {
groupMemberInfo, err := rocksCache.GetGroupMemberInfoFromCache(ctx, groupID, userID)
2022-11-11 18:06:15 +08:00
if err != nil {
2022-12-02 17:27:12 +08:00
return false, false, utils.Wrap(err, "")
2022-11-11 18:06:15 +08:00
}
2022-12-02 17:27:12 +08:00
2022-11-11 18:06:15 +08:00
if groupMemberInfo.MuteEndTime.Unix() >= time.Now().Unix() {
2022-12-02 17:27:12 +08:00
return true, groupMemberInfo.RoleLevel > constant.GroupOrdinaryUsers, nil
2022-11-11 18:06:15 +08:00
}
2022-12-02 17:27:12 +08:00
return false, groupMemberInfo.RoleLevel > constant.GroupOrdinaryUsers, nil
2022-11-11 18:06:15 +08:00
}
2023-01-11 11:15:46 +08:00
func groupIsMuted(ctx context.Context, groupID string) (bool, error) {
groupInfo, err := rocksCache.GetGroupInfoFromCache(ctx, groupID)
2022-11-29 14:41:20 +08:00
if err != nil {
return false, utils.Wrap(err, "GetGroupInfoFromCache failed")
}
if groupInfo.Status == constant.GroupStatusMuted {
return true, nil
}
return false, nil
}
2023-01-11 11:15:46 +08:00
func (rpc *rpcChat) messageVerification(ctx context.Context, data *pbChat.SendMsgReq) (bool, int32, string, []string) {
2022-07-25 19:19:00 +08:00
switch data.MsgData.SessionType {
case constant.SingleChatType:
2022-08-05 15:51:34 +08:00
if utils.IsContain(data.MsgData.SendID, config.Config.Manager.AppManagerUid) {
return true, 0, "", nil
}
if data.MsgData.ContentType <= constant.NotificationEnd && data.MsgData.ContentType >= constant.NotificationBegin {
return true, 0, "", nil
}
2022-11-10 14:51:35 +08:00
log.NewDebug(data.OperationID, *config.Config.MessageVerify.FriendVerify)
2022-05-28 18:10:08 +08:00
reqGetBlackIDListFromCache := &cacheRpc.GetBlackIDListFromCacheReq{UserID: data.MsgData.RecvID, OperationID: data.OperationID}
2023-02-07 20:24:20 +08:00
etcdConn, err := rpc.GetConn(context.Background(), config.Config.RpcRegisterName.OpenImCacheName)
2023-01-11 11:15:46 +08:00
if err != nil {
2022-08-17 12:12:54 +08:00
errMsg := data.OperationID + "getcdv3.GetDefaultConn == nil"
2022-06-16 14:26:03 +08:00
log.NewError(data.OperationID, errMsg)
2022-07-25 19:19:00 +08:00
return true, 0, "", nil
2022-06-16 14:26:03 +08:00
}
2022-04-28 15:10:42 +08:00
cacheClient := cacheRpc.NewCacheClient(etcdConn)
2022-05-28 18:10:08 +08:00
cacheResp, err := cacheClient.GetBlackIDListFromCache(context.Background(), reqGetBlackIDListFromCache)
2022-03-15 19:03:22 +08:00
if err != nil {
2022-05-28 18:10:08 +08:00
log.NewError(data.OperationID, "GetBlackIDListFromCache rpc call failed ", err.Error())
2022-04-28 15:10:42 +08:00
} else {
if cacheResp.CommonResp.ErrCode != 0 {
2022-05-28 18:10:08 +08:00
log.NewError(data.OperationID, "GetBlackIDListFromCache rpc logic call failed ", cacheResp.String())
2022-04-28 15:10:42 +08:00
} else {
2022-05-28 18:10:08 +08:00
if utils.IsContain(data.MsgData.SendID, cacheResp.UserIDList) {
2022-07-25 19:19:00 +08:00
return false, 600, "in black list", nil
2022-04-28 15:10:42 +08:00
}
}
2022-03-15 19:03:22 +08:00
}
2022-11-10 14:51:35 +08:00
log.NewDebug(data.OperationID, *config.Config.MessageVerify.FriendVerify)
if *config.Config.MessageVerify.FriendVerify {
2022-05-28 18:10:08 +08:00
reqGetFriendIDListFromCache := &cacheRpc.GetFriendIDListFromCacheReq{UserID: data.MsgData.RecvID, OperationID: data.OperationID}
2023-02-07 20:24:20 +08:00
etcdConn, err := rpc.GetConn(context.Background(), config.Config.RpcRegisterName.OpenImCacheName)
2023-01-11 11:15:46 +08:00
if err != nil {
2022-08-17 12:12:54 +08:00
errMsg := data.OperationID + "getcdv3.GetDefaultConn == nil"
2022-06-16 14:26:03 +08:00
log.NewError(data.OperationID, errMsg)
2022-07-25 19:19:00 +08:00
return true, 0, "", nil
2022-06-16 14:26:03 +08:00
}
2022-05-28 18:10:08 +08:00
cacheClient := cacheRpc.NewCacheClient(etcdConn)
cacheResp, err := cacheClient.GetFriendIDListFromCache(context.Background(), reqGetFriendIDListFromCache)
if err != nil {
log.NewError(data.OperationID, "GetFriendIDListFromCache rpc call failed ", err.Error())
} else {
if cacheResp.CommonResp.ErrCode != 0 {
log.NewError(data.OperationID, "GetFriendIDListFromCache rpc logic call failed ", cacheResp.String())
} else {
if !utils.IsContain(data.MsgData.SendID, cacheResp.UserIDList) {
2022-07-25 19:19:00 +08:00
return false, 601, "not friend", nil
2022-05-28 18:10:08 +08:00
}
}
}
2022-07-25 19:19:00 +08:00
return true, 0, "", nil
2022-05-28 18:10:08 +08:00
} else {
2022-07-25 19:19:00 +08:00
return true, 0, "", nil
}
case constant.GroupChatType:
2023-01-30 15:28:46 +08:00
userIDList, err := utils.GetGroupMemberUserIDList(ctx, data.MsgData.GroupID, data.OperationID)
2022-10-10 17:51:57 +08:00
if err != nil {
errMsg := data.OperationID + err.Error()
log.NewError(data.OperationID, errMsg)
return false, 201, errMsg, nil
}
2023-02-09 14:40:49 +08:00
if tokenverify.IsManagerUserID(data.MsgData.SendID) {
2022-11-21 10:20:56 +08:00
return true, 0, "", userIDList
}
2022-11-22 14:18:57 +08:00
if data.MsgData.ContentType <= constant.NotificationEnd && data.MsgData.ContentType >= constant.NotificationBegin {
return true, 0, "", userIDList
} else {
if !utils.IsContain(data.MsgData.SendID, userIDList) {
//return returnMsg(&replay, pb, 202, "you are not in group", "", 0)
return false, 202, "you are not in group", nil
}
2022-10-10 17:51:57 +08:00
}
2023-01-11 11:15:46 +08:00
isMute, isAdmin, err := userIsMuteAndIsAdminInGroup(ctx, data.MsgData.GroupID, data.MsgData.SendID)
2022-11-29 14:41:20 +08:00
if err != nil {
errMsg := data.OperationID + err.Error()
return false, 223, errMsg, nil
}
if isMute {
2022-12-02 17:27:12 +08:00
return false, 224, "you are muted", nil
}
if isAdmin {
return true, 0, "", userIDList
2022-11-29 14:41:20 +08:00
}
2023-01-11 11:15:46 +08:00
isMute, err = groupIsMuted(ctx, data.MsgData.GroupID)
2022-11-11 18:06:15 +08:00
if err != nil {
errMsg := data.OperationID + err.Error()
return false, 223, errMsg, nil
}
if isMute {
2022-12-02 17:27:12 +08:00
return false, 225, "group id muted", nil
2022-11-11 18:06:15 +08:00
}
2022-10-10 17:51:57 +08:00
return true, 0, "", userIDList
2022-07-25 19:19:00 +08:00
case constant.SuperGroupChatType:
2023-01-11 11:15:46 +08:00
groupInfo, err := rocksCache.GetGroupInfoFromCache(ctx, data.MsgData.GroupID)
2022-07-25 19:19:00 +08:00
if err != nil {
return false, 201, err.Error(), nil
}
2022-10-10 17:51:57 +08:00
2022-09-29 17:57:26 +08:00
if data.MsgData.ContentType == constant.AdvancedRevoke {
2022-09-29 18:31:36 +08:00
revokeMessage := new(MessageRevoked)
err := utils.JsonStringToStruct(string(data.MsgData.Content), revokeMessage)
2022-09-29 17:57:26 +08:00
if err != nil {
log.Error(data.OperationID, "json unmarshal err:", err.Error())
return false, 201, err.Error(), nil
}
2022-09-29 18:35:47 +08:00
log.Debug(data.OperationID, "revoke message is", *revokeMessage)
2022-09-29 19:29:22 +08:00
if revokeMessage.RevokerID != revokeMessage.SourceMessageSendID {
req := pbChat.GetSuperGroupMsgReq{OperationID: data.OperationID, Seq: revokeMessage.Seq, GroupID: data.MsgData.GroupID}
resp, err := rpc.GetSuperGroupMsg(context.Background(), &req)
if err != nil {
log.Error(data.OperationID, "GetSuperGroupMsgReq err:", err.Error())
} else if resp.ErrCode != 0 {
2022-10-10 17:51:57 +08:00
log.Error(data.OperationID, "GetSuperGroupMsgReq err:", resp.ErrCode, resp.ErrMsg)
2022-09-29 17:57:26 +08:00
} else {
2022-09-29 19:29:22 +08:00
if resp.MsgData != nil && resp.MsgData.ClientMsgID == revokeMessage.ClientMsgID && resp.MsgData.Seq == revokeMessage.Seq {
revokeMessage.SourceMessageSendTime = resp.MsgData.SendTime
revokeMessage.SourceMessageSenderNickname = resp.MsgData.SenderNickname
revokeMessage.SourceMessageSendID = resp.MsgData.SendID
log.Debug(data.OperationID, "new revoke message is ", revokeMessage)
data.MsgData.Content = []byte(utils.StructToJsonString(revokeMessage))
} else {
return false, 201, errors.New("msg err").Error(), nil
}
2022-09-29 17:57:26 +08:00
}
}
}
2022-08-01 19:13:26 +08:00
if groupInfo.GroupType == constant.SuperGroup {
return true, 0, "", nil
} else {
2023-01-30 15:28:46 +08:00
userIDList, err := utils.GetGroupMemberUserIDList(ctx, data.MsgData.GroupID, data.OperationID)
2022-08-30 22:04:13 +08:00
if err != nil {
errMsg := data.OperationID + err.Error()
2022-08-01 19:13:26 +08:00
log.NewError(data.OperationID, errMsg)
return false, 201, errMsg, nil
2022-07-25 19:19:00 +08:00
}
2023-02-09 14:40:49 +08:00
if tokenverify.IsManagerUserID(data.MsgData.SendID) {
2022-11-21 10:20:56 +08:00
return true, 0, "", userIDList
}
2022-11-22 14:18:57 +08:00
if data.MsgData.ContentType <= constant.NotificationEnd && data.MsgData.ContentType >= constant.NotificationBegin {
return true, 0, "", userIDList
} else {
if !utils.IsContain(data.MsgData.SendID, userIDList) {
//return returnMsg(&replay, pb, 202, "you are not in group", "", 0)
return false, 202, "you are not in group", nil
}
2022-08-01 19:13:26 +08:00
}
2023-01-11 11:15:46 +08:00
isMute, isAdmin, err := userIsMuteAndIsAdminInGroup(ctx, data.MsgData.GroupID, data.MsgData.SendID)
2022-11-29 14:41:20 +08:00
if err != nil {
errMsg := data.OperationID + err.Error()
return false, 223, errMsg, nil
}
if isMute {
2022-12-02 17:27:12 +08:00
return false, 224, "you are muted", nil
2022-11-29 14:41:20 +08:00
}
2022-12-02 17:27:12 +08:00
if isAdmin {
return true, 0, "", userIDList
}
2023-01-11 11:15:46 +08:00
isMute, err = groupIsMuted(ctx, data.MsgData.GroupID)
2022-11-11 18:06:15 +08:00
if err != nil {
errMsg := data.OperationID + err.Error()
return false, 223, errMsg, nil
}
if isMute {
2022-12-02 17:27:12 +08:00
return false, 225, "group id muted", nil
2022-11-11 18:06:15 +08:00
}
2022-08-30 22:04:13 +08:00
return true, 0, "", userIDList
2022-07-25 19:19:00 +08:00
}
default:
return true, 0, "", nil
2022-03-15 19:03:22 +08:00
}
2022-07-25 19:19:00 +08:00
2022-02-21 18:18:53 +08:00
}
2021-12-23 17:19:57 +08:00
func (rpc *rpcChat) encapsulateMsgData(msg *sdk_ws.MsgData) {
msg.ServerMsgID = GetMsgID(msg.SendID)
2022-02-22 16:01:32 +08:00
msg.SendTime = utils.GetCurrentTimestampByMill()
2021-12-23 17:19:57 +08:00
switch msg.ContentType {
case constant.Text:
fallthrough
case constant.Picture:
fallthrough
case constant.Voice:
fallthrough
case constant.Video:
fallthrough
case constant.File:
fallthrough
case constant.AtText:
fallthrough
case constant.Merger:
fallthrough
case constant.Card:
fallthrough
case constant.Location:
fallthrough
case constant.Custom:
fallthrough
case constant.Quote:
utils.SetSwitchFromOptions(msg.Options, constant.IsConversationUpdate, true)
utils.SetSwitchFromOptions(msg.Options, constant.IsUnreadCount, true)
utils.SetSwitchFromOptions(msg.Options, constant.IsSenderSync, true)
case constant.Revoke:
utils.SetSwitchFromOptions(msg.Options, constant.IsUnreadCount, false)
utils.SetSwitchFromOptions(msg.Options, constant.IsOfflinePush, false)
case constant.HasReadReceipt:
2022-02-14 16:59:26 +08:00
log.Info("", "this is a test start", msg, msg.Options)
2021-12-23 17:19:57 +08:00
utils.SetSwitchFromOptions(msg.Options, constant.IsConversationUpdate, false)
2022-03-28 18:30:12 +08:00
utils.SetSwitchFromOptions(msg.Options, constant.IsSenderConversationUpdate, false)
2021-12-23 17:19:57 +08:00
utils.SetSwitchFromOptions(msg.Options, constant.IsUnreadCount, false)
utils.SetSwitchFromOptions(msg.Options, constant.IsOfflinePush, false)
2022-02-14 16:59:26 +08:00
log.Info("", "this is a test end", msg, msg.Options)
2021-12-23 17:19:57 +08:00
case constant.Typing:
utils.SetSwitchFromOptions(msg.Options, constant.IsHistory, false)
utils.SetSwitchFromOptions(msg.Options, constant.IsPersistent, false)
utils.SetSwitchFromOptions(msg.Options, constant.IsSenderSync, false)
utils.SetSwitchFromOptions(msg.Options, constant.IsConversationUpdate, false)
2022-03-28 18:30:12 +08:00
utils.SetSwitchFromOptions(msg.Options, constant.IsSenderConversationUpdate, false)
2021-12-23 17:19:57 +08:00
utils.SetSwitchFromOptions(msg.Options, constant.IsUnreadCount, false)
utils.SetSwitchFromOptions(msg.Options, constant.IsOfflinePush, false)
}
}
2023-01-11 11:15:46 +08:00
func (rpc *rpcChat) SendMsg(ctx context.Context, pb *pbChat.SendMsgReq) (*pbChat.SendMsgResp, error) {
2021-12-23 17:19:57 +08:00
replay := pbChat.SendMsgResp{}
2022-06-16 10:26:37 +08:00
log.Info(pb.OperationID, "rpc sendMsg come here ", pb.String())
2022-06-09 12:39:28 +08:00
flag, errCode, errMsg := isMessageHasReadEnabled(pb)
if !flag {
return returnMsg(&replay, pb, errCode, errMsg, "", 0)
}
2022-09-19 22:16:45 +08:00
t1 := time.Now()
2021-12-23 17:19:57 +08:00
rpc.encapsulateMsgData(pb.MsgData)
2022-09-19 22:16:45 +08:00
log.Debug(pb.OperationID, "encapsulateMsgData ", " cost time: ", time.Since(t1))
2022-05-01 12:04:31 +08:00
msgToMQSingle := pbChat.MsgDataToMQ{Token: pb.Token, OperationID: pb.OperationID, MsgData: pb.MsgData}
2022-02-28 17:57:03 +08:00
// callback
2022-08-22 19:19:30 +08:00
t1 = time.Now()
2022-10-21 15:44:55 +08:00
callbackResp := callbackMsgModify(pb)
log.Debug(pb.OperationID, "callbackMsgModify ", callbackResp, "cost time: ", time.Since(t1))
2022-05-26 18:02:00 +08:00
if callbackResp.ErrCode != 0 {
2022-10-21 15:44:55 +08:00
log.Error(pb.OperationID, utils.GetSelfFuncName(), "callbackMsgModify resp: ", callbackResp)
2022-03-03 14:42:50 +08:00
}
2022-05-27 11:08:27 +08:00
log.NewDebug(pb.OperationID, utils.GetSelfFuncName(), "callbackResp: ", callbackResp)
2022-05-26 18:02:00 +08:00
if callbackResp.ActionCode != constant.ActionAllow {
2022-05-27 11:49:38 +08:00
if callbackResp.ErrCode == 0 {
callbackResp.ErrCode = 201
}
2022-10-21 15:44:55 +08:00
log.NewDebug(pb.OperationID, utils.GetSelfFuncName(), "callbackMsgModify result", "end rpc and return", pb.MsgData)
2022-05-26 18:02:00 +08:00
return returnMsg(&replay, pb, int32(callbackResp.ErrCode), callbackResp.ErrMsg, "", 0)
2022-03-03 14:42:50 +08:00
}
2021-12-23 17:19:57 +08:00
switch pb.MsgData.SessionType {
2021-10-21 12:10:38 +08:00
case constant.SingleChatType:
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.SingleChatMsgRecvSuccessCounter)
2022-03-04 14:27:54 +08:00
// callback
2022-08-22 19:19:30 +08:00
t1 = time.Now()
2022-05-26 18:02:00 +08:00
callbackResp := callbackBeforeSendSingleMsg(pb)
2022-09-19 22:16:45 +08:00
log.Debug(pb.OperationID, "callbackBeforeSendSingleMsg ", " cost time: ", time.Since(t1))
2022-05-26 18:02:00 +08:00
if callbackResp.ErrCode != 0 {
log.NewError(pb.OperationID, utils.GetSelfFuncName(), "callbackBeforeSendSingleMsg resp: ", callbackResp)
2022-02-28 17:57:03 +08:00
}
2022-05-26 18:02:00 +08:00
if callbackResp.ActionCode != constant.ActionAllow {
2022-05-27 11:49:38 +08:00
if callbackResp.ErrCode == 0 {
callbackResp.ErrCode = 201
}
2022-05-26 18:02:00 +08:00
log.NewDebug(pb.OperationID, utils.GetSelfFuncName(), "callbackBeforeSendSingleMsg result", "end rpc and return", callbackResp)
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.SingleChatMsgProcessFailedCounter)
2022-05-27 11:49:38 +08:00
return returnMsg(&replay, pb, int32(callbackResp.ErrCode), callbackResp.ErrMsg, "", 0)
2022-02-28 17:57:03 +08:00
}
2022-09-19 22:16:45 +08:00
t1 = time.Now()
2023-01-11 11:15:46 +08:00
flag, errCode, errMsg, _ = rpc.messageVerification(ctx, pb)
2022-09-19 22:16:45 +08:00
log.Debug(pb.OperationID, "messageVerification ", flag, " cost time: ", time.Since(t1))
if !flag {
return returnMsg(&replay, pb, errCode, errMsg, "", 0)
}
2022-08-22 19:19:30 +08:00
t1 = time.Now()
2021-12-23 17:19:57 +08:00
isSend := modifyMessageByUserMessageReceiveOpt(pb.MsgData.RecvID, pb.MsgData.SendID, constant.SingleChatType, pb)
2022-08-22 19:19:30 +08:00
log.Info(pb.OperationID, "modifyMessageByUserMessageReceiveOpt ", " cost time: ", time.Since(t1))
2021-12-06 20:03:59 +08:00
if isSend {
2022-05-01 12:04:31 +08:00
msgToMQSingle.MsgData = pb.MsgData
log.NewInfo(msgToMQSingle.OperationID, msgToMQSingle)
2022-08-22 19:19:30 +08:00
t1 = time.Now()
2023-01-11 11:15:46 +08:00
err1 := rpc.sendMsgToWriter(ctx, &msgToMQSingle, msgToMQSingle.MsgData.RecvID, constant.OnlineStatus)
2022-09-29 17:57:26 +08:00
log.Info(pb.OperationID, "sendMsgToWriter ", " cost time: ", time.Since(t1))
2021-12-06 20:03:59 +08:00
if err1 != nil {
2022-06-16 10:26:37 +08:00
log.NewError(msgToMQSingle.OperationID, "kafka send msg err :RecvID", msgToMQSingle.MsgData.RecvID, msgToMQSingle.String(), err1.Error())
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.SingleChatMsgProcessFailedCounter)
2021-12-06 20:03:59 +08:00
return returnMsg(&replay, pb, 201, "kafka send msg err", "", 0)
}
}
2022-05-01 12:04:31 +08:00
if msgToMQSingle.MsgData.SendID != msgToMQSingle.MsgData.RecvID { //Filter messages sent to yourself
2022-08-22 19:19:30 +08:00
t1 = time.Now()
2023-01-11 11:15:46 +08:00
err2 := rpc.sendMsgToWriter(ctx, &msgToMQSingle, msgToMQSingle.MsgData.SendID, constant.OnlineStatus)
2022-09-29 17:57:26 +08:00
log.Info(pb.OperationID, "sendMsgToWriter ", " cost time: ", time.Since(t1))
2021-12-28 20:44:19 +08:00
if err2 != nil {
2022-05-01 12:04:31 +08:00
log.NewError(msgToMQSingle.OperationID, "kafka send msg err:SendID", msgToMQSingle.MsgData.SendID, msgToMQSingle.String())
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.SingleChatMsgProcessFailedCounter)
2021-12-28 20:44:19 +08:00
return returnMsg(&replay, pb, 201, "kafka send msg err", "", 0)
}
2021-10-21 12:10:38 +08:00
}
2022-03-04 14:27:54 +08:00
// callback
2022-08-22 19:19:30 +08:00
t1 = time.Now()
2022-05-26 18:02:00 +08:00
callbackResp = callbackAfterSendSingleMsg(pb)
2022-08-22 19:19:30 +08:00
log.Info(pb.OperationID, "callbackAfterSendSingleMsg ", " cost time: ", time.Since(t1))
2022-05-26 18:02:00 +08:00
if callbackResp.ErrCode != 0 {
log.NewError(pb.OperationID, utils.GetSelfFuncName(), "callbackAfterSendSingleMsg resp: ", callbackResp)
2022-02-28 17:57:03 +08:00
}
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.SingleChatMsgProcessSuccessCounter)
2022-05-01 12:04:31 +08:00
return returnMsg(&replay, pb, 0, "", msgToMQSingle.MsgData.ServerMsgID, msgToMQSingle.MsgData.SendTime)
2021-10-21 12:10:38 +08:00
case constant.GroupChatType:
2022-03-04 14:27:54 +08:00
// callback
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.GroupChatMsgRecvSuccessCounter)
2022-05-26 18:02:00 +08:00
callbackResp := callbackBeforeSendGroupMsg(pb)
if callbackResp.ErrCode != 0 {
log.NewError(pb.OperationID, utils.GetSelfFuncName(), "callbackBeforeSendGroupMsg resp:", callbackResp)
2022-02-28 17:57:03 +08:00
}
2022-05-26 18:02:00 +08:00
if callbackResp.ActionCode != constant.ActionAllow {
2022-05-27 11:49:38 +08:00
if callbackResp.ErrCode == 0 {
callbackResp.ErrCode = 201
}
2022-05-26 18:02:00 +08:00
log.NewDebug(pb.OperationID, utils.GetSelfFuncName(), "callbackBeforeSendSingleMsg result", "end rpc and return", callbackResp)
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.GroupChatMsgProcessFailedCounter)
2022-05-27 11:49:38 +08:00
return returnMsg(&replay, pb, int32(callbackResp.ErrCode), callbackResp.ErrMsg, "", 0)
2022-02-28 17:57:03 +08:00
}
2022-07-25 19:19:00 +08:00
var memberUserIDList []string
2023-01-11 11:15:46 +08:00
if flag, errCode, errMsg, memberUserIDList = rpc.messageVerification(ctx, pb); !flag {
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.GroupChatMsgProcessFailedCounter)
2022-07-25 19:19:00 +08:00
return returnMsg(&replay, pb, errCode, errMsg, "", 0)
2021-10-21 12:10:38 +08:00
}
2022-07-25 19:19:00 +08:00
log.Debug(pb.OperationID, "GetGroupAllMember userID list", memberUserIDList, "len: ", len(memberUserIDList))
2022-02-09 12:50:36 +08:00
var addUidList []string
switch pb.MsgData.ContentType {
case constant.MemberKickedNotification:
var tips sdk_ws.TipsComm
var memberKickedTips sdk_ws.MemberKickedTips
2022-02-09 13:05:17 +08:00
err := proto.Unmarshal(pb.MsgData.Content, &tips)
if err != nil {
log.Error(pb.OperationID, "Unmarshal err", err.Error())
}
err = proto.Unmarshal(tips.Detail, &memberKickedTips)
if err != nil {
log.Error(pb.OperationID, "Unmarshal err", err.Error())
}
log.Info(pb.OperationID, "data is ", memberKickedTips)
2022-02-09 12:50:36 +08:00
for _, v := range memberKickedTips.KickedUserList {
addUidList = append(addUidList, v.UserID)
}
case constant.MemberQuitNotification:
addUidList = append(addUidList, pb.MsgData.SendID)
2022-04-29 18:47:55 +08:00
default:
}
2022-07-01 14:47:29 +08:00
if len(addUidList) > 0 {
memberUserIDList = append(memberUserIDList, addUidList...)
}
2022-05-13 21:05:39 +08:00
m := make(map[string][]string, 2)
2022-06-14 17:24:40 +08:00
m[constant.OnlineStatus] = memberUserIDList
2022-09-19 22:16:45 +08:00
t1 = time.Now()
2022-05-22 15:57:02 +08:00
2022-04-29 18:47:55 +08:00
//split parallel send
var wg sync.WaitGroup
var sendTag bool
2022-05-22 18:47:16 +08:00
var split = 20
2022-05-13 21:05:39 +08:00
for k, v := range m {
remain := len(v) % split
for i := 0; i < len(v)/split; i++ {
wg.Add(1)
2022-05-22 18:28:51 +08:00
tmp := valueCopy(pb)
// go rpc.sendMsgToGroup(v[i*split:(i+1)*split], *pb, k, &sendTag, &wg)
2023-01-11 16:23:16 +08:00
go rpc.sendMsgToGroupOptimization(ctx, v[i*split:(i+1)*split], tmp, k, &sendTag, &wg)
2022-05-13 21:05:39 +08:00
}
if remain > 0 {
wg.Add(1)
2022-05-22 18:28:51 +08:00
tmp := valueCopy(pb)
// go rpc.sendMsgToGroup(v[split*(len(v)/split):], *pb, k, &sendTag, &wg)
2023-01-11 16:23:16 +08:00
go rpc.sendMsgToGroupOptimization(ctx, v[split*(len(v)/split):], tmp, k, &sendTag, &wg)
2022-05-13 21:05:39 +08:00
}
2022-02-09 12:50:36 +08:00
}
2022-09-19 22:16:45 +08:00
log.Debug(pb.OperationID, "send msg cost time22 ", time.Since(t1), pb.MsgData.ClientMsgID, "uidList : ", len(addUidList))
2022-07-01 20:13:36 +08:00
//wg.Add(1)
//go rpc.sendMsgToGroup(addUidList, *pb, constant.OnlineStatus, &sendTag, &wg)
2022-07-01 19:48:03 +08:00
wg.Wait()
2022-09-19 22:16:45 +08:00
t1 = time.Now()
2022-03-04 14:27:54 +08:00
// callback
2022-05-26 18:02:00 +08:00
callbackResp = callbackAfterSendGroupMsg(pb)
if callbackResp.ErrCode != 0 {
log.NewError(pb.OperationID, utils.GetSelfFuncName(), "callbackAfterSendGroupMsg resp: ", callbackResp)
2022-02-28 17:57:03 +08:00
}
2022-04-29 18:47:55 +08:00
if !sendTag {
2022-07-01 19:29:17 +08:00
log.NewWarn(pb.OperationID, "send tag is ", sendTag)
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.GroupChatMsgProcessFailedCounter)
2022-04-29 18:47:55 +08:00
return returnMsg(&replay, pb, 201, "kafka send msg err", "", 0)
} else {
if pb.MsgData.ContentType == constant.AtText {
go func() {
var conversationReq pbConversation.ModifyConversationFieldReq
var tag bool
var atUserID []string
conversation := pbConversation.Conversation{
OwnerUserID: pb.MsgData.SendID,
ConversationID: utils.GetConversationIDBySessionType(pb.MsgData.GroupID, constant.GroupChatType),
ConversationType: constant.GroupChatType,
GroupID: pb.MsgData.GroupID,
}
conversationReq.Conversation = &conversation
conversationReq.OperationID = pb.OperationID
conversationReq.FieldType = constant.FieldGroupAtType
tagAll := utils.IsContain(constant.AtAllString, pb.MsgData.AtUserIDList)
if tagAll {
atUserID = utils.DifferenceString([]string{constant.AtAllString}, pb.MsgData.AtUserIDList)
if len(atUserID) == 0 { //just @everyone
conversationReq.UserIDList = memberUserIDList
conversation.GroupAtType = constant.AtAll
} else { //@Everyone and @other people
conversationReq.UserIDList = atUserID
conversation.GroupAtType = constant.AtAllAtMe
tag = true
}
} else {
conversationReq.UserIDList = pb.MsgData.AtUserIDList
conversation.GroupAtType = constant.AtMe
}
2023-02-07 20:24:20 +08:00
etcdConn, err := rpc.GetConn(ctx, config.Config.RpcRegisterName.OpenImConversationName)
2023-01-11 11:15:46 +08:00
if err != nil {
2022-08-17 12:12:54 +08:00
errMsg := pb.OperationID + "getcdv3.GetDefaultConn == nil"
2022-06-16 14:26:03 +08:00
log.NewError(pb.OperationID, errMsg)
return
}
2022-04-29 18:47:55 +08:00
client := pbConversation.NewConversationClient(etcdConn)
conversationReply, err := client.ModifyConversationField(context.Background(), &conversationReq)
if err != nil {
log.NewError(conversationReq.OperationID, "ModifyConversationField rpc failed, ", conversationReq.String(), err.Error())
} else if conversationReply.CommonResp.ErrCode != 0 {
log.NewError(conversationReq.OperationID, "ModifyConversationField rpc failed, ", conversationReq.String(), conversationReply.String())
}
if tag {
conversationReq.UserIDList = utils.DifferenceString(atUserID, memberUserIDList)
conversation.GroupAtType = constant.AtAll
2023-02-07 20:24:20 +08:00
etcdConn := rpc.GetDefaultConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImConversationName, pb.OperationID)
2022-06-16 14:26:03 +08:00
if etcdConn == nil {
2022-08-17 12:12:54 +08:00
errMsg := pb.OperationID + "getcdv3.GetDefaultConn == nil"
2022-06-16 14:26:03 +08:00
log.NewError(pb.OperationID, errMsg)
return
}
2022-04-29 18:47:55 +08:00
client := pbConversation.NewConversationClient(etcdConn)
conversationReply, err := client.ModifyConversationField(context.Background(), &conversationReq)
if err != nil {
log.NewError(conversationReq.OperationID, "ModifyConversationField rpc failed, ", conversationReq.String(), err.Error())
} else if conversationReply.CommonResp.ErrCode != 0 {
log.NewError(conversationReq.OperationID, "ModifyConversationField rpc failed, ", conversationReq.String(), conversationReply.String())
}
}
}()
}
2022-09-19 22:16:45 +08:00
log.Debug(pb.OperationID, "send msg cost time3 ", time.Since(t1), pb.MsgData.ClientMsgID)
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.GroupChatMsgProcessSuccessCounter)
2022-05-01 12:04:31 +08:00
return returnMsg(&replay, pb, 0, "", msgToMQSingle.MsgData.ServerMsgID, msgToMQSingle.MsgData.SendTime)
2022-04-29 18:47:55 +08:00
}
2022-03-30 18:23:05 +08:00
case constant.NotificationChatType:
2022-09-19 22:16:45 +08:00
t1 = time.Now()
2022-05-01 12:04:31 +08:00
msgToMQSingle.MsgData = pb.MsgData
log.NewInfo(msgToMQSingle.OperationID, msgToMQSingle)
2023-01-11 11:15:46 +08:00
err1 := rpc.sendMsgToWriter(ctx, &msgToMQSingle, msgToMQSingle.MsgData.RecvID, constant.OnlineStatus)
2022-03-30 18:23:05 +08:00
if err1 != nil {
2022-05-01 12:04:31 +08:00
log.NewError(msgToMQSingle.OperationID, "kafka send msg err:RecvID", msgToMQSingle.MsgData.RecvID, msgToMQSingle.String())
2022-03-30 18:23:05 +08:00
return returnMsg(&replay, pb, 201, "kafka send msg err", "", 0)
}
2022-05-01 12:04:31 +08:00
if msgToMQSingle.MsgData.SendID != msgToMQSingle.MsgData.RecvID { //Filter messages sent to yourself
2023-01-11 11:15:46 +08:00
err2 := rpc.sendMsgToWriter(ctx, &msgToMQSingle, msgToMQSingle.MsgData.SendID, constant.OnlineStatus)
2022-03-30 18:23:05 +08:00
if err2 != nil {
2022-05-01 12:04:31 +08:00
log.NewError(msgToMQSingle.OperationID, "kafka send msg err:SendID", msgToMQSingle.MsgData.SendID, msgToMQSingle.String())
2022-03-30 18:23:05 +08:00
return returnMsg(&replay, pb, 201, "kafka send msg err", "", 0)
}
}
2022-05-22 15:09:21 +08:00
2022-09-19 22:16:45 +08:00
log.Debug(pb.OperationID, "send msg cost time ", time.Since(t1), pb.MsgData.ClientMsgID)
2022-05-01 12:04:31 +08:00
return returnMsg(&replay, pb, 0, "", msgToMQSingle.MsgData.ServerMsgID, msgToMQSingle.MsgData.SendTime)
2022-05-28 18:10:08 +08:00
case constant.SuperGroupChatType:
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.WorkSuperGroupChatMsgRecvSuccessCounter)
2022-05-28 18:10:08 +08:00
// callback
2022-06-30 18:50:10 +08:00
callbackResp := callbackBeforeSendGroupMsg(pb)
2022-05-28 18:10:08 +08:00
if callbackResp.ErrCode != 0 {
log.NewError(pb.OperationID, utils.GetSelfFuncName(), "callbackBeforeSendSuperGroupMsg resp: ", callbackResp)
}
if callbackResp.ActionCode != constant.ActionAllow {
if callbackResp.ErrCode == 0 {
callbackResp.ErrCode = 201
}
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.WorkSuperGroupChatMsgProcessFailedCounter)
2022-05-28 18:10:08 +08:00
log.NewDebug(pb.OperationID, utils.GetSelfFuncName(), "callbackBeforeSendSuperGroupMsg result", "end rpc and return", callbackResp)
return returnMsg(&replay, pb, int32(callbackResp.ErrCode), callbackResp.ErrMsg, "", 0)
}
2023-01-11 11:15:46 +08:00
if flag, errCode, errMsg, _ = rpc.messageVerification(ctx, pb); !flag {
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.WorkSuperGroupChatMsgProcessFailedCounter)
2022-07-25 19:19:00 +08:00
return returnMsg(&replay, pb, errCode, errMsg, "", 0)
}
2022-05-28 18:10:08 +08:00
msgToMQSingle.MsgData = pb.MsgData
log.NewInfo(msgToMQSingle.OperationID, msgToMQSingle)
2023-01-11 16:23:16 +08:00
err1 := rpc.sendMsgToWriter(ctx, &msgToMQSingle, msgToMQSingle.MsgData.GroupID, constant.OnlineStatus)
2022-07-25 21:36:21 +08:00
if err1 != nil {
log.NewError(msgToMQSingle.OperationID, "kafka send msg err:RecvID", msgToMQSingle.MsgData.RecvID, msgToMQSingle.String())
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.WorkSuperGroupChatMsgProcessFailedCounter)
2022-07-25 21:36:21 +08:00
return returnMsg(&replay, pb, 201, "kafka send msg err", "", 0)
}
2022-08-09 15:12:13 +08:00
// callback
callbackResp = callbackAfterSendGroupMsg(pb)
if callbackResp.ErrCode != 0 {
log.NewError(pb.OperationID, utils.GetSelfFuncName(), "callbackAfterSendSuperGroupMsg resp: ", callbackResp)
}
2022-09-15 12:07:28 +08:00
promePkg.PromeInc(promePkg.WorkSuperGroupChatMsgProcessSuccessCounter)
2022-05-28 18:10:08 +08:00
return returnMsg(&replay, pb, 0, "", msgToMQSingle.MsgData.ServerMsgID, msgToMQSingle.MsgData.SendTime)
2021-10-21 12:10:38 +08:00
default:
2022-05-23 08:44:20 +08:00
return returnMsg(&replay, pb, 203, "unknown sessionType", "", 0)
2021-12-20 16:29:24 +08:00
}
2021-05-26 19:40:38 +08:00
}
2021-12-10 17:30:11 +08:00
2023-01-11 11:15:46 +08:00
func (rpc *rpcChat) sendMsgToWriter(ctx context.Context, m *pbChat.MsgDataToMQ, key string, status string) error {
2022-05-11 18:33:48 +08:00
switch status {
case constant.OnlineStatus:
2022-09-19 22:16:45 +08:00
if m.MsgData.ContentType == constant.SignalingNotification {
rpcPushMsg := pbPush.PushMsgReq{OperationID: m.OperationID, MsgData: m.MsgData, PushToUserID: key}
2023-02-07 20:24:20 +08:00
grpcConn, err := rpc.GetConn(ctx, config.Config.RpcRegisterName.OpenImPushName)
2023-01-11 11:15:46 +08:00
if err != nil {
return err
2022-09-19 22:16:45 +08:00
}
msgClient := pbPush.NewPushMsgServiceClient(grpcConn)
2023-01-11 11:15:46 +08:00
_, err = msgClient.PushMsg(context.Background(), &rpcPushMsg)
2022-09-19 22:16:45 +08:00
if err != nil {
log.Error(rpcPushMsg.OperationID, "rpc send failed", rpcPushMsg.OperationID, "push data", rpcPushMsg.String(), "err", err.Error())
return err
} else {
return nil
}
}
2022-09-29 17:57:26 +08:00
pid, offset, err := rpc.messageWriter.SendMessage(m, key, m.OperationID)
2022-05-11 18:33:48 +08:00
if err != nil {
log.Error(m.OperationID, "kafka send failed", "send data", m.String(), "pid", pid, "offset", offset, "err", err.Error(), "key", key, status)
2022-05-21 10:59:44 +08:00
} else {
2022-09-29 17:57:26 +08:00
// log.NewWarn(m.OperationID, "sendMsgToWriter client msgID ", m.MsgData.ClientMsgID)
2022-05-11 18:33:48 +08:00
}
return err
case constant.OfflineStatus:
2022-09-29 17:57:26 +08:00
pid, offset, err := rpc.messageWriter.SendMessage(m, key, m.OperationID)
2022-05-11 18:33:48 +08:00
if err != nil {
log.Error(m.OperationID, "kafka send failed", "send data", m.String(), "pid", pid, "offset", offset, "err", err.Error(), "key", key, status)
}
return err
2021-05-26 19:40:38 +08:00
}
2022-05-11 18:33:48 +08:00
return errors.New("status error")
2021-05-26 19:40:38 +08:00
}
func GetMsgID(sendID string) string {
t := time.Now().Format("2006-01-02 15:04:05")
2022-02-25 18:35:04 +08:00
return utils.Md5(t + "-" + sendID + "-" + strconv.Itoa(rand.Int()))
2021-05-26 19:40:38 +08:00
}
2021-12-21 21:40:50 +08:00
2021-12-23 17:19:57 +08:00
func returnMsg(replay *pbChat.SendMsgResp, pb *pbChat.SendMsgReq, errCode int32, errMsg, serverMsgID string, sendTime int64) (*pbChat.SendMsgResp, error) {
2021-06-28 15:35:18 +08:00
replay.ErrCode = errCode
replay.ErrMsg = errMsg
replay.ServerMsgID = serverMsgID
2021-12-23 17:19:57 +08:00
replay.ClientMsgID = pb.MsgData.ClientMsgID
2021-06-28 15:35:18 +08:00
replay.SendTime = sendTime
return replay, nil
}
2021-12-21 21:40:50 +08:00
2021-12-23 17:19:57 +08:00
func modifyMessageByUserMessageReceiveOpt(userID, sourceID string, sessionType int, pb *pbChat.SendMsgReq) bool {
2022-06-16 12:53:39 +08:00
opt, err := db.DB.GetUserGlobalMsgRecvOpt(userID)
if err != nil {
log.NewError(pb.OperationID, "GetUserGlobalMsgRecvOpt from redis err", userID, pb.String(), err.Error())
}
switch opt {
case constant.ReceiveMessage:
case constant.NotReceiveMessage:
return false
case constant.ReceiveNotNotifyMessage:
if pb.MsgData.Options == nil {
pb.MsgData.Options = make(map[string]bool, 10)
}
utils.SetSwitchFromOptions(pb.MsgData.Options, constant.IsOfflinePush, false)
return true
}
2021-12-06 20:03:59 +08:00
conversationID := utils.GetConversationIDBySessionType(sourceID, sessionType)
2022-06-16 12:53:39 +08:00
singleOpt, sErr := db.DB.GetSingleConversationRecvMsgOpt(userID, conversationID)
if sErr != nil && sErr != go_redis.Nil {
log.NewError(pb.OperationID, "GetSingleConversationMsgOpt from redis err", conversationID, pb.String(), sErr.Error())
2021-12-06 20:03:59 +08:00
return true
}
2022-06-16 12:53:39 +08:00
switch singleOpt {
2021-12-06 20:03:59 +08:00
case constant.ReceiveMessage:
return true
case constant.NotReceiveMessage:
2022-08-05 19:10:08 +08:00
if utils.IsContainInt(int(pb.MsgData.ContentType), ExcludeContentType) {
return true
}
2021-12-06 20:03:59 +08:00
return false
case constant.ReceiveNotNotifyMessage:
2021-12-23 17:19:57 +08:00
if pb.MsgData.Options == nil {
pb.MsgData.Options = make(map[string]bool, 10)
2021-12-08 18:30:23 +08:00
}
2021-12-23 17:19:57 +08:00
utils.SetSwitchFromOptions(pb.MsgData.Options, constant.IsOfflinePush, false)
2021-12-06 20:03:59 +08:00
return true
}
return true
}
2021-12-21 21:40:50 +08:00
2022-05-22 18:28:51 +08:00
func modifyMessageByUserMessageReceiveOptoptimization(userID, sourceID string, sessionType int, operationID string, options *map[string]bool) bool {
conversationID := utils.GetConversationIDBySessionType(sourceID, sessionType)
opt, err := db.DB.GetSingleConversationRecvMsgOpt(userID, conversationID)
2022-06-15 16:50:45 +08:00
if err != nil && err != go_redis.Nil {
2022-05-22 18:28:51 +08:00
log.NewError(operationID, "GetSingleConversationMsgOpt from redis err", userID, conversationID, err.Error())
return true
}
switch opt {
case constant.ReceiveMessage:
return true
case constant.NotReceiveMessage:
return false
case constant.ReceiveNotNotifyMessage:
if *options == nil {
*options = make(map[string]bool, 10)
}
utils.SetSwitchFromOptions(*options, constant.IsOfflinePush, false)
return true
}
return true
}
2021-12-21 21:40:50 +08:00
type NotificationMsg struct {
2022-06-22 19:48:35 +08:00
SendID string
RecvID string
Content []byte // open_im_sdk.TipsComm
MsgFrom int32
ContentType int32
SessionType int32
OperationID string
SenderNickname string
SenderFaceURL string
2021-12-21 21:40:50 +08:00
}
2022-01-14 13:44:46 +08:00
func Notification(n *NotificationMsg) {
2021-12-23 17:19:57 +08:00
var req pbChat.SendMsgReq
var msg sdk_ws.MsgData
var offlineInfo sdk_ws.OfflinePushInfo
2021-12-27 11:49:01 +08:00
var title, desc, ex string
2022-02-08 15:30:48 +08:00
var pushSwitch, unReadCount bool
var reliabilityLevel int
2021-12-23 17:19:57 +08:00
req.OperationID = n.OperationID
msg.SendID = n.SendID
msg.RecvID = n.RecvID
msg.Content = n.Content
msg.MsgFrom = n.MsgFrom
msg.ContentType = n.ContentType
msg.SessionType = n.SessionType
2022-01-18 11:54:48 +08:00
msg.CreateTime = utils.GetCurrentTimestampByMill()
2021-12-23 17:19:57 +08:00
msg.ClientMsgID = utils.GetMsgID(n.SendID)
2022-02-08 15:30:48 +08:00
msg.Options = make(map[string]bool, 7)
2022-06-22 19:48:35 +08:00
msg.SenderNickname = n.SenderNickname
msg.SenderFaceURL = n.SenderFaceURL
2021-12-23 17:19:57 +08:00
switch n.SessionType {
2022-06-07 19:02:19 +08:00
case constant.GroupChatType, constant.SuperGroupChatType:
2021-12-23 17:19:57 +08:00
msg.RecvID = ""
msg.GroupID = n.RecvID
}
offlineInfo.IOSBadgeCount = config.Config.IOSPush.BadgeCount
offlineInfo.IOSPushSound = config.Config.IOSPush.PushSound
2022-02-08 15:30:48 +08:00
switch msg.ContentType {
case constant.GroupCreatedNotification:
pushSwitch = config.Config.Notification.GroupCreated.OfflinePush.PushSwitch
title = config.Config.Notification.GroupCreated.OfflinePush.Title
desc = config.Config.Notification.GroupCreated.OfflinePush.Desc
ex = config.Config.Notification.GroupCreated.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.GroupCreated.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.GroupCreated.Conversation.UnreadCount
case constant.GroupInfoSetNotification:
pushSwitch = config.Config.Notification.GroupInfoSet.OfflinePush.PushSwitch
title = config.Config.Notification.GroupInfoSet.OfflinePush.Title
desc = config.Config.Notification.GroupInfoSet.OfflinePush.Desc
ex = config.Config.Notification.GroupInfoSet.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.GroupInfoSet.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.GroupInfoSet.Conversation.UnreadCount
case constant.JoinGroupApplicationNotification:
pushSwitch = config.Config.Notification.JoinGroupApplication.OfflinePush.PushSwitch
title = config.Config.Notification.JoinGroupApplication.OfflinePush.Title
desc = config.Config.Notification.JoinGroupApplication.OfflinePush.Desc
ex = config.Config.Notification.JoinGroupApplication.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.JoinGroupApplication.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.JoinGroupApplication.Conversation.UnreadCount
case constant.MemberQuitNotification:
pushSwitch = config.Config.Notification.MemberQuit.OfflinePush.PushSwitch
title = config.Config.Notification.MemberQuit.OfflinePush.Title
desc = config.Config.Notification.MemberQuit.OfflinePush.Desc
ex = config.Config.Notification.MemberQuit.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.MemberQuit.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.MemberQuit.Conversation.UnreadCount
case constant.GroupApplicationAcceptedNotification:
pushSwitch = config.Config.Notification.GroupApplicationAccepted.OfflinePush.PushSwitch
title = config.Config.Notification.GroupApplicationAccepted.OfflinePush.Title
desc = config.Config.Notification.GroupApplicationAccepted.OfflinePush.Desc
ex = config.Config.Notification.GroupApplicationAccepted.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.GroupApplicationAccepted.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.GroupApplicationAccepted.Conversation.UnreadCount
case constant.GroupApplicationRejectedNotification:
pushSwitch = config.Config.Notification.GroupApplicationRejected.OfflinePush.PushSwitch
title = config.Config.Notification.GroupApplicationRejected.OfflinePush.Title
desc = config.Config.Notification.GroupApplicationRejected.OfflinePush.Desc
ex = config.Config.Notification.GroupApplicationRejected.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.GroupApplicationRejected.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.GroupApplicationRejected.Conversation.UnreadCount
case constant.GroupOwnerTransferredNotification:
pushSwitch = config.Config.Notification.GroupOwnerTransferred.OfflinePush.PushSwitch
title = config.Config.Notification.GroupOwnerTransferred.OfflinePush.Title
desc = config.Config.Notification.GroupOwnerTransferred.OfflinePush.Desc
ex = config.Config.Notification.GroupOwnerTransferred.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.GroupOwnerTransferred.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.GroupOwnerTransferred.Conversation.UnreadCount
case constant.MemberKickedNotification:
pushSwitch = config.Config.Notification.MemberKicked.OfflinePush.PushSwitch
title = config.Config.Notification.MemberKicked.OfflinePush.Title
desc = config.Config.Notification.MemberKicked.OfflinePush.Desc
ex = config.Config.Notification.MemberKicked.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.MemberKicked.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.MemberKicked.Conversation.UnreadCount
case constant.MemberInvitedNotification:
pushSwitch = config.Config.Notification.MemberInvited.OfflinePush.PushSwitch
title = config.Config.Notification.MemberInvited.OfflinePush.Title
desc = config.Config.Notification.MemberInvited.OfflinePush.Desc
ex = config.Config.Notification.MemberInvited.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.MemberInvited.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.MemberInvited.Conversation.UnreadCount
case constant.MemberEnterNotification:
pushSwitch = config.Config.Notification.MemberEnter.OfflinePush.PushSwitch
title = config.Config.Notification.MemberEnter.OfflinePush.Title
desc = config.Config.Notification.MemberEnter.OfflinePush.Desc
ex = config.Config.Notification.MemberEnter.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.MemberEnter.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.MemberEnter.Conversation.UnreadCount
case constant.UserInfoUpdatedNotification:
pushSwitch = config.Config.Notification.UserInfoUpdated.OfflinePush.PushSwitch
title = config.Config.Notification.UserInfoUpdated.OfflinePush.Title
desc = config.Config.Notification.UserInfoUpdated.OfflinePush.Desc
ex = config.Config.Notification.UserInfoUpdated.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.UserInfoUpdated.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.UserInfoUpdated.Conversation.UnreadCount
case constant.FriendApplicationNotification:
pushSwitch = config.Config.Notification.FriendApplication.OfflinePush.PushSwitch
title = config.Config.Notification.FriendApplication.OfflinePush.Title
desc = config.Config.Notification.FriendApplication.OfflinePush.Desc
ex = config.Config.Notification.FriendApplication.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.FriendApplication.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.FriendApplication.Conversation.UnreadCount
2022-02-08 18:06:35 +08:00
case constant.FriendApplicationApprovedNotification:
pushSwitch = config.Config.Notification.FriendApplicationApproved.OfflinePush.PushSwitch
title = config.Config.Notification.FriendApplicationApproved.OfflinePush.Title
desc = config.Config.Notification.FriendApplicationApproved.OfflinePush.Desc
ex = config.Config.Notification.FriendApplicationApproved.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.FriendApplicationApproved.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.FriendApplicationApproved.Conversation.UnreadCount
case constant.FriendApplicationRejectedNotification:
pushSwitch = config.Config.Notification.FriendApplicationRejected.OfflinePush.PushSwitch
title = config.Config.Notification.FriendApplicationRejected.OfflinePush.Title
desc = config.Config.Notification.FriendApplicationRejected.OfflinePush.Desc
ex = config.Config.Notification.FriendApplicationRejected.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.FriendApplicationRejected.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.FriendApplicationRejected.Conversation.UnreadCount
case constant.FriendAddedNotification:
pushSwitch = config.Config.Notification.FriendAdded.OfflinePush.PushSwitch
title = config.Config.Notification.FriendAdded.OfflinePush.Title
desc = config.Config.Notification.FriendAdded.OfflinePush.Desc
ex = config.Config.Notification.FriendAdded.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.FriendAdded.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.FriendAdded.Conversation.UnreadCount
case constant.FriendDeletedNotification:
pushSwitch = config.Config.Notification.FriendDeleted.OfflinePush.PushSwitch
title = config.Config.Notification.FriendDeleted.OfflinePush.Title
desc = config.Config.Notification.FriendDeleted.OfflinePush.Desc
ex = config.Config.Notification.FriendDeleted.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.FriendDeleted.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.FriendDeleted.Conversation.UnreadCount
case constant.FriendRemarkSetNotification:
pushSwitch = config.Config.Notification.FriendRemarkSet.OfflinePush.PushSwitch
title = config.Config.Notification.FriendRemarkSet.OfflinePush.Title
desc = config.Config.Notification.FriendRemarkSet.OfflinePush.Desc
ex = config.Config.Notification.FriendRemarkSet.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.FriendRemarkSet.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.FriendRemarkSet.Conversation.UnreadCount
case constant.BlackAddedNotification:
pushSwitch = config.Config.Notification.BlackAdded.OfflinePush.PushSwitch
title = config.Config.Notification.BlackAdded.OfflinePush.Title
desc = config.Config.Notification.BlackAdded.OfflinePush.Desc
ex = config.Config.Notification.BlackAdded.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.BlackAdded.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.BlackAdded.Conversation.UnreadCount
case constant.BlackDeletedNotification:
pushSwitch = config.Config.Notification.BlackDeleted.OfflinePush.PushSwitch
title = config.Config.Notification.BlackDeleted.OfflinePush.Title
desc = config.Config.Notification.BlackDeleted.OfflinePush.Desc
ex = config.Config.Notification.BlackDeleted.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.BlackDeleted.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.BlackDeleted.Conversation.UnreadCount
2022-02-15 14:11:20 +08:00
case constant.ConversationOptChangeNotification:
pushSwitch = config.Config.Notification.ConversationOptUpdate.OfflinePush.PushSwitch
title = config.Config.Notification.ConversationOptUpdate.OfflinePush.Title
desc = config.Config.Notification.ConversationOptUpdate.OfflinePush.Desc
ex = config.Config.Notification.ConversationOptUpdate.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.ConversationOptUpdate.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.ConversationOptUpdate.Conversation.UnreadCount
2022-03-24 15:33:30 +08:00
case constant.GroupDismissedNotification:
pushSwitch = config.Config.Notification.GroupDismissed.OfflinePush.PushSwitch
title = config.Config.Notification.GroupDismissed.OfflinePush.Title
desc = config.Config.Notification.GroupDismissed.OfflinePush.Desc
ex = config.Config.Notification.GroupDismissed.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.GroupDismissed.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.GroupDismissed.Conversation.UnreadCount
2022-03-29 16:04:15 +08:00
case constant.GroupMutedNotification:
pushSwitch = config.Config.Notification.GroupMuted.OfflinePush.PushSwitch
title = config.Config.Notification.GroupMuted.OfflinePush.Title
desc = config.Config.Notification.GroupMuted.OfflinePush.Desc
ex = config.Config.Notification.GroupMuted.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.GroupMuted.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.GroupMuted.Conversation.UnreadCount
case constant.GroupCancelMutedNotification:
pushSwitch = config.Config.Notification.GroupCancelMuted.OfflinePush.PushSwitch
title = config.Config.Notification.GroupCancelMuted.OfflinePush.Title
desc = config.Config.Notification.GroupCancelMuted.OfflinePush.Desc
ex = config.Config.Notification.GroupCancelMuted.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.GroupCancelMuted.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.GroupCancelMuted.Conversation.UnreadCount
case constant.GroupMemberMutedNotification:
pushSwitch = config.Config.Notification.GroupMemberMuted.OfflinePush.PushSwitch
title = config.Config.Notification.GroupMemberMuted.OfflinePush.Title
desc = config.Config.Notification.GroupMemberMuted.OfflinePush.Desc
ex = config.Config.Notification.GroupMemberMuted.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.GroupMemberMuted.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.GroupMemberMuted.Conversation.UnreadCount
case constant.GroupMemberCancelMutedNotification:
pushSwitch = config.Config.Notification.GroupMemberCancelMuted.OfflinePush.PushSwitch
title = config.Config.Notification.GroupMemberCancelMuted.OfflinePush.Title
desc = config.Config.Notification.GroupMemberCancelMuted.OfflinePush.Desc
ex = config.Config.Notification.GroupMemberCancelMuted.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.GroupMemberCancelMuted.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.GroupMemberCancelMuted.Conversation.UnreadCount
2022-04-19 16:40:57 +08:00
2022-04-19 18:46:02 +08:00
case constant.GroupMemberInfoSetNotification:
pushSwitch = config.Config.Notification.GroupMemberInfoSet.OfflinePush.PushSwitch
title = config.Config.Notification.GroupMemberInfoSet.OfflinePush.Title
desc = config.Config.Notification.GroupMemberInfoSet.OfflinePush.Desc
ex = config.Config.Notification.GroupMemberInfoSet.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.GroupMemberInfoSet.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.GroupMemberInfoSet.Conversation.UnreadCount
case constant.OrganizationChangedNotification:
pushSwitch = config.Config.Notification.OrganizationChanged.OfflinePush.PushSwitch
title = config.Config.Notification.OrganizationChanged.OfflinePush.Title
desc = config.Config.Notification.OrganizationChanged.OfflinePush.Desc
ex = config.Config.Notification.OrganizationChanged.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.OrganizationChanged.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.OrganizationChanged.Conversation.UnreadCount
2022-04-22 17:19:35 +08:00
case constant.WorkMomentNotification:
pushSwitch = config.Config.Notification.WorkMomentsNotification.OfflinePush.PushSwitch
title = config.Config.Notification.WorkMomentsNotification.OfflinePush.Title
desc = config.Config.Notification.WorkMomentsNotification.OfflinePush.Desc
ex = config.Config.Notification.WorkMomentsNotification.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.WorkMomentsNotification.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.WorkMomentsNotification.Conversation.UnreadCount
case constant.ConversationPrivateChatNotification:
pushSwitch = config.Config.Notification.ConversationSetPrivate.OfflinePush.PushSwitch
title = config.Config.Notification.ConversationSetPrivate.OfflinePush.Title
desc = config.Config.Notification.ConversationSetPrivate.OfflinePush.Desc
ex = config.Config.Notification.ConversationSetPrivate.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.ConversationSetPrivate.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.ConversationSetPrivate.Conversation.UnreadCount
2022-12-07 10:56:57 +08:00
case constant.FriendInfoUpdatedNotification:
pushSwitch = config.Config.Notification.FriendInfoUpdated.OfflinePush.PushSwitch
title = config.Config.Notification.FriendInfoUpdated.OfflinePush.Title
desc = config.Config.Notification.FriendInfoUpdated.OfflinePush.Desc
ex = config.Config.Notification.FriendInfoUpdated.OfflinePush.Ext
reliabilityLevel = config.Config.Notification.FriendInfoUpdated.Conversation.ReliabilityLevel
unReadCount = config.Config.Notification.FriendInfoUpdated.Conversation.UnreadCount
2022-08-17 10:18:01 +08:00
case constant.DeleteMessageNotification:
2022-05-25 14:55:14 +08:00
reliabilityLevel = constant.ReliableNotificationNoMsg
2022-08-17 10:18:01 +08:00
case constant.ConversationUnreadNotification, constant.SuperGroupUpdateNotification:
2022-06-07 19:02:19 +08:00
reliabilityLevel = constant.UnreliableNotification
2022-02-08 15:30:48 +08:00
}
switch reliabilityLevel {
case constant.UnreliableNotification:
utils.SetSwitchFromOptions(msg.Options, constant.IsHistory, false)
utils.SetSwitchFromOptions(msg.Options, constant.IsPersistent, false)
utils.SetSwitchFromOptions(msg.Options, constant.IsConversationUpdate, false)
2022-03-28 18:30:12 +08:00
utils.SetSwitchFromOptions(msg.Options, constant.IsSenderConversationUpdate, false)
2022-02-08 15:30:48 +08:00
case constant.ReliableNotificationNoMsg:
utils.SetSwitchFromOptions(msg.Options, constant.IsConversationUpdate, false)
2022-03-28 18:30:12 +08:00
utils.SetSwitchFromOptions(msg.Options, constant.IsSenderConversationUpdate, false)
2022-02-08 15:30:48 +08:00
case constant.ReliableNotificationMsg:
2022-02-08 18:06:35 +08:00
2022-02-08 15:30:48 +08:00
}
utils.SetSwitchFromOptions(msg.Options, constant.IsUnreadCount, unReadCount)
2021-12-23 17:19:57 +08:00
utils.SetSwitchFromOptions(msg.Options, constant.IsOfflinePush, pushSwitch)
offlineInfo.Title = title
offlineInfo.Desc = desc
2021-12-27 11:49:01 +08:00
offlineInfo.Ex = ex
2021-12-23 17:19:57 +08:00
msg.OfflinePushInfo = &offlineInfo
req.MsgData = &msg
2023-02-07 20:24:20 +08:00
etcdConn := rpc.GetDefaultConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImMsgName, req.OperationID)
2022-06-16 14:26:03 +08:00
if etcdConn == nil {
2022-08-17 12:12:54 +08:00
errMsg := req.OperationID + "getcdv3.GetDefaultConn == nil"
2022-06-16 14:26:03 +08:00
log.NewError(req.OperationID, errMsg)
return
}
2022-07-20 21:15:30 +08:00
client := pbChat.NewMsgClient(etcdConn)
2021-12-23 17:19:57 +08:00
reply, err := client.SendMsg(context.Background(), &req)
if err != nil {
log.NewError(req.OperationID, "SendMsg rpc failed, ", req.String(), err.Error())
} else if reply.ErrCode != 0 {
2022-03-30 15:52:04 +08:00
log.NewError(req.OperationID, "SendMsg rpc failed, ", req.String(), reply.ErrCode, reply.ErrMsg)
2021-12-23 17:19:57 +08:00
}
2021-12-21 21:40:50 +08:00
}
2022-09-21 18:44:21 +08:00
2022-05-13 21:05:39 +08:00
func getOnlineAndOfflineUserIDList(memberList []string, m map[string][]string, operationID string) {
var onllUserIDList, offlUserIDList []string
2022-05-11 18:33:48 +08:00
var wsResult []*pbRelay.GetUsersOnlineStatusResp_SuccessResult
req := &pbRelay.GetUsersOnlineStatusReq{}
req.UserIDList = memberList
req.OperationID = operationID
req.OpUserID = config.Config.Manager.AppManagerUid[0]
flag := false
2023-02-07 20:24:20 +08:00
grpcCons := rpc.GetDefaultGatewayConn4Unique(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), operationID)
2022-05-11 18:33:48 +08:00
for _, v := range grpcCons {
2022-07-20 21:15:30 +08:00
client := pbRelay.NewRelayClient(v)
2022-05-11 18:33:48 +08:00
reply, err := client.GetUsersOnlineStatus(context.Background(), req)
if err != nil {
log.NewError(operationID, "GetUsersOnlineStatus rpc err", req.String(), err.Error())
continue
} else {
if reply.ErrCode == 0 {
wsResult = append(wsResult, reply.SuccessResult...)
}
}
}
log.NewInfo(operationID, "call GetUsersOnlineStatus rpc server is success", wsResult)
//Online data merge of each node
for _, v1 := range memberList {
flag = false
for _, v2 := range wsResult {
if v2.UserID == v1 {
flag = true
onllUserIDList = append(onllUserIDList, v1)
}
}
if !flag {
offlUserIDList = append(offlUserIDList, v1)
}
}
2022-05-13 21:05:39 +08:00
m[constant.OnlineStatus] = onllUserIDList
m[constant.OfflineStatus] = offlUserIDList
2022-05-11 18:33:48 +08:00
}
2022-05-22 18:28:51 +08:00
func valueCopy(pb *pbChat.SendMsgReq) *pbChat.SendMsgReq {
offlinePushInfo := sdk_ws.OfflinePushInfo{}
if pb.MsgData.OfflinePushInfo != nil {
offlinePushInfo = *pb.MsgData.OfflinePushInfo
}
msgData := sdk_ws.MsgData{}
msgData = *pb.MsgData
msgData.OfflinePushInfo = &offlinePushInfo
options := make(map[string]bool, 10)
for key, value := range pb.MsgData.Options {
options[key] = value
}
msgData.Options = options
2022-05-23 09:54:43 +08:00
return &pbChat.SendMsgReq{Token: pb.Token, OperationID: pb.OperationID, MsgData: &msgData}
2022-05-22 18:28:51 +08:00
}
2023-01-11 16:23:16 +08:00
func (rpc *rpcChat) sendMsgToGroup(ctx context.Context, list []string, pb pbChat.SendMsgReq, status string, sendTag *bool, wg *sync.WaitGroup) {
2022-05-11 18:33:48 +08:00
// log.Debug(pb.OperationID, "split userID ", list)
2022-05-13 15:59:38 +08:00
offlinePushInfo := sdk_ws.OfflinePushInfo{}
2022-05-11 18:33:48 +08:00
if pb.MsgData.OfflinePushInfo != nil {
2022-05-13 15:59:38 +08:00
offlinePushInfo = *pb.MsgData.OfflinePushInfo
2022-05-11 18:33:48 +08:00
}
2022-05-13 15:59:38 +08:00
msgData := sdk_ws.MsgData{}
msgData = *pb.MsgData
msgData.OfflinePushInfo = &offlinePushInfo
groupPB := pbChat.SendMsgReq{Token: pb.Token, OperationID: pb.OperationID, MsgData: &msgData}
msgToMQGroup := pbChat.MsgDataToMQ{Token: pb.Token, OperationID: pb.OperationID, MsgData: &msgData}
2022-05-11 18:33:48 +08:00
for _, v := range list {
2022-05-13 15:59:38 +08:00
options := make(map[string]bool, 10)
for key, value := range pb.MsgData.Options {
options[key] = value
}
2022-05-11 18:33:48 +08:00
groupPB.MsgData.RecvID = v
2022-05-13 15:59:38 +08:00
groupPB.MsgData.Options = options
isSend := modifyMessageByUserMessageReceiveOpt(v, msgData.GroupID, constant.GroupChatType, &groupPB)
2022-05-11 18:33:48 +08:00
if isSend {
msgToMQGroup.MsgData = groupPB.MsgData
2022-09-29 17:57:26 +08:00
// log.Debug(groupPB.OperationID, "sendMsgToWriter, ", v, groupID, msgToMQGroup.String())
2023-01-11 16:23:16 +08:00
err := rpc.sendMsgToWriter(ctx, &msgToMQGroup, v, status)
2022-05-11 18:33:48 +08:00
if err != nil {
log.NewError(msgToMQGroup.OperationID, "kafka send msg err:UserId", v, msgToMQGroup.String())
} else {
*sendTag = true
}
} else {
2022-09-29 17:57:26 +08:00
log.Debug(groupPB.OperationID, "not sendMsgToWriter, ", v)
2022-05-11 18:33:48 +08:00
}
}
wg.Done()
}
2022-05-22 18:28:51 +08:00
2023-01-11 16:23:16 +08:00
func (rpc *rpcChat) sendMsgToGroupOptimization(ctx context.Context, list []string, groupPB *pbChat.SendMsgReq, status string, sendTag *bool, wg *sync.WaitGroup) {
2022-05-22 18:28:51 +08:00
msgToMQGroup := pbChat.MsgDataToMQ{Token: groupPB.Token, OperationID: groupPB.OperationID, MsgData: groupPB.MsgData}
2022-08-09 15:54:16 +08:00
tempOptions := make(map[string]bool, 1)
for k, v := range groupPB.MsgData.Options {
tempOptions[k] = v
}
2022-05-22 18:28:51 +08:00
for _, v := range list {
groupPB.MsgData.RecvID = v
2022-08-09 15:41:16 +08:00
options := make(map[string]bool, 1)
2022-08-09 15:54:16 +08:00
for k, v := range tempOptions {
2022-08-09 15:41:16 +08:00
options[k] = v
}
groupPB.MsgData.Options = options
2022-05-22 18:28:51 +08:00
isSend := modifyMessageByUserMessageReceiveOpt(v, groupPB.MsgData.GroupID, constant.GroupChatType, groupPB)
if isSend {
2022-06-07 13:44:52 +08:00
if v == "" || groupPB.MsgData.SendID == "" {
2022-06-07 14:02:11 +08:00
log.Error(msgToMQGroup.OperationID, "sendMsgToGroupOptimization userID nil ", msgToMQGroup.String())
continue
2022-06-07 13:44:52 +08:00
}
2023-01-11 16:23:16 +08:00
err := rpc.sendMsgToWriter(ctx, &msgToMQGroup, v, status)
2022-05-22 18:28:51 +08:00
if err != nil {
log.NewError(msgToMQGroup.OperationID, "kafka send msg err:UserId", v, msgToMQGroup.String())
} else {
*sendTag = true
}
} else {
2022-09-29 17:57:26 +08:00
log.Debug(groupPB.OperationID, "not sendMsgToWriter, ", v)
2022-05-22 18:28:51 +08:00
}
}
wg.Done()
}