wangchuxiao
2 years ago
commit
8793271541
@ -0,0 +1,514 @@
|
|||||||
|
package cache
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
|
||||||
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"
|
||||||
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
|
||||||
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/sdkws"
|
||||||
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
|
||||||
|
"github.com/gogo/protobuf/jsonpb"
|
||||||
|
|
||||||
|
"google.golang.org/protobuf/proto"
|
||||||
|
|
||||||
|
"github.com/go-redis/redis/v8"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
NotificationUserIncrSeq = "REDIS_USER_INCR_SEQ:" // user incr seq
|
||||||
|
NotificationUserMinSeq = "REDIS_USER_MIN_SEQ:"
|
||||||
|
NotificationGetuiToken = "GETUI_TOKEN"
|
||||||
|
NotificationGetuiTaskID = "GETUI_TASK_ID"
|
||||||
|
NotificationMessageCache = "MESSAGE_CACHE:"
|
||||||
|
NotificationSignalCache = "SIGNAL_CACHE:"
|
||||||
|
NotificationSignalListCache = "SIGNAL_LIST_CACHE:"
|
||||||
|
NotificationFcmToken = "FCM_TOKEN:"
|
||||||
|
NotificationGroupUserMinSeq = "GROUP_USER_MIN_SEQ:"
|
||||||
|
NotificationGroupMaxSeq = "GROUP_MAX_SEQ:"
|
||||||
|
NotificationGroupMinSeq = "GROUP_MIN_SEQ:"
|
||||||
|
NotificationSendMsgFailedFlag = "SEND_MSG_FAILED_FLAG:"
|
||||||
|
NotificationUserBadgeUnreadCountSum = "USER_BADGE_UNREAD_COUNT_SUM:"
|
||||||
|
NotificationExTypeKeyLocker = "EX_LOCK:"
|
||||||
|
NotificationUidPidToken = "UID_PID_TOKEN_STATUS:"
|
||||||
|
)
|
||||||
|
|
||||||
|
type NotificationModel interface {
|
||||||
|
IncrUserSeq(ctx context.Context, userID string) (int64, error)
|
||||||
|
GetUserMaxSeq(ctx context.Context, userID string) (int64, error)
|
||||||
|
SetUserMaxSeq(ctx context.Context, userID string, maxSeq int64) error
|
||||||
|
SetUserMinSeq(ctx context.Context, userID string, minSeq int64) (err error)
|
||||||
|
GetUserMinSeq(ctx context.Context, userID string) (int64, error)
|
||||||
|
SetGroupUserMinSeq(ctx context.Context, groupID, userID string, minSeq int64) (err error)
|
||||||
|
GetGroupUserMinSeq(ctx context.Context, groupID, userID string) (int64, error)
|
||||||
|
GetGroupMaxSeq(ctx context.Context, groupID string) (int64, error)
|
||||||
|
GetGroupMinSeq(ctx context.Context, groupID string) (int64, error)
|
||||||
|
IncrGroupMaxSeq(ctx context.Context, groupID string) (int64, error)
|
||||||
|
SetGroupMaxSeq(ctx context.Context, groupID string, maxSeq int64) error
|
||||||
|
SetGroupMinSeq(ctx context.Context, groupID string, minSeq int64) error
|
||||||
|
AddTokenFlag(ctx context.Context, userID string, platformID int, token string, flag int) error
|
||||||
|
GetTokensWithoutError(ctx context.Context, userID, platformID string) (map[string]int, error)
|
||||||
|
SetTokenMapByUidPid(ctx context.Context, userID string, platform string, m map[string]int) error
|
||||||
|
DeleteTokenByUidPid(ctx context.Context, userID string, platform string, fields []string) error
|
||||||
|
GetMessagesBySeq(ctx context.Context, userID string, seqList []int64) (seqMsg []*sdkws.MsgData, failedSeqList []int64, err error)
|
||||||
|
SetMessageToCache(ctx context.Context, userID string, msgList []*sdkws.MsgData) (int, error)
|
||||||
|
DeleteMessageFromCache(ctx context.Context, userID string, msgList []*sdkws.MsgData) error
|
||||||
|
CleanUpOneUserAllMsg(ctx context.Context, userID string) error
|
||||||
|
HandleSignalInvite(ctx context.Context, msg *sdkws.MsgData, pushToUserID string) (isSend bool, err error)
|
||||||
|
GetSignalInvitationInfoByClientMsgID(ctx context.Context, clientMsgID string) (invitationInfo *sdkws.SignalInviteReq, err error)
|
||||||
|
GetAvailableSignalInvitationInfo(ctx context.Context, userID string) (invitationInfo *sdkws.SignalInviteReq, err error)
|
||||||
|
DelUserSignalList(ctx context.Context, userID string) error
|
||||||
|
DelMsgFromCache(ctx context.Context, userID string, seqList []int64) error
|
||||||
|
SetGetuiToken(ctx context.Context, token string, expireTime int64) error
|
||||||
|
GetGetuiToken(ctx context.Context) (string, error)
|
||||||
|
SetGetuiTaskID(ctx context.Context, taskID string, expireTime int64) error
|
||||||
|
GetGetuiTaskID(ctx context.Context) (string, error)
|
||||||
|
SetSendMsgStatus(ctx context.Context, id string, status int32) error
|
||||||
|
GetSendMsgStatus(ctx context.Context, id string) (int32, error)
|
||||||
|
SetFcmToken(ctx context.Context, account string, platformID int, fcmToken string, expireTime int64) (err error)
|
||||||
|
GetFcmToken(ctx context.Context, account string, platformID int) (string, error)
|
||||||
|
DelFcmToken(ctx context.Context, account string, platformID int) error
|
||||||
|
IncrUserBadgeUnreadCountSum(ctx context.Context, userID string) (int, error)
|
||||||
|
SetUserBadgeUnreadCountSum(ctx context.Context, userID string, value int) error
|
||||||
|
GetUserBadgeUnreadCountSum(ctx context.Context, userID string) (int, error)
|
||||||
|
JudgeMessageReactionExist(ctx context.Context, clientMsgID string, sessionType int32) (bool, error)
|
||||||
|
GetOneMessageAllReactionList(ctx context.Context, clientMsgID string, sessionType int32) (map[string]string, error)
|
||||||
|
DeleteOneMessageKey(ctx context.Context, clientMsgID string, sessionType int32, subKey string) error
|
||||||
|
SetMessageReactionExpire(ctx context.Context, clientMsgID string, sessionType int32, expiration time.Duration) (bool, error)
|
||||||
|
GetMessageTypeKeyValue(ctx context.Context, clientMsgID string, sessionType int32, typeKey string) (string, error)
|
||||||
|
SetMessageTypeKeyValue(ctx context.Context, clientMsgID string, sessionType int32, typeKey, value string) error
|
||||||
|
LockMessageTypeKey(ctx context.Context, clientMsgID string, TypeKey string) error
|
||||||
|
UnLockMessageTypeKey(ctx context.Context, clientMsgID string, TypeKey string) error
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewNotificationCacheModel(client redis.UniversalClient) NotificationModel {
|
||||||
|
return ¬ificationCache{rdb: client}
|
||||||
|
}
|
||||||
|
|
||||||
|
type notificationCache struct {
|
||||||
|
rdb redis.UniversalClient
|
||||||
|
}
|
||||||
|
|
||||||
|
// 兼容老版本调用
|
||||||
|
func (c *notificationCache) DelKeys() {
|
||||||
|
for _, key := range []string{"GROUP_CACHE:", "FRIEND_RELATION_CACHE:", "BLACK_LIST_CACHE:", "USER_INFO_CACHE:", "GROUP_INFO_CACHE:", "JOINED_GROUP_LIST_CACHE:",
|
||||||
|
"GROUP_MEMBER_INFO_CACHE:", "GROUP_ALL_MEMBER_INFO_CACHE:", "ALL_FRIEND_INFO_CACHE:"} {
|
||||||
|
fName := utils.GetSelfFuncName()
|
||||||
|
var cursor uint64
|
||||||
|
var n int
|
||||||
|
for {
|
||||||
|
var keys []string
|
||||||
|
var err error
|
||||||
|
keys, cursor, err = c.rdb.Scan(context.Background(), cursor, key+"*", scanCount).Result()
|
||||||
|
if err != nil {
|
||||||
|
panic(err.Error())
|
||||||
|
}
|
||||||
|
n += len(keys)
|
||||||
|
// for each for redis cluster
|
||||||
|
for _, key := range keys {
|
||||||
|
if err = c.rdb.Del(context.Background(), key).Err(); err != nil {
|
||||||
|
log.NewError("", fName, key, err.Error())
|
||||||
|
err = c.rdb.Del(context.Background(), key).Err()
|
||||||
|
if err != nil {
|
||||||
|
panic(err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if cursor == 0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) IncrUserSeq(ctx context.Context, userID string) (int64, error) {
|
||||||
|
return utils.Wrap2(c.rdb.Get(ctx, NotificationUserIncrSeq+userID).Int64())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetUserMaxSeq(ctx context.Context, userID string) (int64, error) {
|
||||||
|
return utils.Wrap2(c.rdb.Get(ctx, NotificationUserIncrSeq+userID).Int64())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetUserMaxSeq(ctx context.Context, userID string, maxSeq int64) error {
|
||||||
|
return errs.Wrap(c.rdb.Set(ctx, NotificationUserIncrSeq+userID, maxSeq, 0).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetUserMinSeq(ctx context.Context, userID string, minSeq int64) (err error) {
|
||||||
|
return errs.Wrap(c.rdb.Set(ctx, NotificationUserMinSeq+userID, minSeq, 0).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetUserMinSeq(ctx context.Context, userID string) (int64, error) {
|
||||||
|
return utils.Wrap2(c.rdb.Get(ctx, NotificationUserMinSeq+userID).Int64())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetGroupUserMinSeq(ctx context.Context, groupID, userID string, minSeq int64) (err error) {
|
||||||
|
key := NotificationGroupUserMinSeq + "g:" + groupID + "u:" + userID
|
||||||
|
return errs.Wrap(c.rdb.Set(ctx, key, minSeq, 0).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetGroupUserMinSeq(ctx context.Context, groupID, userID string) (int64, error) {
|
||||||
|
key := NotificationGroupUserMinSeq + "g:" + groupID + "u:" + userID
|
||||||
|
return utils.Wrap2(c.rdb.Get(ctx, key).Int64())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetGroupMaxSeq(ctx context.Context, groupID string) (int64, error) {
|
||||||
|
return utils.Wrap2(c.rdb.Get(ctx, NotificationGroupMaxSeq+groupID).Int64())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetGroupMinSeq(ctx context.Context, groupID string) (int64, error) {
|
||||||
|
return utils.Wrap2(c.rdb.Get(ctx, NotificationGroupMinSeq+groupID).Int64())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) IncrGroupMaxSeq(ctx context.Context, groupID string) (int64, error) {
|
||||||
|
key := NotificationGroupMaxSeq + groupID
|
||||||
|
seq, err := c.rdb.Incr(ctx, key).Uint64()
|
||||||
|
return int64(seq), errs.Wrap(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetGroupMaxSeq(ctx context.Context, groupID string, maxSeq int64) error {
|
||||||
|
key := NotificationGroupMaxSeq + groupID
|
||||||
|
return errs.Wrap(c.rdb.Set(ctx, key, maxSeq, 0).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetGroupMinSeq(ctx context.Context, groupID string, minSeq int64) error {
|
||||||
|
key := NotificationGroupMinSeq + groupID
|
||||||
|
return errs.Wrap(c.rdb.Set(ctx, key, minSeq, 0).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) AddTokenFlag(ctx context.Context, userID string, platformID int, token string, flag int) error {
|
||||||
|
key := NotificationUidPidToken + userID + ":" + constant.PlatformIDToName(platformID)
|
||||||
|
return errs.Wrap(c.rdb.HSet(ctx, key, token, flag).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetTokensWithoutError(ctx context.Context, userID, platformID string) (map[string]int, error) {
|
||||||
|
key := NotificationUidPidToken + userID + ":" + platformID
|
||||||
|
m, err := c.rdb.HGetAll(ctx, key).Result()
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.Wrap(err)
|
||||||
|
}
|
||||||
|
mm := make(map[string]int)
|
||||||
|
for k, v := range m {
|
||||||
|
mm[k] = utils.StringToInt(v)
|
||||||
|
}
|
||||||
|
return mm, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetTokenMapByUidPid(ctx context.Context, userID string, platform string, m map[string]int) error {
|
||||||
|
key := NotificationUidPidToken + userID + ":" + platform
|
||||||
|
mm := make(map[string]interface{})
|
||||||
|
for k, v := range m {
|
||||||
|
mm[k] = v
|
||||||
|
}
|
||||||
|
return errs.Wrap(c.rdb.HSet(ctx, key, mm).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) DeleteTokenByUidPid(ctx context.Context, userID string, platform string, fields []string) error {
|
||||||
|
key := NotificationUidPidToken + userID + ":" + platform
|
||||||
|
return errs.Wrap(c.rdb.HDel(ctx, key, fields...).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) getMessageCacheKey(sourceID string, seq int64) string {
|
||||||
|
return NotificationMessageCache + sourceID + "_" + strconv.Itoa(int(seq))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) allMessageCacheKey(sourceID string) string {
|
||||||
|
return NotificationMessageCache + sourceID + "_*"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetMessagesBySeq(ctx context.Context, userID string, seqs []int64) (seqMsgs []*sdkws.MsgData, failedSeqs []int64, err error) {
|
||||||
|
pipe := c.rdb.Pipeline()
|
||||||
|
for _, v := range seqs {
|
||||||
|
//MESSAGE_CACHE:169.254.225.224_reliability1653387820_0_1
|
||||||
|
key := c.getMessageCacheKey(userID, v)
|
||||||
|
if err := pipe.Get(ctx, key).Err(); err != nil && err != redis.Nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
result, err := pipe.Exec(ctx)
|
||||||
|
for i, v := range result {
|
||||||
|
if v.Err() != nil {
|
||||||
|
failedSeqs = append(failedSeqs, seqs[i])
|
||||||
|
} else {
|
||||||
|
msg := sdkws.MsgData{}
|
||||||
|
err = jsonpb.UnmarshalString(v.String(), &msg)
|
||||||
|
if err != nil {
|
||||||
|
failedSeqs = append(failedSeqs, seqs[i])
|
||||||
|
} else {
|
||||||
|
seqMsgs = append(seqMsgs, &msg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return seqMsgs, failedSeqs, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetMessageToCache(ctx context.Context, userID string, msgList []*sdkws.MsgData) (int, error) {
|
||||||
|
pipe := c.rdb.Pipeline()
|
||||||
|
var failedMsgs []sdkws.MsgData
|
||||||
|
for _, msg := range msgList {
|
||||||
|
key := c.getMessageCacheKey(userID, msg.Seq)
|
||||||
|
s, err := utils.Pb2String(msg)
|
||||||
|
if err != nil {
|
||||||
|
return 0, errs.Wrap(err)
|
||||||
|
}
|
||||||
|
err = pipe.Set(ctx, key, s, time.Duration(config.Config.MsgCacheTimeout)*time.Second).Err()
|
||||||
|
if err != nil {
|
||||||
|
return 0, errs.Wrap(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(failedMsgs) != 0 {
|
||||||
|
return len(failedMsgs), fmt.Errorf("set msg to notificationCache failed, failed lists: %v, %s", failedMsgs, userID)
|
||||||
|
}
|
||||||
|
_, err := pipe.Exec(ctx)
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) DeleteMessageFromCache(ctx context.Context, userID string, msgList []*sdkws.MsgData) error {
|
||||||
|
pipe := c.rdb.Pipeline()
|
||||||
|
for _, v := range msgList {
|
||||||
|
if err := pipe.Del(ctx, c.getMessageCacheKey(userID, v.Seq)).Err(); err != nil {
|
||||||
|
return errs.Wrap(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err := pipe.Exec(ctx)
|
||||||
|
return errs.Wrap(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) CleanUpOneUserAllMsg(ctx context.Context, userID string) error {
|
||||||
|
vals, err := c.rdb.Keys(ctx, c.allMessageCacheKey(userID)).Result()
|
||||||
|
if err == redis.Nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return errs.Wrap(err)
|
||||||
|
}
|
||||||
|
pipe := c.rdb.Pipeline()
|
||||||
|
for _, v := range vals {
|
||||||
|
if err := pipe.Del(ctx, v).Err(); err != nil {
|
||||||
|
return errs.Wrap(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = pipe.Exec(ctx)
|
||||||
|
return errs.Wrap(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) HandleSignalInvite(ctx context.Context, msg *sdkws.MsgData, pushToUserID string) (isSend bool, err error) {
|
||||||
|
req := &sdkws.SignalReq{}
|
||||||
|
if err := proto.Unmarshal(msg.Content, req); err != nil {
|
||||||
|
return false, errs.Wrap(err)
|
||||||
|
}
|
||||||
|
var inviteeUserIDs []string
|
||||||
|
var isInviteSignal bool
|
||||||
|
switch signalInfo := req.Payload.(type) {
|
||||||
|
case *sdkws.SignalReq_Invite:
|
||||||
|
inviteeUserIDs = signalInfo.Invite.Invitation.InviteeUserIDList
|
||||||
|
isInviteSignal = true
|
||||||
|
case *sdkws.SignalReq_InviteInGroup:
|
||||||
|
inviteeUserIDs = signalInfo.InviteInGroup.Invitation.InviteeUserIDList
|
||||||
|
isInviteSignal = true
|
||||||
|
if !utils.Contain(pushToUserID, inviteeUserIDs...) {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
case *sdkws.SignalReq_HungUp, *sdkws.SignalReq_Cancel, *sdkws.SignalReq_Reject, *sdkws.SignalReq_Accept:
|
||||||
|
return false, errs.Wrap(errors.New("signalInfo do not need offlinePush"))
|
||||||
|
default:
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
if isInviteSignal {
|
||||||
|
pipe := c.rdb.Pipeline()
|
||||||
|
for _, userID := range inviteeUserIDs {
|
||||||
|
timeout, err := strconv.Atoi(config.Config.Rtc.SignalTimeout)
|
||||||
|
if err != nil {
|
||||||
|
return false, errs.Wrap(err)
|
||||||
|
}
|
||||||
|
keys := NotificationSignalListCache + userID
|
||||||
|
err = pipe.LPush(ctx, keys, msg.ClientMsgID).Err()
|
||||||
|
if err != nil {
|
||||||
|
return false, errs.Wrap(err)
|
||||||
|
}
|
||||||
|
err = pipe.Expire(ctx, keys, time.Duration(timeout)*time.Second).Err()
|
||||||
|
if err != nil {
|
||||||
|
return false, errs.Wrap(err)
|
||||||
|
}
|
||||||
|
key := NotificationSignalCache + msg.ClientMsgID
|
||||||
|
err = pipe.Set(ctx, key, msg.Content, time.Duration(timeout)*time.Second).Err()
|
||||||
|
if err != nil {
|
||||||
|
return false, errs.Wrap(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err := pipe.Exec(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return false, errs.Wrap(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetSignalInvitationInfoByClientMsgID(ctx context.Context, clientMsgID string) (signalInviteReq *sdkws.SignalInviteReq, err error) {
|
||||||
|
bytes, err := c.rdb.Get(ctx, NotificationSignalCache+clientMsgID).Bytes()
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.Wrap(err)
|
||||||
|
}
|
||||||
|
signalReq := &sdkws.SignalReq{}
|
||||||
|
if err = proto.Unmarshal(bytes, signalReq); err != nil {
|
||||||
|
return nil, errs.Wrap(err)
|
||||||
|
}
|
||||||
|
signalInviteReq = &sdkws.SignalInviteReq{}
|
||||||
|
switch req := signalReq.Payload.(type) {
|
||||||
|
case *sdkws.SignalReq_Invite:
|
||||||
|
signalInviteReq.Invitation = req.Invite.Invitation
|
||||||
|
signalInviteReq.OpUserID = req.Invite.OpUserID
|
||||||
|
case *sdkws.SignalReq_InviteInGroup:
|
||||||
|
signalInviteReq.Invitation = req.InviteInGroup.Invitation
|
||||||
|
signalInviteReq.OpUserID = req.InviteInGroup.OpUserID
|
||||||
|
}
|
||||||
|
return signalInviteReq, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetAvailableSignalInvitationInfo(ctx context.Context, userID string) (invitationInfo *sdkws.SignalInviteReq, err error) {
|
||||||
|
key, err := c.rdb.LPop(ctx, NotificationSignalListCache+userID).Result()
|
||||||
|
if err != nil {
|
||||||
|
return nil, errs.Wrap(err)
|
||||||
|
}
|
||||||
|
invitationInfo, err = c.GetSignalInvitationInfoByClientMsgID(ctx, key)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return invitationInfo, errs.Wrap(c.DelUserSignalList(ctx, userID))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) DelUserSignalList(ctx context.Context, userID string) error {
|
||||||
|
return errs.Wrap(c.rdb.Del(ctx, NotificationSignalListCache+userID).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) DelMsgFromCache(ctx context.Context, userID string, seqs []int64) error {
|
||||||
|
for _, seq := range seqs {
|
||||||
|
key := c.getMessageCacheKey(userID, seq)
|
||||||
|
result, err := c.rdb.Get(ctx, key).Result()
|
||||||
|
if err != nil {
|
||||||
|
if err == redis.Nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
return errs.Wrap(err)
|
||||||
|
}
|
||||||
|
var msg sdkws.MsgData
|
||||||
|
if err := jsonpb.UnmarshalString(result, &msg); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
msg.Status = constant.MsgDeleted
|
||||||
|
s, err := utils.Pb2String(&msg)
|
||||||
|
if err != nil {
|
||||||
|
return errs.Wrap(err)
|
||||||
|
}
|
||||||
|
if err := c.rdb.Set(ctx, key, s, time.Duration(config.Config.MsgCacheTimeout)*time.Second).Err(); err != nil {
|
||||||
|
return errs.Wrap(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetGetuiToken(ctx context.Context, token string, expireTime int64) error {
|
||||||
|
return errs.Wrap(c.rdb.Set(ctx, NotificationGetuiToken, token, time.Duration(expireTime)*time.Second).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetGetuiToken(ctx context.Context) (string, error) {
|
||||||
|
return utils.Wrap2(c.rdb.Get(ctx, NotificationGetuiToken).Result())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetGetuiTaskID(ctx context.Context, taskID string, expireTime int64) error {
|
||||||
|
return errs.Wrap(c.rdb.Set(ctx, NotificationGetuiTaskID, taskID, time.Duration(expireTime)*time.Second).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetGetuiTaskID(ctx context.Context) (string, error) {
|
||||||
|
return utils.Wrap2(c.rdb.Get(ctx, NotificationGetuiTaskID).Result())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetSendMsgStatus(ctx context.Context, id string, status int32) error {
|
||||||
|
return errs.Wrap(c.rdb.Set(ctx, NotificationSendMsgFailedFlag+id, status, time.Hour*24).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetSendMsgStatus(ctx context.Context, id string) (int32, error) {
|
||||||
|
result, err := c.rdb.Get(ctx, NotificationSendMsgFailedFlag+id).Int()
|
||||||
|
return int32(result), errs.Wrap(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetFcmToken(ctx context.Context, account string, platformID int, fcmToken string, expireTime int64) (err error) {
|
||||||
|
return errs.Wrap(c.rdb.Set(ctx, NotificationFcmToken+account+":"+strconv.Itoa(platformID), fcmToken, time.Duration(expireTime)*time.Second).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetFcmToken(ctx context.Context, account string, platformID int) (string, error) {
|
||||||
|
return utils.Wrap2(c.rdb.Get(ctx, NotificationFcmToken+account+":"+strconv.Itoa(platformID)).Result())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) DelFcmToken(ctx context.Context, account string, platformID int) error {
|
||||||
|
return errs.Wrap(c.rdb.Del(ctx, NotificationFcmToken+account+":"+strconv.Itoa(platformID)).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) IncrUserBadgeUnreadCountSum(ctx context.Context, userID string) (int, error) {
|
||||||
|
seq, err := c.rdb.Incr(ctx, NotificationUserBadgeUnreadCountSum+userID).Result()
|
||||||
|
return int(seq), errs.Wrap(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetUserBadgeUnreadCountSum(ctx context.Context, userID string, value int) error {
|
||||||
|
return errs.Wrap(c.rdb.Set(ctx, NotificationUserBadgeUnreadCountSum+userID, value, 0).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetUserBadgeUnreadCountSum(ctx context.Context, userID string) (int, error) {
|
||||||
|
return utils.Wrap2(c.rdb.Get(ctx, NotificationUserBadgeUnreadCountSum+userID).Int())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) LockMessageTypeKey(ctx context.Context, clientMsgID string, TypeKey string) error {
|
||||||
|
key := NotificationExTypeKeyLocker + clientMsgID + "_" + TypeKey
|
||||||
|
return errs.Wrap(c.rdb.SetNX(ctx, key, 1, time.Minute).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) UnLockMessageTypeKey(ctx context.Context, clientMsgID string, TypeKey string) error {
|
||||||
|
key := NotificationExTypeKeyLocker + clientMsgID + "_" + TypeKey
|
||||||
|
return errs.Wrap(c.rdb.Del(ctx, key).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) getMessageReactionExPrefix(clientMsgID string, sessionType int32) string {
|
||||||
|
switch sessionType {
|
||||||
|
case constant.SingleChatType:
|
||||||
|
return "EX_SINGLE_" + clientMsgID
|
||||||
|
case constant.GroupChatType:
|
||||||
|
return "EX_GROUP_" + clientMsgID
|
||||||
|
case constant.SuperGroupChatType:
|
||||||
|
return "EX_SUPER_GROUP_" + clientMsgID
|
||||||
|
case constant.NotificationChatType:
|
||||||
|
return "EX_NOTIFICATION" + clientMsgID
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) JudgeMessageReactionExist(ctx context.Context, clientMsgID string, sessionType int32) (bool, error) {
|
||||||
|
n, err := c.rdb.Exists(ctx, c.getMessageReactionExPrefix(clientMsgID, sessionType)).Result()
|
||||||
|
if err != nil {
|
||||||
|
return false, utils.Wrap(err, "")
|
||||||
|
}
|
||||||
|
return n > 0, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetMessageTypeKeyValue(ctx context.Context, clientMsgID string, sessionType int32, typeKey, value string) error {
|
||||||
|
return errs.Wrap(c.rdb.HSet(ctx, c.getMessageReactionExPrefix(clientMsgID, sessionType), typeKey, value).Err())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) SetMessageReactionExpire(ctx context.Context, clientMsgID string, sessionType int32, expiration time.Duration) (bool, error) {
|
||||||
|
return utils.Wrap2(c.rdb.Expire(ctx, c.getMessageReactionExPrefix(clientMsgID, sessionType), expiration).Result())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetMessageTypeKeyValue(ctx context.Context, clientMsgID string, sessionType int32, typeKey string) (string, error) {
|
||||||
|
return utils.Wrap2(c.rdb.HGet(ctx, c.getMessageReactionExPrefix(clientMsgID, sessionType), typeKey).Result())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) GetOneMessageAllReactionList(ctx context.Context, clientMsgID string, sessionType int32) (map[string]string, error) {
|
||||||
|
return utils.Wrap2(c.rdb.HGetAll(ctx, c.getMessageReactionExPrefix(clientMsgID, sessionType)).Result())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *notificationCache) DeleteOneMessageKey(ctx context.Context, clientMsgID string, sessionType int32, subKey string) error {
|
||||||
|
return errs.Wrap(c.rdb.HDel(ctx, c.getMessageReactionExPrefix(clientMsgID, sessionType), subKey).Err())
|
||||||
|
}
|
||||||
@ -1,28 +1,133 @@
|
|||||||
package unrelation
|
package unrelation
|
||||||
|
|
||||||
import "github.com/OpenIMSDK/Open-IM-Server/pkg/proto/sdkws"
|
import (
|
||||||
|
"context"
|
||||||
type MsgModel struct {
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"
|
||||||
SendID string `bson:"send_id"`
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/sdkws"
|
||||||
RecvID string `bson:"recv_id"`
|
"strconv"
|
||||||
GroupID string `bson:"group_id"`
|
"strings"
|
||||||
ClientMsgID string `bson:"client_msg_id"` // 客户端消息ID
|
)
|
||||||
ServerMsgID string `bson:"server_msg_id"` // 服务端消息ID
|
|
||||||
SenderPlatformID int32 `bson:"sender_platform_id"`
|
const (
|
||||||
SenderNickname string `bson:"sender_nickname"`
|
singleGocMsgNum = 5000
|
||||||
SenderFaceURL string `bson:"sender_face_url"`
|
Msg = "msg"
|
||||||
SessionType int32 `bson:"session_type"`
|
OldestList = 0
|
||||||
MsgFrom int32 `bson:"msg_from"`
|
NewestList = -1
|
||||||
ContentType int32 `bson:"contentType"`
|
)
|
||||||
Content []byte `bson:"content"`
|
|
||||||
Seq int64 `bson:"seq"`
|
type MsgDocModel struct {
|
||||||
SendTime int64 `bson:"sendTime"`
|
DocID string `bson:"uid"`
|
||||||
CreateTime int64 `bson:"createTime"`
|
Msg []MsgInfoModel `bson:"msg"`
|
||||||
Status int32 `bson:"status"`
|
}
|
||||||
Options map[string]bool `bson:"options"`
|
|
||||||
OfflinePushInfo *sdkws.OfflinePushInfo `bson:"offlinePushInfo"`
|
type MsgInfoModel struct {
|
||||||
AtUserIDList []string `bson:"atUserIDList"`
|
SendTime int64 `bson:"sendtime"`
|
||||||
MsgDataList []byte `bson:"msgDataList"`
|
Msg []byte `bson:"msg"`
|
||||||
AttachedInfo string `bson:"attachedInfo"`
|
}
|
||||||
Ex string `bson:"ex"`
|
|
||||||
|
type MsgDocModelInterface interface {
|
||||||
|
PushMsgsToDoc(ctx context.Context, docID string, msgsToMongo []MsgInfoModel) error
|
||||||
|
Create(ctx context.Context, model *MsgDocModel) error
|
||||||
|
UpdateMsgStatusByIndexInOneDoc(ctx context.Context, docID string, msg *sdkws.MsgData, seqIndex int, status int32) error
|
||||||
|
FindOneByDocID(ctx context.Context, docID string) (*MsgDocModel, error)
|
||||||
|
GetNewestMsg(ctx context.Context, sourceID string) (*MsgInfoModel, error)
|
||||||
|
GetOldestMsg(ctx context.Context, sourceID string) (*MsgInfoModel, error)
|
||||||
|
Delete(ctx context.Context, docIDs []string) error
|
||||||
|
GetMsgsByIndex(ctx context.Context, sourceID string, index int64) (*MsgDocModel, error)
|
||||||
|
UpdateOneDoc(ctx context.Context, msg *MsgDocModel) error
|
||||||
|
}
|
||||||
|
|
||||||
|
func (MsgDocModel) TableName() string {
|
||||||
|
return Msg
|
||||||
|
}
|
||||||
|
|
||||||
|
func (MsgDocModel) GetSingleGocMsgNum() int64 {
|
||||||
|
return singleGocMsgNum
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MsgDocModel) IsFull() bool {
|
||||||
|
index, _ := strconv.Atoi(strings.Split(m.DocID, ":")[1])
|
||||||
|
if index == 0 {
|
||||||
|
if len(m.Msg) >= singleGocMsgNum-1 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(m.Msg) >= singleGocMsgNum {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m MsgDocModel) GetDocID(sourceID string, seq int64) string {
|
||||||
|
seqSuffix := seq / singleGocMsgNum
|
||||||
|
return m.indexGen(sourceID, seqSuffix)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m MsgDocModel) GetSeqDocIDList(userID string, maxSeq int64) []string {
|
||||||
|
seqMaxSuffix := maxSeq / singleGocMsgNum
|
||||||
|
var seqUserIDs []string
|
||||||
|
for i := 0; i <= int(seqMaxSuffix); i++ {
|
||||||
|
seqUserID := m.indexGen(userID, int64(i))
|
||||||
|
seqUserIDs = append(seqUserIDs, seqUserID)
|
||||||
|
}
|
||||||
|
return seqUserIDs
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m MsgDocModel) getSeqSuperGroupID(groupID string, seq int64) string {
|
||||||
|
seqSuffix := seq / singleGocMsgNum
|
||||||
|
return m.superGroupIndexGen(groupID, seqSuffix)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m MsgDocModel) superGroupIndexGen(groupID string, seqSuffix int64) string {
|
||||||
|
return "super_group_" + groupID + ":" + strconv.FormatInt(int64(seqSuffix), 10)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m MsgDocModel) GetDocIDSeqsMap(sourceID string, seqs []int64) map[string][]int64 {
|
||||||
|
t := make(map[string][]int64)
|
||||||
|
for i := 0; i < len(seqs); i++ {
|
||||||
|
docID := m.GetDocID(sourceID, seqs[i])
|
||||||
|
if value, ok := t[docID]; !ok {
|
||||||
|
var temp []int64
|
||||||
|
t[docID] = append(temp, seqs[i])
|
||||||
|
} else {
|
||||||
|
t[docID] = append(value, seqs[i])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return t
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m MsgDocModel) getMsgIndex(seq uint32) int {
|
||||||
|
seqSuffix := seq / singleGocMsgNum
|
||||||
|
var index uint32
|
||||||
|
if seqSuffix == 0 {
|
||||||
|
index = (seq - seqSuffix*singleGocMsgNum) - 1
|
||||||
|
} else {
|
||||||
|
index = seq - seqSuffix*singleGocMsgNum
|
||||||
|
}
|
||||||
|
return int(index)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m MsgDocModel) indexGen(sourceID string, seqSuffix int64) string {
|
||||||
|
return sourceID + ":" + strconv.FormatInt(seqSuffix, 10)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (MsgDocModel) GenExceptionMessageBySeqs(seqs []int64) (exceptionMsg []*sdkws.MsgData) {
|
||||||
|
for _, v := range seqs {
|
||||||
|
msg := new(sdkws.MsgData)
|
||||||
|
msg.Seq = v
|
||||||
|
exceptionMsg = append(exceptionMsg, msg)
|
||||||
|
}
|
||||||
|
return exceptionMsg
|
||||||
|
}
|
||||||
|
|
||||||
|
func (MsgDocModel) GenExceptionSuperGroupMessageBySeqs(seqs []int64, groupID string) (exceptionMsg []*sdkws.MsgData) {
|
||||||
|
for _, v := range seqs {
|
||||||
|
msg := new(sdkws.MsgData)
|
||||||
|
msg.Seq = v
|
||||||
|
msg.GroupID = groupID
|
||||||
|
msg.SessionType = constant.SuperGroupChatType
|
||||||
|
exceptionMsg = append(exceptionMsg, msg)
|
||||||
|
}
|
||||||
|
return exceptionMsg
|
||||||
}
|
}
|
||||||
|
|||||||
@ -0,0 +1,134 @@
|
|||||||
|
package unrelation
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
table "github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/table/unrelation"
|
||||||
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/sdkws"
|
||||||
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
|
||||||
|
"go.mongodb.org/mongo-driver/bson"
|
||||||
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||||
|
"go.mongodb.org/mongo-driver/mongo"
|
||||||
|
"go.mongodb.org/mongo-driver/mongo/options"
|
||||||
|
"google.golang.org/protobuf/proto"
|
||||||
|
)
|
||||||
|
|
||||||
|
var ErrNotificationListNotExist = errors.New("user not have msg in mongoDB")
|
||||||
|
var ErrNotificationNotFound = errors.New("msg not found")
|
||||||
|
|
||||||
|
type NotificationMongoDriver struct {
|
||||||
|
MsgCollection *mongo.Collection
|
||||||
|
msg table.NotificationDocModel
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewNotificationMongoDriver(database *mongo.Database) table.NotificationDocModelInterface {
|
||||||
|
return &NotificationMongoDriver{MsgCollection: database.Collection(table.NotificationDocModel{}.TableName())}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *NotificationMongoDriver) PushMsgsToDoc(ctx context.Context, docID string, msgsToMongo []table.NotificationInfoModel) error {
|
||||||
|
filter := bson.M{"uid": docID}
|
||||||
|
return m.MsgCollection.FindOneAndUpdate(ctx, filter, bson.M{"$push": bson.M{"msg": bson.M{"$each": msgsToMongo}}}).Err()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *NotificationMongoDriver) Create(ctx context.Context, model *table.NotificationDocModel) error {
|
||||||
|
_, err := m.MsgCollection.InsertOne(ctx, model)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *NotificationMongoDriver) UpdateMsgStatusByIndexInOneDoc(ctx context.Context, docID string, msg *sdkws.MsgData, seqIndex int, status int32) error {
|
||||||
|
msg.Status = status
|
||||||
|
bytes, err := proto.Marshal(msg)
|
||||||
|
if err != nil {
|
||||||
|
return utils.Wrap(err, "")
|
||||||
|
}
|
||||||
|
_, err = m.MsgCollection.UpdateOne(ctx, bson.M{"uid": docID}, bson.M{"$set": bson.M{fmt.Sprintf("msg.%d.msg", seqIndex): bytes}})
|
||||||
|
if err != nil {
|
||||||
|
return utils.Wrap(err, "")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *NotificationMongoDriver) FindOneByDocID(ctx context.Context, docID string) (*table.NotificationDocModel, error) {
|
||||||
|
doc := &table.NotificationDocModel{}
|
||||||
|
err := m.MsgCollection.FindOne(ctx, bson.M{"uid": docID}).Decode(doc)
|
||||||
|
return doc, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *NotificationMongoDriver) GetMsgsByIndex(ctx context.Context, sourceID string, index int64) (*table.NotificationDocModel, error) {
|
||||||
|
findOpts := options.Find().SetLimit(1).SetSkip(index).SetSort(bson.M{"uid": 1})
|
||||||
|
cursor, err := m.MsgCollection.Find(ctx, bson.M{"uid": primitive.Regex{Pattern: fmt.Sprintf("^%s:", sourceID)}}, findOpts)
|
||||||
|
if err != nil {
|
||||||
|
return nil, utils.Wrap(err, "")
|
||||||
|
}
|
||||||
|
var msgs []table.NotificationDocModel
|
||||||
|
err = cursor.All(context.Background(), &msgs)
|
||||||
|
if err != nil {
|
||||||
|
return nil, utils.Wrap(err, fmt.Sprintf("cursor is %s", cursor.Current.String()))
|
||||||
|
}
|
||||||
|
if len(msgs) > 0 {
|
||||||
|
return &msgs[0], nil
|
||||||
|
}
|
||||||
|
return nil, ErrMsgListNotExist
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *NotificationMongoDriver) GetNewestMsg(ctx context.Context, sourceID string) (*table.NotificationInfoModel, error) {
|
||||||
|
var msgDocs []table.NotificationDocModel
|
||||||
|
cursor, err := m.MsgCollection.Find(ctx, bson.M{"uid": bson.M{"$regex": fmt.Sprintf("^%s:", sourceID)}}, options.Find().SetLimit(1).SetSort(bson.M{"uid": -1}))
|
||||||
|
if err != nil {
|
||||||
|
return nil, utils.Wrap(err, "")
|
||||||
|
}
|
||||||
|
err = cursor.All(ctx, &msgDocs)
|
||||||
|
if err != nil {
|
||||||
|
return nil, utils.Wrap(err, "")
|
||||||
|
}
|
||||||
|
if len(msgDocs) > 0 {
|
||||||
|
if len(msgDocs[0].Msg) > 0 {
|
||||||
|
return &msgDocs[0].Msg[len(msgDocs[0].Msg)-1], nil
|
||||||
|
}
|
||||||
|
return nil, errors.New("len(msgDocs[0].Msg) < 0")
|
||||||
|
}
|
||||||
|
return nil, ErrMsgNotFound
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *NotificationMongoDriver) GetOldestMsg(ctx context.Context, sourceID string) (*table.NotificationInfoModel, error) {
|
||||||
|
var msgDocs []table.NotificationDocModel
|
||||||
|
cursor, err := m.MsgCollection.Find(ctx, bson.M{"uid": bson.M{"$regex": fmt.Sprintf("^%s:", sourceID)}}, options.Find().SetLimit(1).SetSort(bson.M{"uid": 1}))
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
err = cursor.All(ctx, &msgDocs)
|
||||||
|
if err != nil {
|
||||||
|
return nil, utils.Wrap(err, "")
|
||||||
|
}
|
||||||
|
var oldestMsg table.NotificationInfoModel
|
||||||
|
if len(msgDocs) > 0 {
|
||||||
|
for _, v := range msgDocs[0].Msg {
|
||||||
|
if v.SendTime != 0 {
|
||||||
|
oldestMsg = v
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(oldestMsg.Msg) == 0 {
|
||||||
|
if len(msgDocs[0].Msg) > 0 {
|
||||||
|
oldestMsg = msgDocs[0].Msg[0]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return &oldestMsg, nil
|
||||||
|
}
|
||||||
|
return nil, ErrMsgNotFound
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *NotificationMongoDriver) Delete(ctx context.Context, docIDs []string) error {
|
||||||
|
if docIDs == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
_, err := m.MsgCollection.DeleteMany(ctx, bson.M{"uid": bson.M{"$in": docIDs}})
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *NotificationMongoDriver) UpdateOneDoc(ctx context.Context, msg *table.NotificationDocModel) error {
|
||||||
|
_, err := m.MsgCollection.UpdateOne(ctx, bson.M{"uid": msg.DocID}, bson.M{"$set": bson.M{"msg": msg.Msg}})
|
||||||
|
return err
|
||||||
|
}
|
||||||
Loading…
Reference in new issue