mark msg as read server

pkg/common/db/controller/msg.go

@@ -299,8 +299,18 @@ func (db *commonMsgDatabase) RevokeMsg(ctx context.Context, conversationID strin
 	return db.BatchInsertBlock(ctx, conversationID, []any{revoke}, updateKeyRevoke, seq)
 }
 
-func (db *commonMsgDatabase) MarkSingleChatMsgsAsRead(ctx context.Context, userID string, conversationID string, seqs []int64) error {
-	return db.msgDocDatabase.MarkSingleChatMsgsAsRead(ctx, userID, conversationID, seqs)
+func (db *commonMsgDatabase) MarkSingleChatMsgsAsRead(ctx context.Context, userID string, conversationID string, totalSeqs []int64) error {
+	for docID, seqs := range db.msg.GetDocIDSeqsMap(conversationID, totalSeqs) {
+		var indexes []int64
+		for _, seq := range seqs {
+			indexes = append(indexes, db.msg.GetMsgIndex(seq))
+		}
+		if err := db.msgDocDatabase.MarkSingleChatMsgsAsRead(ctx, userID, docID, indexes); err != nil {
+			log.ZError(ctx, "MarkSingleChatMsgsAsRead", err, "userID", userID, "docID", docID, "indexes", indexes)
+			return err
+		}
+	}
+	return nil
 }
 
 func (db *commonMsgDatabase) DeleteMessageFromCache(ctx context.Context, conversationID string, msgs []*sdkws.MsgData) error {

pkg/common/db/table/unrelation/msg.go

@@ -51,6 +51,7 @@ type MsgDataModel struct {
 	SendTime         int64             `bson:"send_time"`
 	CreateTime       int64             `bson:"create_time"`
 	Status           int32             `bson:"status"`
+	IsRead           bool              `bson:"is_read"`
 	Options          map[string]bool   `bson:"options"`
 	OfflinePush      *OfflinePushModel `bson:"offline_push"`
 	AtUserIDList     []string          `bson:"at_user_id_list"`
@@ -62,6 +63,7 @@ type MsgInfoModel struct {
 	Msg     *MsgDataModel `bson:"msg"`
 	Revoke  *RevokeModel  `bson:"revoke"`
 	DelList []string      `bson:"del_list"`
+	IsRead  bool          `bson:"is_read"`
 }
 
 type MsgDocModelInterface interface {
@@ -78,7 +80,7 @@ type MsgDocModelInterface interface {
 	DeleteDocs(ctx context.Context, docIDs []string) error
 	GetMsgDocModelByIndex(ctx context.Context, conversationID string, index, sort int64) (*MsgDocModel, error)
 	DeleteMsgsInOneDocByIndex(ctx context.Context, docID string, indexes []int) error
-	MarkSingleChatMsgsAsRead(ctx context.Context, userID string, conversationID string, seqs []int64) error
+	MarkSingleChatMsgsAsRead(ctx context.Context, userID string, docID string, indexes []int64) error
 }
 
 func (MsgDocModel) TableName() string {

pkg/common/db/unrelation/msg.go

@@ -280,11 +280,17 @@ func (m *MsgMongoDriver) IsExistDocID(ctx context.Context, docID string) (bool,
 	return count > 0, nil
 }
 
-func (m *MsgMongoDriver) MarkSingleChatMsgsAsRead(ctx context.Context, userID string, conversationID string, seqs []int64) error {
-	indexs := make([]int64, 0, len(seqs))
-	for _, seq := range seqs {
-		indexs = append(indexs, m.model.GetMsgIndex(seq))
+func (m *MsgMongoDriver) MarkSingleChatMsgsAsRead(ctx context.Context, userID string, docID string, indexes []int64) error {
+	updates := bson.M{
+		"$set": bson.M{},
+	}
+	for _, index := range indexes {
+		updates["$set"].(bson.M)[fmt.Sprintf("msgs.%d.is_read", index)] = true
+	}
+	filter := bson.M{"doc_id": docID}
+	_, err := m.MsgCollection.UpdateMany(ctx, filter, updates)
+	if err != nil {
+		return utils.Wrap(err, "")
 	}
-
 	return nil
 }