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

26
dao/strategy_dao.go
View File

@@ -4,45 +4,39 @@ import (
"context"
"cid/model/entity"
"gitee.com/red-future---jilin-g/common/mongo"
"go.mongodb.org/mongo-driver/v2/bson"
"go.mongodb.org/mongo-driver/v2/mongo/options"
)
var Strategy = &strategyDao{
NoCache: true,
}
var Strategy = &strategyDao{}
type strategyDao struct {
NoCache bool
}
func (d *strategyDao) SetNoCache() {
Strategy.NoCache = true
}
// GetByName 根据名称获取策略
func (d *strategyDao) GetByName(ctx context.Context, name string) (strategy *entity.Strategy, err error) {
err = mongo.FindOne(ctx, d.NoCache, bson.M{"name": name}, &strategy, "strategies")
err = mongo.DB().FindOne(ctx, bson.M{"name": name}, &strategy, "strategies")
return
}
// GetByID 根据ID获取策略
func (d *strategyDao) GetByID(ctx context.Context, id string) (strategy *entity.Strategy, err error) {
err = mongo.FindOne(ctx, d.NoCache, bson.M{"_id": id}, &strategy, "strategies")
err = mongo.DB().FindOne(ctx, bson.M{"_id": id}, &strategy, "strategies")
return
}
// GetByTenantLevel 根据租户级别获取策略
func (d *strategyDao) GetByTenantLevel(ctx context.Context, tenantLevel string) (strategy *entity.Strategy, err error) {
err = mongo.FindOne(ctx, d.NoCache, bson.M{"tenantLevel": tenantLevel, "status": "active"}, &strategy, "strategies",
err = mongo.DB().FindOne(ctx, bson.M{"tenantLevel": tenantLevel, "status": "active"}, &strategy, "strategies",
options.FindOne().SetSort(bson.M{"priority": -1, "createdAt": 1}))
return
}
// Create 创建策略
func (d *strategyDao) Create(ctx context.Context, strategy *entity.Strategy) (id string, err error) {
ids, err := mongo.Insert(ctx, []interface{}{strategy}, "strategies")
ids, err := mongo.DB().Insert(ctx, []interface{}{strategy}, "strategies")
if err != nil {
return "", err
}
@@ -54,7 +48,7 @@ func (d *strategyDao) Create(ctx context.Context, strategy *entity.Strategy) (id
// Update 更新策略
func (d *strategyDao) Update(ctx context.Context, strategy *entity.Strategy) (affected int64, err error) {
result, err := mongo.Update(ctx, bson.M{"_id": strategy.Id}, bson.M{"$set": strategy}, "strategies")
result, err := mongo.DB().Update(ctx, bson.M{"_id": strategy.Id}, bson.M{"$set": strategy}, "strategies")
if err != nil {
return 0, err
}
@@ -63,7 +57,7 @@ func (d *strategyDao) Update(ctx context.Context, strategy *entity.Strategy) (af
// Delete 删除策略
func (d *strategyDao) Delete(ctx context.Context, id string) (affected int64, err error) {
count, err := mongo.Delete(ctx, bson.M{"_id": id}, "strategies")
count, err := mongo.DB().Delete(ctx, bson.M{"_id": id}, "strategies")
if err != nil {
return 0, err
}
@@ -83,14 +77,14 @@ func (d *strategyDao) GetList(ctx context.Context, page, size int, tenantLevel,
}
// 获取总数
total, err = mongo.Count(ctx, d.NoCache, filter, "strategies")
total, err = mongo.DB().Count(ctx, filter, "strategies")
if err != nil {
return
}
// 分页查询
offset := (page - 1) * size
err = mongo.Find(ctx, d.NoCache, filter, &list, "strategies",
err = mongo.DB().Find(ctx, filter, &list, "strategies",
options.Find().SetSort(bson.M{"priority": -1, "createdAt": -1}).
SetSkip(int64(offset)).
SetLimit(int64(size)))