From b6e659664c15bf734bfab379b11fa94d3137a80e Mon Sep 17 00:00:00 2001 From: Lansong <2095621390@qq.com> Date: Fri, 19 Jan 2024 21:13:02 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E6=94=B9=E8=A1=A8=E5=90=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- biz/infrastructure/mapper/balance/mongo.go | 12 ++++++------ biz/infrastructure/mapper/stock/mongo.go | 12 ++++++------ 2 files changed, 12 insertions(+), 12 deletions(-) diff --git a/biz/infrastructure/mapper/balance/mongo.go b/biz/infrastructure/mapper/balance/mongo.go index b7dd8fe..4f4ff6c 100644 --- a/biz/infrastructure/mapper/balance/mongo.go +++ b/biz/infrastructure/mapper/balance/mongo.go @@ -11,9 +11,9 @@ import ( "go.mongodb.org/mongo-driver/mongo" ) -const CollectionName = "Balance" +const CollectionName = "balance" -var PrefixUserCacheKey = "cache:Balance:" +var PrefixBalanceCacheKey = "cache:balance:" var _ IBalanceMongoMapper = (*MongoMapper)(nil) @@ -48,7 +48,7 @@ func (m *MongoMapper) Insert(ctx context.Context, data *Balance) (string, error) data.ID = primitive.NewObjectID() } - key := PrefixUserCacheKey + data.ID.Hex() + key := PrefixBalanceCacheKey + data.ID.Hex() ID, err := m.conn.InsertOne(ctx, key, data) if err != nil { return "", err @@ -62,7 +62,7 @@ func (m *MongoMapper) FindOne(ctx context.Context, id string) (*Balance, error) return nil, consts.ErrInvalidObjectId } var data Balance - key := PrefixUserCacheKey + id + key := PrefixBalanceCacheKey + id err = m.conn.FindOne(ctx, key, &data, bson.M{consts.ID: oid}) switch { case err == nil: @@ -92,7 +92,7 @@ func (b *Balance) ToBson() bson.M { } func (m *MongoMapper) Update(ctx context.Context, data *Balance, oldBalance *Balance) (*mongo.UpdateResult, error) { - key := PrefixUserCacheKey + oldBalance.ID.Hex() + key := PrefixBalanceCacheKey + oldBalance.ID.Hex() res, err := m.conn.UpdateOne(ctx, key, oldBalance.ToBson(), bson.M{"$set": data}) return res, err } @@ -102,7 +102,7 @@ func (m *MongoMapper) Delete(ctx context.Context, id string) (int64, error) { if err != nil { return 0, err } - key := PrefixUserCacheKey + id + key := PrefixBalanceCacheKey + id res, err := m.conn.DeleteOne(ctx, key, bson.M{consts.ID: oid}) return res, err } diff --git a/biz/infrastructure/mapper/stock/mongo.go b/biz/infrastructure/mapper/stock/mongo.go index c9bf4e7..bbf45a1 100644 --- a/biz/infrastructure/mapper/stock/mongo.go +++ b/biz/infrastructure/mapper/stock/mongo.go @@ -11,9 +11,9 @@ import ( "go.mongodb.org/mongo-driver/mongo" ) -const CollectionName = "Stock" +const CollectionName = "stock" -var PrefixUserCacheKey = "cache:Stock:" +var PrefixStockCacheKey = "cache:stock:" var _ IStockMongoMapper = (*MongoMapper)(nil) @@ -66,7 +66,7 @@ func (m *MongoMapper) Insert(ctx context.Context, data *Stock) (string, error) { data.ID = primitive.NewObjectID() } - key := PrefixUserCacheKey + data.ID.Hex() + key := PrefixStockCacheKey + data.ID.Hex() ID, err := m.conn.InsertOne(ctx, key, data) if err != nil { return "", err @@ -80,7 +80,7 @@ func (m *MongoMapper) FindOne(ctx context.Context, id string) (*Stock, error) { return nil, consts.ErrInvalidObjectId } var data Stock - key := PrefixUserCacheKey + id + key := PrefixStockCacheKey + id err = m.conn.FindOne(ctx, key, &data, bson.M{"_id": oid}) switch { case err == nil: @@ -93,7 +93,7 @@ func (m *MongoMapper) FindOne(ctx context.Context, id string) (*Stock, error) { } func (m *MongoMapper) Update(ctx context.Context, data *Stock, oldAmount *int64) (*mongo.UpdateResult, error) { - key := PrefixUserCacheKey + data.ID.Hex() + key := PrefixStockCacheKey + data.ID.Hex() res, err := m.conn.UpdateOne(ctx, key, bson.M{consts.ID: data.ID, consts.Amount: oldAmount}, bson.M{"$set": data}) return res, err } @@ -103,7 +103,7 @@ func (m *MongoMapper) Delete(ctx context.Context, id string) (int64, error) { if err != nil { return 0, err } - key := PrefixUserCacheKey + id + key := PrefixStockCacheKey + id res, err := m.conn.DeleteOne(ctx, key, bson.M{"_id": oid}) return res, err }