You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
265 lines
9.8 KiB
265 lines
9.8 KiB
// Copyright © 2023 OpenIM. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package msg
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
cbapi "github.com/openimsdk/open-im-server/v3/pkg/callbackstruct"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/storage/model"
|
|
"github.com/openimsdk/protocol/constant"
|
|
"github.com/openimsdk/protocol/conversation"
|
|
"github.com/openimsdk/protocol/msg"
|
|
"github.com/openimsdk/protocol/sdkws"
|
|
"github.com/openimsdk/tools/errs"
|
|
"github.com/openimsdk/tools/log"
|
|
"github.com/openimsdk/tools/utils/datautil"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
func (m *msgServer) GetConversationsHasReadAndMaxSeq(ctx context.Context, req *msg.GetConversationsHasReadAndMaxSeqReq) (*msg.GetConversationsHasReadAndMaxSeqResp, error) {
|
|
var conversationIDs []string
|
|
if len(req.ConversationIDs) == 0 {
|
|
var err error
|
|
conversationIDs, err = m.ConversationLocalCache.GetConversationIDs(ctx, req.UserID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
} else {
|
|
conversationIDs = req.ConversationIDs
|
|
}
|
|
|
|
hasReadSeqs, err := m.MsgDatabase.GetHasReadSeqs(ctx, req.UserID, conversationIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
conversations, err := m.ConversationLocalCache.GetConversations(ctx, req.UserID, conversationIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
conversationMaxSeqMap := make(map[string]int64)
|
|
for _, conversation := range conversations {
|
|
if conversation.MaxSeq != 0 {
|
|
conversationMaxSeqMap[conversation.ConversationID] = conversation.MaxSeq
|
|
}
|
|
}
|
|
maxSeqs, err := m.MsgDatabase.GetMaxSeqsWithTime(ctx, conversationIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
resp := &msg.GetConversationsHasReadAndMaxSeqResp{Seqs: make(map[string]*msg.Seqs)}
|
|
for conversationID, maxSeq := range maxSeqs {
|
|
resp.Seqs[conversationID] = &msg.Seqs{
|
|
HasReadSeq: hasReadSeqs[conversationID],
|
|
MaxSeq: maxSeq.Seq,
|
|
MaxSeqTime: maxSeq.Time,
|
|
}
|
|
if v, ok := conversationMaxSeqMap[conversationID]; ok {
|
|
resp.Seqs[conversationID].MaxSeq = v
|
|
}
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
func (m *msgServer) SetConversationHasReadSeq(ctx context.Context, req *msg.SetConversationHasReadSeqReq) (*msg.SetConversationHasReadSeqResp, error) {
|
|
maxSeq, err := m.MsgDatabase.GetMaxSeq(ctx, req.ConversationID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if req.HasReadSeq > maxSeq {
|
|
return nil, errs.ErrArgs.WrapMsg("hasReadSeq must not be bigger than maxSeq")
|
|
}
|
|
if err := m.MsgDatabase.SetHasReadSeq(ctx, req.UserID, req.ConversationID, req.HasReadSeq); err != nil {
|
|
return nil, err
|
|
}
|
|
m.sendMarkAsReadNotification(ctx, req.ConversationID, constant.SingleChatType, req.UserID, req.UserID, nil, req.HasReadSeq)
|
|
return &msg.SetConversationHasReadSeqResp{}, nil
|
|
}
|
|
|
|
func (m *msgServer) MarkMsgsAsRead(ctx context.Context, req *msg.MarkMsgsAsReadReq) (*msg.MarkMsgsAsReadResp, error) {
|
|
if len(req.Seqs) < 1 {
|
|
return nil, errs.ErrArgs.WrapMsg("seqs must not be empty")
|
|
}
|
|
maxSeq, err := m.MsgDatabase.GetMaxSeq(ctx, req.ConversationID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
hasReadSeq := req.Seqs[len(req.Seqs)-1]
|
|
if hasReadSeq > maxSeq {
|
|
return nil, errs.ErrArgs.WrapMsg("hasReadSeq must not be bigger than maxSeq")
|
|
}
|
|
conversation, err := m.ConversationLocalCache.GetConversation(ctx, req.UserID, req.ConversationID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := m.MsgDatabase.MarkSingleChatMsgsAsRead(ctx, req.UserID, req.ConversationID, req.Seqs); err != nil {
|
|
return nil, err
|
|
}
|
|
currentHasReadSeq, err := m.MsgDatabase.GetHasReadSeq(ctx, req.UserID, req.ConversationID)
|
|
if err != nil && !errors.Is(err, redis.Nil) {
|
|
return nil, err
|
|
}
|
|
if hasReadSeq > currentHasReadSeq {
|
|
err = m.MsgDatabase.SetHasReadSeq(ctx, req.UserID, req.ConversationID, hasReadSeq)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
reqCallback := &cbapi.CallbackSingleMsgReadReq{
|
|
ConversationID: conversation.ConversationID,
|
|
UserID: req.UserID,
|
|
Seqs: req.Seqs,
|
|
ContentType: conversation.ConversationType,
|
|
}
|
|
m.webhookAfterSingleMsgRead(ctx, &m.config.WebhooksConfig.AfterSingleMsgRead, reqCallback)
|
|
m.recordBurnDeadlines(ctx, conversation, req.UserID, req.Seqs)
|
|
m.sendMarkAsReadNotification(ctx, req.ConversationID, conversation.ConversationType, req.UserID,
|
|
m.conversationAndGetRecvID(conversation, req.UserID), req.Seqs, hasReadSeq)
|
|
return &msg.MarkMsgsAsReadResp{}, nil
|
|
}
|
|
|
|
func (m *msgServer) MarkConversationAsRead(ctx context.Context, req *msg.MarkConversationAsReadReq) (*msg.MarkConversationAsReadResp, error) {
|
|
conversation, err := m.ConversationLocalCache.GetConversation(ctx, req.UserID, req.ConversationID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
hasReadSeq, err := m.MsgDatabase.GetHasReadSeq(ctx, req.UserID, req.ConversationID)
|
|
if err != nil && !errors.Is(err, redis.Nil) {
|
|
return nil, err
|
|
}
|
|
var seqs []int64
|
|
|
|
log.ZDebug(ctx, "MarkConversationAsRead", "hasReadSeq", hasReadSeq, "req.HasReadSeq", req.HasReadSeq)
|
|
if conversation.ConversationType == constant.SingleChatType {
|
|
for i := hasReadSeq + 1; i <= req.HasReadSeq; i++ {
|
|
seqs = append(seqs, i)
|
|
}
|
|
// avoid client missed call MarkConversationMessageAsRead by order
|
|
for _, val := range req.Seqs {
|
|
if !datautil.Contain(val, seqs...) {
|
|
seqs = append(seqs, val)
|
|
}
|
|
}
|
|
if len(seqs) > 0 {
|
|
log.ZDebug(ctx, "MarkConversationAsRead", "seqs", seqs, "conversationID", req.ConversationID)
|
|
if err = m.MsgDatabase.MarkSingleChatMsgsAsRead(ctx, req.UserID, req.ConversationID, seqs); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
if req.HasReadSeq > hasReadSeq {
|
|
err = m.MsgDatabase.SetHasReadSeq(ctx, req.UserID, req.ConversationID, req.HasReadSeq)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
hasReadSeq = req.HasReadSeq
|
|
}
|
|
m.recordBurnDeadlines(ctx, conversation, req.UserID, seqs)
|
|
m.sendMarkAsReadNotification(ctx, req.ConversationID, conversation.ConversationType, req.UserID,
|
|
m.conversationAndGetRecvID(conversation, req.UserID), seqs, hasReadSeq)
|
|
} else if conversation.ConversationType == constant.ReadGroupChatType ||
|
|
conversation.ConversationType == constant.NotificationChatType {
|
|
if req.HasReadSeq > hasReadSeq {
|
|
err = m.MsgDatabase.SetHasReadSeq(ctx, req.UserID, req.ConversationID, req.HasReadSeq)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
hasReadSeq = req.HasReadSeq
|
|
}
|
|
m.sendMarkAsReadNotification(ctx, req.ConversationID, constant.SingleChatType, req.UserID,
|
|
req.UserID, seqs, hasReadSeq)
|
|
}
|
|
|
|
if conversation.ConversationType == constant.SingleChatType {
|
|
reqCall := &cbapi.CallbackSingleMsgReadReq{
|
|
ConversationID: conversation.ConversationID,
|
|
UserID: conversation.OwnerUserID,
|
|
Seqs: req.Seqs,
|
|
ContentType: conversation.ConversationType,
|
|
}
|
|
m.webhookAfterSingleMsgRead(ctx, &m.config.WebhooksConfig.AfterSingleMsgRead, reqCall)
|
|
} else if conversation.ConversationType == constant.ReadGroupChatType {
|
|
reqCall := &cbapi.CallbackGroupMsgReadReq{
|
|
SendID: conversation.OwnerUserID,
|
|
ReceiveID: req.UserID,
|
|
UnreadMsgNum: req.HasReadSeq,
|
|
ContentType: int64(conversation.ConversationType),
|
|
}
|
|
m.webhookAfterGroupMsgRead(ctx, &m.config.WebhooksConfig.AfterGroupMsgRead, reqCall)
|
|
}
|
|
return &msg.MarkConversationAsReadResp{}, nil
|
|
}
|
|
|
|
// recordBurnDeadlines 在「单聊」场景下，根据对端的 MsgBurnDuration 为本次已读的每条消息
|
|
// 记录一份「阅后即焚」截止时间到 mongo。后续由 cron 推进 min_seq 让消息从该用户视图消失。
|
|
//
|
|
// 设计要点：
|
|
// 1. 仅单聊：群聊有自己的不同语义（多接收方 + 仅 watermark 已读），暂不在此实现。
|
|
// 2. 仅当对端 MsgBurnDuration > 0 时才记录；为 0 表示对端未开启该功能。
|
|
// 3. 同一 (UserID, ConversationID, Seq) 已存在则不覆盖：保证以「首次阅读时刻」为基准，
|
|
// 避免多端重复 MarkAsRead 导致 deadline 被往后推。
|
|
// 4. 失败仅记录日志，不影响主流程的已读语义。
|
|
func (m *msgServer) recordBurnDeadlines(ctx context.Context, conv *conversation.Conversation, readerUserID string, seqs []int64) {
|
|
if len(seqs) == 0 {
|
|
return
|
|
}
|
|
if conv.ConversationType != constant.SingleChatType {
|
|
return
|
|
}
|
|
peerID := m.conversationAndGetRecvID(conv, readerUserID)
|
|
if peerID == "" || peerID == readerUserID {
|
|
return
|
|
}
|
|
peerInfo, err := m.UserLocalCache.GetUserInfo(ctx, peerID)
|
|
if err != nil {
|
|
log.ZWarn(ctx, "recordBurnDeadlines GetUserInfo failed", err, "peerID", peerID)
|
|
return
|
|
}
|
|
if peerInfo == nil || peerInfo.MsgBurnDuration <= 0 {
|
|
return
|
|
}
|
|
now := time.Now().UnixMilli()
|
|
deadline := now + int64(peerInfo.MsgBurnDuration)*1000
|
|
items := make([]*model.MsgBurnDeadline, 0, len(seqs))
|
|
for _, seq := range seqs {
|
|
items = append(items, &model.MsgBurnDeadline{
|
|
UserID: readerUserID,
|
|
ConversationID: conv.ConversationID,
|
|
Seq: seq,
|
|
DeadlineMs: deadline,
|
|
CreateTime: now,
|
|
})
|
|
}
|
|
if err := m.msgBurnDeadlineDB.UpsertIfAbsent(ctx, items); err != nil {
|
|
log.ZError(ctx, "recordBurnDeadlines UpsertIfAbsent failed", err,
|
|
"userID", readerUserID, "conversationID", conv.ConversationID, "seqs", seqs)
|
|
}
|
|
}
|
|
|
|
func (m *msgServer) sendMarkAsReadNotification(ctx context.Context, conversationID string, sessionType int32, sendID, recvID string, seqs []int64, hasReadSeq int64) {
|
|
tips := &sdkws.MarkAsReadTips{
|
|
MarkAsReadUserID: sendID,
|
|
ConversationID: conversationID,
|
|
Seqs: seqs,
|
|
HasReadSeq: hasReadSeq,
|
|
}
|
|
m.notificationSender.NotificationWithSessionType(ctx, sendID, recvID, constant.HasReadReceipt, sessionType, tips)
|
|
|
|
}
|