blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-12-17 19:47:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

update skip logic.

Browse Source
This commit is contained in:
Monet Lee 2024-08-13 17:09:11 +08:00
parent 7c482c9466
commit d467381b6f
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/common/storage/database/mgo/msg.go
View File

@ -1231,6 +1231,7 @@ func (m *MsgMgo) GetDocIDs(ctx context.Context) ([]string, error) {
limit := 5000 limit := 5000
var skip int var skip int
var docIDs []string var docIDs []string
var offset int
count, err := m.coll.CountDocuments(ctx, bson.M{}) count, err := m.coll.CountDocuments(ctx, bson.M{})
if err != nil { if err != nil {
@ -1241,7 +1242,8 @@ func (m *MsgMgo) GetDocIDs(ctx context.Context) ([]string, error) {
skip = 0 skip = 0
} else { } else {
rand.Seed(uint64(time.Now().UnixMilli())) rand.Seed(uint64(time.Now().UnixMilli()))
skip = rand.Intn(int(count)) skip = rand.Intn(int(count / int64(limit)))
offset = skip * (int(count) / limit)
} }
res, err := mongoutil.Aggregate[*model.MsgDocModel](ctx, m.coll, []bson.M{ res, err := mongoutil.Aggregate[*model.MsgDocModel](ctx, m.coll, []bson.M{
@ -1251,7 +1253,7 @@ func (m *MsgMgo) GetDocIDs(ctx context.Context) ([]string, error) {
}, },
}, },
{ {
"$skip": skip, "$skip": offset,
}, },
{ {
"$limit": limit, "$limit": limit,