red-future/cid
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

mongo.go重构

Browse Source
This commit is contained in:
张斌
2025-12-30 10:55:35 +08:00
parent ec2cc8e995
commit b48613bca1
9 changed files with 65 additions and 102 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
dao/ad_position_dao.go
View File

@@ -13,16 +13,9 @@ import (
"go.mongodb.org/mongo-driver/v2/mongo/options"
)
var AdPosition = &adPosition{
NoCache: true,
}
var AdPosition = &adPosition{}
type adPosition struct {
NoCache bool
}
func (d *adPosition) SetNoCache() {
AdPosition.NoCache = true
}
// Insert 插入广告位
@@ -36,7 +29,7 @@ func (d *adPosition) Insert(ctx context.Context, adPosition *entity.AdPosition)
g.Log().Infof(ctx, "获取到stream消息: %v", streamMsg)
}
_, err = mongo.Insert(ctx, []interface{}{adPosition}, entity.AdPositionCollection)
_, err = mongo.DB().Insert(ctx, []interface{}{adPosition}, entity.AdPositionCollection)
return
}
@@ -124,7 +117,7 @@ func (d *adPosition) Update(ctx context.Context, req *dto.UpdateAdPositionReq) (
if len(updateFields) > 0 {
update := bson.M{"$set": updateFields}
_, err = mongo.Update(ctx, filter, update, entity.AdPositionCollection)
_, err = mongo.DB().Update(ctx, filter, update, entity.AdPositionCollection)
}
return
}
@@ -138,7 +131,7 @@ func (d *adPosition) UpdateStatus(ctx context.Context, id, status string) (err e
filter := bson.M{"_id": objectId}
update := bson.M{"$set": bson.M{"status": status}}
_, err = mongo.Update(ctx, filter, update, entity.AdPositionCollection)
_, err = mongo.DB().Update(ctx, filter, update, entity.AdPositionCollection)
return
}
@@ -151,7 +144,7 @@ func (d *adPosition) UpdateStatistics(ctx context.Context, id string, stats map[
filter := bson.M{"_id": objectId}
update := bson.M{"$set": stats}
_, err = mongo.Update(ctx, filter, update, entity.AdPositionCollection)
_, err = mongo.DB().Update(ctx, filter, update, entity.AdPositionCollection)
return
}
@@ -164,7 +157,7 @@ func (d *adPosition) GetOne(ctx context.Context, id string) (adPosition *entity.
filter := bson.M{"_id": objectId}
adPosition = &entity.AdPosition{}
err = mongo.FindOne(ctx, d.NoCache, filter, adPosition, entity.AdPositionCollection)
err = mongo.DB().FindOne(ctx, filter, adPosition, entity.AdPositionCollection)
return
}
@@ -173,7 +166,7 @@ func (d *adPosition) GetByCode(ctx context.Context, code string) (adPosition *en
filter := bson.M{"positionCode": code}
adPosition = &entity.AdPosition{}
err = mongo.FindOne(ctx, d.NoCache, filter, adPosition, entity.AdPositionCollection)
err = mongo.DB().FindOne(ctx, filter, adPosition, entity.AdPositionCollection)
return
}
@@ -215,7 +208,7 @@ func (d *adPosition) buildListFilter(req *dto.ListAdPositionReq) bson.M {
// checkTotalCount 检查总数
func (d *adPosition) checkTotalCount(ctx context.Context, filter bson.M) (total int64, err error) {
total, err = mongo.Count(ctx, d.NoCache, filter, entity.AdPositionCollection)
total, err = mongo.DB().Count(ctx, filter, entity.AdPositionCollection)
return
}
@@ -248,7 +241,7 @@ func (d *adPosition) List(ctx context.Context, req *dto.ListAdPositionReq) (list
opts := options.Find().SetLimit(limit).SetSkip(skip).SetSort(sort)
err = mongo.Find(ctx, d.NoCache, filter, &list, entity.AdPositionCollection, opts)
err = mongo.DB().Find(ctx, filter, &list, entity.AdPositionCollection, opts)
return
}
@@ -260,6 +253,6 @@ func (d *adPosition) GetAvailableAdPositions(ctx context.Context) (list []*entit
opts := options.Find().SetSort(bson.M{"createdAt": -1})
err = mongo.Find(ctx, d.NoCache, filter, &list, entity.AdPositionCollection, opts)
err = mongo.DB().Find(ctx, filter, &list, entity.AdPositionCollection, opts)
return
}