初始化项目

dao/strategy_dao.go

@@ -1,10 +1,12 @@
 package dao
 
 import (
-	"cid/model/entity"
 	"context"
 
-	"github.com/gogf/gf/v2/frame/g"
+	"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{}
@@ -13,86 +15,77 @@ type strategyDao struct{}
 
 // GetByName 根据名称获取策略
 func (d *strategyDao) GetByName(ctx context.Context, name string) (strategy *entity.Strategy, err error) {
-	err = g.DB().Model("strategies").
-		Where("name = ?", name).
-		Scan(&strategy)
+	err = mongo.FindOne(ctx, bson.M{"name": name}, &strategy, "strategies")
 	return
 }
 
 // GetByID 根据ID获取策略
-func (d *strategyDao) GetByID(ctx context.Context, id int64) (strategy *entity.Strategy, err error) {
-	err = g.DB().Model("strategies").
-		Where("id = ?", id).
-		Scan(&strategy)
+func (d *strategyDao) GetByID(ctx context.Context, id string) (strategy *entity.Strategy, err error) {
+	err = mongo.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 = g.DB().Model("strategies").
-		Where("tenant_level = ? AND status = ?", tenantLevel, "active").
-		Order("priority DESC, created_at ASC").
-		Scan(&strategy)
+	err = mongo.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 int64, err error) {
-	result, err := g.DB().Model("strategies").Insert(strategy)
+func (d *strategyDao) Create(ctx context.Context, strategy *entity.Strategy) (id string, err error) {
+	ids, err := mongo.Insert(ctx, []interface{}{strategy}, "strategies")
 	if err != nil {
-		return 0, err
+		return "", err
+	}
+	if len(ids) > 0 {
+		id = ids[0].(string)
 	}
-	id, err = result.LastInsertId()
 	return
 }
 
 // Update 更新策略
 func (d *strategyDao) Update(ctx context.Context, strategy *entity.Strategy) (affected int64, err error) {
-	result, err := g.DB().Model("strategies").
-		Where("id = ?", strategy.Id).
-		Update(strategy)
+	result, err := mongo.Update(ctx, bson.M{"_id": strategy.Id}, bson.M{"$set": strategy}, "strategies")
 	if err != nil {
 		return 0, err
 	}
-	return result.RowsAffected()
+	return result.ModifiedCount, nil
 }
 
 // Delete 删除策略
-func (d *strategyDao) Delete(ctx context.Context, id int64) (affected int64, err error) {
-	result, err := g.DB().Model("strategies").
-		Where("id = ?", id).
-		Delete()
+func (d *strategyDao) Delete(ctx context.Context, id string) (affected int64, err error) {
+	count, err := mongo.Delete(ctx, bson.M{"_id": id}, "strategies")
 	if err != nil {
 		return 0, err
 	}
-	return result.RowsAffected()
+	return count, nil
 }
 
 // GetList 获取策略列表
 func (d *strategyDao) GetList(ctx context.Context, page, size int, tenantLevel, status string) (list []*entity.Strategy, total int64, err error) {
-	model := g.DB().Model("strategies")
+	filter := bson.M{}
 
 	// 筛选条件
 	if tenantLevel != "" {
-		model = model.Where("tenant_level = ?", tenantLevel)
+		filter["tenantLevel"] = tenantLevel
 	}
 	if status != "" {
-		model = model.Where("status = ?", status)
+		filter["status"] = status
 	}
 
 	// 获取总数
-	count, err := model.Count()
+	total, err = mongo.Count(ctx, filter, "strategies")
 	if err != nil {
 		return
 	}
-	total = int64(count)
 
 	// 分页查询
 	offset := (page - 1) * size
-	err = model.Order("priority DESC, created_at DESC").
-		Offset(offset).
-		Limit(size).
-		Scan(&list)
+	err = mongo.Find(ctx, filter, &list, "strategies",
+		options.Find().SetSort(bson.M{"priority": -1, "createdAt": -1}).
+			SetSkip(int64(offset)).
+			SetLimit(int64(size)))
 
 	return
 }