RevokeMsg

internal/msgtransfer/online_msg_to_mongo_handler.go

@@ -2,6 +2,7 @@ package msgtransfer
 
 import (
 	"context"
+	"encoding/json"
 
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/sdkws"
 
@@ -52,7 +53,8 @@ func (mc *OnlineHistoryMongoConsumerHandler) handleChatWs2Mongo(ctx context.Cont
 		log.ZError(ctx, "remove cache msg from redis err", err, "msg", msgFromMQ.MsgData, "conversationID", msgFromMQ.ConversationID)
 	}
 	for _, v := range msgFromMQ.MsgData {
-		if v.ContentType == constant.DeleteMessageNotification {
+		switch v.ContentType {
+		case constant.DeleteMessageNotification:
 			deleteMessageTips := sdkws.DeleteMessageTips{}
 			err := proto.Unmarshal(v.Content, &deleteMessageTips)
 			if err != nil {
@@ -61,10 +63,31 @@ func (mc *OnlineHistoryMongoConsumerHandler) handleChatWs2Mongo(ctx context.Cont
 			}
 			if totalUnExistSeqs, err := mc.msgDatabase.DelMsgBySeqs(ctx, deleteMessageTips.UserID, deleteMessageTips.Seqs); err != nil {
 				log.ZError(ctx, "DelMsgBySeqs", err, "userIDs", deleteMessageTips.UserID, "seqs", deleteMessageTips.Seqs, "totalUnExistSeqs", totalUnExistSeqs)
+				continue
+			}
+		case constant.MsgRevokeNotification:
+			var elem sdkws.NotificationElem
+			if err := json.Unmarshal(v.Content, &elem); err != nil {
+				log.ZError(ctx, "json.Unmarshal NotificationElem", err, "content", string(v.Content))
+				continue
+			}
+			var tips sdkws.RevokeMsgTips
+			if err := json.Unmarshal([]byte(elem.Detail), &tips); err != nil {
+				log.ZError(ctx, "json.Unmarshal RevokeMsgTips", err, "content", string(v.Content))
+				continue
+			}
+			v.Seq = tips.Seq
+			data, err := proto.Marshal(v)
+			if err != nil {
+				log.ZError(ctx, "proto.Marshal MsgData", err)
+				continue
+			}
+			if err := mc.msgDatabase.RevokeMsg(ctx, tips.ConversationID, tips.Seq, data); err != nil {
+				log.ZError(ctx, "RevokeMsg", err, "conversationID", tips.ConversationID, "seq", tips.Seq)
+				continue
 			}
 		}
 	}
-
 }
 
 func (OnlineHistoryMongoConsumerHandler) Setup(_ sarama.ConsumerGroupSession) error   { return nil }

internal/rpc/msg/revoke.go

@@ -2,19 +2,77 @@ package msg
 
 import (
 	"context"
+	"encoding/json"
+	"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/tokenverify"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/msg"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/sdkws"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
 )
 
 func (m *msgServer) RevokeMsg(ctx context.Context, req *msg.RevokeMsgReq) (*msg.RevokeMsgResp, error) {
-	//msgs := []*sdkws.MsgData{}
-	//if err := m.MsgDatabase.MsgToMongoMQ(ctx, "", req.ConversationID, msgs, 0); err != nil {
-	//	return nil, err
-	//}
-	//msg := sdkws.MsgData{
-	//	SendID: "",
-	//}
-	//
-	//m.SendMsg(ctx, &msg.SendMsgReq{})
-
+	if req.UserID == "" {
+		return nil, errs.ErrArgs.Wrap("user_id is empty")
+	}
+	if req.RecvID == "" && req.GroupID == "" {
+		return nil, errs.ErrArgs.Wrap("recv_id and group_id are empty")
+	}
+	if req.RecvID != "" && req.GroupID != "" {
+		return nil, errs.ErrArgs.Wrap("recv_id and group_id cannot exist at the same time")
+	}
+	if err := tokenverify.CheckAccessV3(ctx, req.RecvID); err != nil {
+		return nil, err
+	}
+	var sessionType int32
+	var conversationID string
+	if req.GroupID == "" {
+		sessionType = constant.SingleChatType
+		conversationID = utils.GenConversationUniqueKeyForSingle(req.UserID, req.RecvID)
+	} else {
+		sessionType = constant.SuperGroupChatType
+		conversationID = utils.GenConversationUniqueKeyForGroup(req.GroupID)
+	}
+	tips := sdkws.RevokeMsgTips{
+		RevokerUserID:  req.UserID,
+		ClientMsgID:    "",
+		RevokeTime:     utils.GetCurrentTimestampByMill(),
+		Seq:            req.Seq,
+		SesstionType:   sessionType,
+		ConversationID: conversationID,
+	}
+	detail, err := json.Marshal(&tips)
+	if err != nil {
+		return nil, err
+	}
+	notificationElem := sdkws.NotificationElem{Detail: string(detail)}
+	content, err := json.Marshal(&notificationElem)
+	if err != nil {
+		return nil, errs.Wrap(err)
+	}
+	msgData := sdkws.MsgData{
+		SendID:      req.UserID,
+		RecvID:      req.RecvID,
+		GroupID:     req.GroupID,
+		Content:     content,
+		MsgFrom:     constant.SysMsgType,
+		ContentType: constant.MsgRevokeNotification,
+		SessionType: sessionType,
+		CreateTime:  utils.GetCurrentTimestampByMill(),
+		ClientMsgID: utils.GetMsgID(req.UserID),
+		Options: config.GetOptionsByNotification(config.NotificationConf{
+			IsSendMsg:        true,
+			ReliabilityLevel: 2,
+		}),
+		OfflinePushInfo: nil,
+	}
+	if msgData.SessionType == constant.SuperGroupChatType {
+		msgData.GroupID = msgData.RecvID
+	}
+	_, err = m.SendMsg(ctx, &msg.SendMsgReq{MsgData: &msgData})
+	if err != nil {
+		return nil, err
+	}
 	return &msg.RevokeMsgResp{}, nil
 }