Open-IM-Server/pkg/common/db/table/unrelation/extend_msg_set.go

package unrelation

import (
	"context"
	"strconv"
	"strings"

	"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/sdkws"
)

const (
	CExtendMsgSet = "extend_msgs"

	ExtendMsgMaxNum = 100
)

type ExtendMsgSetModel struct {
	ConversationID   string                    `bson:"source_id"           json:"conversationID"`
	SessionType      int32                     `bson:"session_type"        json:"sessionType"`
	ExtendMsgs       map[string]ExtendMsgModel `bson:"extend_msgs"         json:"extendMsgs"`
	ExtendMsgNum     int32                     `bson:"extend_msg_num"      json:"extendMsgNum"`
	CreateTime       int64                     `bson:"create_time"         json:"createTime"`       // this block's create time
	MaxMsgUpdateTime int64                     `bson:"max_msg_update_time" json:"maxMsgUpdateTime"` // index find msg
}

type KeyValueModel struct {
	TypeKey          string `bson:"type_key"           json:"typeKey"`
	Value            string `bson:"value"              json:"value"`
	LatestUpdateTime int64  `bson:"latest_update_time" json:"latestUpdateTime"`
}

type ExtendMsgModel struct {
	ReactionExtensionList map[string]KeyValueModel `bson:"reaction_extension_list" json:"reactionExtensionList"`
	ClientMsgID           string                   `bson:"client_msg_id"           json:"clientMsgID"`
	MsgFirstModifyTime    int64                    `bson:"msg_first_modify_time"   json:"msgFirstModifyTime"` // this extendMsg create time
	AttachedInfo          string                   `bson:"attached_info"           json:"attachedInfo"`
	Ex                    string                   `bson:"ex"                      json:"ex"`
}

type ExtendMsgSetModelInterface interface {
	CreateExtendMsgSet(ctx context.Context, set *ExtendMsgSetModel) error
	GetAllExtendMsgSet(
		ctx context.Context,
		conversationID string,
		opts *GetAllExtendMsgSetOpts,
	) (sets []*ExtendMsgSetModel, err error)
	GetExtendMsgSet(
		ctx context.Context,
		conversationID string,
		sessionType int32,
		maxMsgUpdateTime int64,
	) (*ExtendMsgSetModel, error)
	InsertExtendMsg(ctx context.Context, conversationID string, sessionType int32, msg *ExtendMsgModel) error
	InsertOrUpdateReactionExtendMsgSet(
		ctx context.Context,
		conversationID string,
		sessionType int32,
		clientMsgID string,
		msgFirstModifyTime int64,
		reactionExtensionList map[string]*KeyValueModel,
	) error
	DeleteReactionExtendMsgSet(
		ctx context.Context,
		conversationID string,
		sessionType int32,
		clientMsgID string,
		msgFirstModifyTime int64,
		reactionExtensionList map[string]*KeyValueModel,
	) error
	TakeExtendMsg(
		ctx context.Context,
		conversationID string,
		sessionType int32,
		clientMsgID string,
		maxMsgUpdateTime int64,
	) (extendMsg *ExtendMsgModel, err error)
}

func (ExtendMsgSetModel) TableName() string {
	return CExtendMsgSet
}

func (ExtendMsgSetModel) GetExtendMsgMaxNum() int32 {
	return ExtendMsgMaxNum
}

func (ExtendMsgSetModel) GetConversationID(ID string, index int32) string {
	return ID + ":" + strconv.Itoa(int(index))
}

func (e *ExtendMsgSetModel) SplitConversationIDAndGetIndex() int32 {
	l := strings.Split(e.ConversationID, ":")
	index, _ := strconv.Atoi(l[len(l)-1])
	return int32(index)
}

type GetAllExtendMsgSetOpts struct {
	ExcludeExtendMsgs bool
}

func (ExtendMsgSetModel) Pb2Model(reactionExtensionList map[string]*sdkws.KeyValue) map[string]*KeyValueModel {
	r := make(map[string]*KeyValueModel)
	for key, value := range reactionExtensionList {
		r[key] = &KeyValueModel{
			TypeKey:          value.TypeKey,
			Value:            value.Value,
			LatestUpdateTime: value.LatestUpdateTime,
		}
	}
	return r
}