From b3c3dadadbc3f54330a5f064f10a88d718c1bd3d Mon Sep 17 00:00:00 2001
From: wangchuxiao <wangchuxiao97@outlook.com>
Date: Mon, 7 Aug 2023 19:55:09 +0800
Subject: [PATCH] feat: msg doc len changed

---
 internal/tools/msg_doc_convert.go       |  3 +++
 pkg/common/db/unrelation/msg_convert.go |  1 +
 pkg/msgprocessor/conversation.go        | 10 ++++++++++
 3 files changed, 14 insertions(+)

diff --git a/internal/tools/msg_doc_convert.go b/internal/tools/msg_doc_convert.go
index 5b684508b..4880415aa 100644
--- a/internal/tools/msg_doc_convert.go
+++ b/internal/tools/msg_doc_convert.go
@@ -14,6 +14,9 @@ func (c *MsgTool) ConvertTools() {
 		log.ZError(ctx, "get all conversation ids failed", err)
 		return
 	}
+	for _, conversationID := range conversationIDs {
+		conversationIDs = append(conversationIDs, msgprocessor.GetNotificationConversationIDByConversationID(conversationID))
+	}
 	userIDs, err := c.userDatabase.GetAllUserID(ctx, 0, 0)
 	if err != nil {
 		log.ZError(ctx, "get all user ids failed", err)
diff --git a/pkg/common/db/unrelation/msg_convert.go b/pkg/common/db/unrelation/msg_convert.go
index 362bfd6fc..c8e932522 100644
--- a/pkg/common/db/unrelation/msg_convert.go
+++ b/pkg/common/db/unrelation/msg_convert.go
@@ -49,6 +49,7 @@ func (m *MsgMongoDriver) ConvertMsgsDocLen(ctx context.Context, conversationIDs
 					} else {
 						msgDocModel.Msg = msgDoc.Msg[index:end]
 					}
+					newMsgDocs = append(newMsgDocs, msgDocModel)
 					index = end
 				} else {
 					break
diff --git a/pkg/msgprocessor/conversation.go b/pkg/msgprocessor/conversation.go
index a63b6391a..34d30fcdf 100644
--- a/pkg/msgprocessor/conversation.go
+++ b/pkg/msgprocessor/conversation.go
@@ -100,6 +100,16 @@ func GetConversationIDBySessionType(sessionType int, ids ...string) string {
 	return ""
 }
 
+func GetNotificationConversationIDByConversationID(conversationID string) string {
+	l := strings.Split(conversationID, "_")
+	if len(l) > 1 {
+		l[0] = "n"
+		return strings.Join(l, "_")
+	} else {
+		return ""
+	}
+}
+
 func GetNotificationConversationID(sessionType int, ids ...string) string {
 	sort.Strings(ids)
 	if len(ids) > 2 || len(ids) < 1 {