From 267e7c3b08e5a4b0c7717fba59936d8834c5a6f3 Mon Sep 17 00:00:00 2001 From: wangchuxiao Date: Tue, 1 Nov 2022 19:14:33 +0800 Subject: [PATCH] test cron --- pkg/common/db/mongoModel.go | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/pkg/common/db/mongoModel.go b/pkg/common/db/mongoModel.go index 2164648df..4d767827b 100644 --- a/pkg/common/db/mongoModel.go +++ b/pkg/common/db/mongoModel.go @@ -12,6 +12,7 @@ import ( "fmt" "github.com/go-redis/redis/v8" "github.com/gogo/protobuf/sortkeys" + "go.mongodb.org/mongo-driver/bson/primitive" "go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo/options" "math/rand" @@ -262,26 +263,26 @@ func (d *DataBases) GetMsgBySeqList(uid string, seqList []uint32, operationID st func (d *DataBases) GetUserMsgListByIndex(ID string, index int64) (*UserChat, error) { ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second) c := d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cChat) - //regex := fmt.Sprintf("/^%s/", ID) + regex := fmt.Sprintf("/^%s/", ID) findOpts := options.Find().SetLimit(1).SetSkip(index).SetSort(bson.M{"uid": 1}) - var msgs UserChat - // bson.M{"$regex": primitive.Regex{Pattern: regex}} + var msgs []UserChat + _ = bson.M{"$regex": primitive.Regex{Pattern: regex}} cursor, err := c.Find(ctx, bson.M{"uid": "3729483847:0"}, findOpts) if err != nil { return nil, utils.Wrap(err, "") } v, err := cursor.Current.Values() log.NewInfo("", "values", v, err) - err = cursor.Decode(&msgs) + err = cursor.All(context.Background(), &msgs) if err != nil { return nil, utils.Wrap(err, fmt.Sprintf("cursor is %s", cursor.Current.String())) } - //if len(msgs) > 0 { - // return &msgs[0], err - //} else { - // return nil, errors.New("get msg list failed") - //} - return &msgs, nil + if len(msgs) > 0 { + return &msgs[0], err + } else { + return nil, errors.New("get msg list failed") + } + //return &msgs, nil } func (d *DataBases) DelMongoMsgs(IDList []string) error {