mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-10-27 14:02:15 +08:00
146 lines
4.3 KiB
Go
146 lines
4.3 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 model
|
|
|
|
import (
|
|
"github.com/openimsdk/protocol/sdkws"
|
|
"github.com/openimsdk/tools/errs"
|
|
"strconv"
|
|
)
|
|
|
|
const (
|
|
singleGocMsgNum = 100
|
|
singleGocMsgNum5000 = 5000
|
|
MsgTableName = "msg"
|
|
OldestList = 0
|
|
NewestList = -1
|
|
)
|
|
|
|
var ErrMsgListNotExist = errs.New("user not have msg in mongoDB")
|
|
|
|
type MsgDocModel struct {
|
|
DocID string `bson:"doc_id"`
|
|
Msg []*MsgInfoModel `bson:"msgs"`
|
|
}
|
|
|
|
type RevokeModel struct {
|
|
Role int32 `bson:"role"`
|
|
UserID string `bson:"user_id"`
|
|
Nickname string `bson:"nickname"`
|
|
Time int64 `bson:"time"`
|
|
}
|
|
|
|
type OfflinePushModel struct {
|
|
Title string `bson:"title"`
|
|
Desc string `bson:"desc"`
|
|
Ex string `bson:"ex"`
|
|
IOSPushSound string `bson:"ios_push_sound"`
|
|
IOSBadgeCount bool `bson:"ios_badge_count"`
|
|
}
|
|
|
|
type MsgDataModel struct {
|
|
SendID string `bson:"send_id"`
|
|
RecvID string `bson:"recv_id"`
|
|
GroupID string `bson:"group_id"`
|
|
ClientMsgID string `bson:"client_msg_id"`
|
|
ServerMsgID string `bson:"server_msg_id"`
|
|
SenderPlatformID int32 `bson:"sender_platform_id"`
|
|
SenderNickname string `bson:"sender_nickname"`
|
|
SenderFaceURL string `bson:"sender_face_url"`
|
|
SessionType int32 `bson:"session_type"`
|
|
MsgFrom int32 `bson:"msg_from"`
|
|
ContentType int32 `bson:"content_type"`
|
|
Content string `bson:"content"`
|
|
Seq int64 `bson:"seq"`
|
|
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"`
|
|
AttachedInfo string `bson:"attached_info"`
|
|
Ex string `bson:"ex"`
|
|
}
|
|
|
|
type MsgInfoModel struct {
|
|
Msg *MsgDataModel `bson:"msg"`
|
|
Revoke *RevokeModel `bson:"revoke"`
|
|
DelList []string `bson:"del_list"`
|
|
IsRead bool `bson:"is_read"`
|
|
}
|
|
|
|
type UserCount struct {
|
|
UserID string `bson:"user_id"`
|
|
Count int64 `bson:"count"`
|
|
}
|
|
|
|
type GroupCount struct {
|
|
GroupID string `bson:"group_id"`
|
|
Count int64 `bson:"count"`
|
|
}
|
|
|
|
func (*MsgDocModel) TableName() string {
|
|
return MsgTableName
|
|
}
|
|
|
|
func (*MsgDocModel) GetSingleGocMsgNum() int64 {
|
|
return singleGocMsgNum
|
|
}
|
|
|
|
func (*MsgDocModel) GetSingleGocMsgNum5000() int64 {
|
|
return singleGocMsgNum5000
|
|
}
|
|
|
|
func (m *MsgDocModel) IsFull() bool {
|
|
return m.Msg[len(m.Msg)-1].Msg != nil
|
|
}
|
|
|
|
func (m *MsgDocModel) GetDocID(conversationID string, seq int64) string {
|
|
seqSuffix := (seq - 1) / singleGocMsgNum
|
|
return m.indexGen(conversationID, seqSuffix)
|
|
}
|
|
|
|
func (m *MsgDocModel) GetDocIDSeqsMap(conversationID string, seqs []int64) map[string][]int64 {
|
|
t := make(map[string][]int64)
|
|
for i := 0; i < len(seqs); i++ {
|
|
docID := m.GetDocID(conversationID, seqs[i])
|
|
if value, ok := t[docID]; !ok {
|
|
var temp []int64
|
|
t[docID] = append(temp, seqs[i])
|
|
} else {
|
|
t[docID] = append(value, seqs[i])
|
|
}
|
|
}
|
|
return t
|
|
}
|
|
|
|
func (*MsgDocModel) GetMsgIndex(seq int64) int64 {
|
|
return (seq - 1) % singleGocMsgNum
|
|
}
|
|
|
|
func (*MsgDocModel) indexGen(conversationID string, seqSuffix int64) string {
|
|
return conversationID + ":" + strconv.FormatInt(seqSuffix, 10)
|
|
}
|
|
|
|
func (*MsgDocModel) GenExceptionMessageBySeqs(seqs []int64) (exceptionMsg []*sdkws.MsgData) {
|
|
for _, v := range seqs {
|
|
msgModel := new(sdkws.MsgData)
|
|
msgModel.Seq = v
|
|
exceptionMsg = append(exceptionMsg, msgModel)
|
|
}
|
|
return exceptionMsg
|
|
}
|