mirror of
https://github.com/openimsdk/open-im-server.git
synced 2026-06-25 20:00:27 +08:00
279 lines
10 KiB
Go
279 lines
10 KiB
Go
// 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
|
||
// 为本次已读的每条消息同时给接收者和发送者各记录一份「阅后即焚」截止时间。
|
||
// cron 到期后会分别推进两人各自的 min_seq，双方都看不到该消息。
|
||
//
|
||
// 设计要点：
|
||
// 1. 仅单聊。
|
||
// 2. 仅当发送者 MsgBurnDuration > 0 时才记录；0 表示未开启。
|
||
// 3. $setOnInsert 确保同一 (UserID, ConversationID, Seq) 已存在时不覆盖，
|
||
// 以「首次阅读时刻」为 deadline 基准，多端重复 MarkAsRead 不会往后推。
|
||
// 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
|
||
// 每条消息同时为接收者和发送者各写一条 deadline，双方消息同步焚毁。
|
||
items := make([]*model.MsgBurnDeadline, 0, len(seqs)*2)
|
||
for _, seq := range seqs {
|
||
items = append(items,
|
||
&model.MsgBurnDeadline{
|
||
UserID: readerUserID,
|
||
ConversationID: conv.ConversationID,
|
||
Seq: seq,
|
||
PeerID: peerID,
|
||
DeadlineMs: deadline,
|
||
CreateTime: now,
|
||
},
|
||
&model.MsgBurnDeadline{
|
||
UserID: peerID,
|
||
ConversationID: conv.ConversationID,
|
||
Seq: seq,
|
||
PeerID: readerUserID,
|
||
DeadlineMs: deadline,
|
||
CreateTime: now,
|
||
},
|
||
)
|
||
}
|
||
if err := m.msgBurnDeadlineDB.UpsertIfAbsent(ctx, items); err != nil {
|
||
log.ZError(ctx, "recordBurnDeadlines UpsertIfAbsent failed", err,
|
||
"readerUserID", readerUserID, "peerID", peerID,
|
||
"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)
|
||
|
||
}
|