mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-10-26 21:22:16 +08:00
redis msg cache
This commit is contained in:
parent
1219b47c79
commit
d44fab8518
@ -39,12 +39,6 @@ type MsgMgo struct {
|
|||||||
model model.MsgDocModel
|
model model.MsgDocModel
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *MsgMgo) PushMsgsToDoc(ctx context.Context, docID string, msgsToMongo []model.MsgInfoModel) error {
|
|
||||||
filter := bson.M{"doc_id": docID}
|
|
||||||
update := bson.M{"$push": bson.M{"msgs": bson.M{"$each": msgsToMongo}}}
|
|
||||||
return mongoutil.UpdateOne(ctx, m.coll, filter, update, false)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MsgMgo) Create(ctx context.Context, msg *model.MsgDocModel) error {
|
func (m *MsgMgo) Create(ctx context.Context, msg *model.MsgDocModel) error {
|
||||||
return mongoutil.InsertMany(ctx, m.coll, []*model.MsgDocModel{msg})
|
return mongoutil.InsertMany(ctx, m.coll, []*model.MsgDocModel{msg})
|
||||||
}
|
}
|
||||||
@ -77,16 +71,6 @@ func (m *MsgMgo) PushUnique(ctx context.Context, docID string, index int64, key
|
|||||||
return mongoutil.UpdateOneResult(ctx, m.coll, filter, update)
|
return mongoutil.UpdateOneResult(ctx, m.coll, filter, update)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *MsgMgo) UpdateMsgContent(ctx context.Context, docID string, index int64, msg []byte) error {
|
|
||||||
filter := bson.M{"doc_id": docID}
|
|
||||||
update := bson.M{"$set": bson.M{fmt.Sprintf("msgs.%d.msg", index): msg}}
|
|
||||||
return mongoutil.UpdateOne(ctx, m.coll, filter, update, false)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MsgMgo) IsExistDocID(ctx context.Context, docID string) (bool, error) {
|
|
||||||
return mongoutil.Exist(ctx, m.coll, bson.M{"doc_id": docID})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MsgMgo) FindOneByDocID(ctx context.Context, docID string) (*model.MsgDocModel, error) {
|
func (m *MsgMgo) FindOneByDocID(ctx context.Context, docID string) (*model.MsgDocModel, error) {
|
||||||
return mongoutil.FindOne[*model.MsgDocModel](ctx, m.coll, bson.M{"doc_id": docID})
|
return mongoutil.FindOne[*model.MsgDocModel](ctx, m.coll, bson.M{"doc_id": docID})
|
||||||
}
|
}
|
||||||
@ -215,13 +199,6 @@ func (m *MsgMgo) GetOldestMsg(ctx context.Context, conversationID string) (*mode
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *MsgMgo) DeleteDocs(ctx context.Context, docIDs []string) error {
|
|
||||||
if len(docIDs) == 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return mongoutil.DeleteMany(ctx, m.coll, bson.M{"doc_id": bson.M{"$in": docIDs}})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MsgMgo) GetMsgDocModelByIndex(ctx context.Context, conversationID string, index, sort int64) (*model.MsgDocModel, error) {
|
func (m *MsgMgo) GetMsgDocModelByIndex(ctx context.Context, conversationID string, index, sort int64) (*model.MsgDocModel, error) {
|
||||||
if sort != 1 && sort != -1 {
|
if sort != 1 && sort != -1 {
|
||||||
return nil, errs.ErrArgs.WrapMsg("mongo sort must be 1 or -1")
|
return nil, errs.ErrArgs.WrapMsg("mongo sort must be 1 or -1")
|
||||||
@ -420,22 +397,6 @@ func (m *MsgMgo) searchMessage(ctx context.Context, req *msg.SearchMessageReq) (
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *MsgMgo) getDocRange(ctx context.Context, id primitive.ObjectID, index []int64) ([]*model.MsgInfoModel, error) {
|
|
||||||
if len(index) == 0 {
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
pipeline := bson.A{
|
|
||||||
bson.M{"$match": bson.M{"_id": id}},
|
|
||||||
bson.M{"$project": "$msgs"},
|
|
||||||
}
|
|
||||||
msgs, err := mongoutil.Aggregate[*model.MsgInfoModel](ctx, m.coll, pipeline)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return msgs, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MsgMgo) SearchMessage(ctx context.Context, req *msg.SearchMessageReq) (int64, []*model.MsgInfoModel, error) {
|
func (m *MsgMgo) SearchMessage(ctx context.Context, req *msg.SearchMessageReq) (int64, []*model.MsgInfoModel, error) {
|
||||||
count, data, err := m.searchMessage(ctx, req)
|
count, data, err := m.searchMessage(ctx, req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -980,18 +941,6 @@ func (m *MsgMgo) GetRandBeforeMsg(ctx context.Context, ts int64, limit int) ([]*
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *MsgMgo) DeleteMsgByIndex(ctx context.Context, docID string, index []int) error {
|
|
||||||
if len(index) == 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
model := &model.MsgInfoModel{DelList: []string{}}
|
|
||||||
set := make(map[string]any)
|
|
||||||
for i := range index {
|
|
||||||
set[fmt.Sprintf("msgs.%d", i)] = model
|
|
||||||
}
|
|
||||||
return mongoutil.UpdateOne(ctx, m.coll, bson.M{"doc_id": docID}, bson.M{"$set": set}, true)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MsgMgo) DeleteDoc(ctx context.Context, docID string) error {
|
func (m *MsgMgo) DeleteDoc(ctx context.Context, docID string) error {
|
||||||
return mongoutil.DeleteOne(ctx, m.coll, bson.M{"doc_id": docID})
|
return mongoutil.DeleteOne(ctx, m.coll, bson.M{"doc_id": docID})
|
||||||
}
|
}
|
||||||
|
|||||||
@ -24,29 +24,20 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Msg interface {
|
type Msg interface {
|
||||||
//PushMsgsToDoc(ctx context.Context, docID string, msgsToMongo []model.MsgInfoModel) error
|
|
||||||
Create(ctx context.Context, model *model.MsgDocModel) error
|
Create(ctx context.Context, model *model.MsgDocModel) error
|
||||||
UpdateMsg(ctx context.Context, docID string, index int64, key string, value any) (*mongo.UpdateResult, error)
|
UpdateMsg(ctx context.Context, docID string, index int64, key string, value any) (*mongo.UpdateResult, error)
|
||||||
PushUnique(ctx context.Context, docID string, index int64, key string, value any) (*mongo.UpdateResult, error)
|
PushUnique(ctx context.Context, docID string, index int64, key string, value any) (*mongo.UpdateResult, error)
|
||||||
UpdateMsgContent(ctx context.Context, docID string, index int64, msg []byte) error
|
|
||||||
IsExistDocID(ctx context.Context, docID string) (bool, error)
|
|
||||||
FindOneByDocID(ctx context.Context, docID string) (*model.MsgDocModel, error)
|
FindOneByDocID(ctx context.Context, docID string) (*model.MsgDocModel, error)
|
||||||
GetMsgBySeqIndexIn1Doc(ctx context.Context, userID, docID string, seqs []int64) ([]*model.MsgInfoModel, error)
|
GetMsgBySeqIndexIn1Doc(ctx context.Context, userID, docID string, seqs []int64) ([]*model.MsgInfoModel, error)
|
||||||
GetNewestMsg(ctx context.Context, conversationID string) (*model.MsgInfoModel, error)
|
GetNewestMsg(ctx context.Context, conversationID string) (*model.MsgInfoModel, error)
|
||||||
GetOldestMsg(ctx context.Context, conversationID string) (*model.MsgInfoModel, error)
|
GetOldestMsg(ctx context.Context, conversationID string) (*model.MsgInfoModel, error)
|
||||||
DeleteDocs(ctx context.Context, docIDs []string) error
|
|
||||||
GetMsgDocModelByIndex(ctx context.Context, conversationID string, index, sort int64) (*model.MsgDocModel, error)
|
|
||||||
DeleteMsgsInOneDocByIndex(ctx context.Context, docID string, indexes []int) error
|
DeleteMsgsInOneDocByIndex(ctx context.Context, docID string, indexes []int) error
|
||||||
MarkSingleChatMsgsAsRead(ctx context.Context, userID string, docID string, indexes []int64) error
|
MarkSingleChatMsgsAsRead(ctx context.Context, userID string, docID string, indexes []int64) error
|
||||||
SearchMessage(ctx context.Context, req *msg.SearchMessageReq) (int64, []*model.MsgInfoModel, error)
|
SearchMessage(ctx context.Context, req *msg.SearchMessageReq) (int64, []*model.MsgInfoModel, error)
|
||||||
RangeUserSendCount(ctx context.Context, start time.Time, end time.Time, group bool, ase bool, pageNumber int32, showNumber int32) (msgCount int64, userCount int64, users []*model.UserCount, dateCount map[string]int64, err error)
|
RangeUserSendCount(ctx context.Context, start time.Time, end time.Time, group bool, ase bool, pageNumber int32, showNumber int32) (msgCount int64, userCount int64, users []*model.UserCount, dateCount map[string]int64, err error)
|
||||||
RangeGroupSendCount(ctx context.Context, start time.Time, end time.Time, ase bool, pageNumber int32, showNumber int32) (msgCount int64, userCount int64, groups []*model.GroupCount, dateCount map[string]int64, err error)
|
RangeGroupSendCount(ctx context.Context, start time.Time, end time.Time, ase bool, pageNumber int32, showNumber int32) (msgCount int64, userCount int64, groups []*model.GroupCount, dateCount map[string]int64, err error)
|
||||||
|
|
||||||
DeleteDoc(ctx context.Context, docID string) error
|
DeleteDoc(ctx context.Context, docID string) error
|
||||||
DeleteMsgByIndex(ctx context.Context, docID string, index []int) error
|
|
||||||
GetRandBeforeMsg(ctx context.Context, ts int64, limit int) ([]*model.MsgDocModel, error)
|
GetRandBeforeMsg(ctx context.Context, ts int64, limit int) ([]*model.MsgDocModel, error)
|
||||||
|
|
||||||
GetLastMessageSeqByTime(ctx context.Context, conversationID string, time int64) (int64, error)
|
GetLastMessageSeqByTime(ctx context.Context, conversationID string, time int64) (int64, error)
|
||||||
|
|
||||||
FindSeqs(ctx context.Context, conversationID string, seqs []int64) ([]*model.MsgInfoModel, error)
|
FindSeqs(ctx context.Context, conversationID string, seqs []int64) ([]*model.MsgInfoModel, error)
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user