cid/dao/strategy_dao.go

package dao

import (
	"context"

	"cid/model/entity"

	"gitea.com/red-future/common/beans"
	"gitea.com/red-future/common/db/mongo"
	"go.mongodb.org/mongo-driver/v2/bson"
)

var Strategy = &strategyDao{}

type strategyDao struct {
}

// GetByName 根据名称获取策略
func (d *strategyDao) GetByName(ctx context.Context, name string) (strategy *entity.Strategy, err error) {
	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.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.DB().FindOne(ctx, bson.M{"tenantLevel": tenantLevel, "status": "active"}, &strategy, "strategies")
	return
}

// Create 创建策略
func (d *strategyDao) Create(ctx context.Context, strategy *entity.Strategy) (id string, err error) {
	ids, err := mongo.DB().Insert(ctx, []interface{}{strategy}, "strategies")
	if err != nil {
		return "", err
	}
	if len(ids) > 0 {
		id = ids[0].(string)
	}
	return
}

// Update 更新策略
func (d *strategyDao) Update(ctx context.Context, strategy *entity.Strategy) (affected int64, err error) {
	result, err := mongo.DB().Update(ctx, bson.M{"_id": strategy.Id}, bson.M{"$set": strategy}, "strategies")
	if err != nil {
		return 0, err
	}
	return result, nil
}

// Delete 删除策略
func (d *strategyDao) Delete(ctx context.Context, id string) (affected int64, err error) {
	count, err := mongo.DB().Delete(ctx, bson.M{"_id": id}, "strategies")
	if err != nil {
		return 0, err
	}
	return count, nil
}

// GetList 获取策略列表
func (d *strategyDao) GetList(ctx context.Context, page, size int, tenantLevel, status string) (list []*entity.Strategy, total int64, err error) {
	filter := bson.M{}

	// 筛选条件
	if tenantLevel != "" {
		filter["tenantLevel"] = tenantLevel
	}
	if status != "" {
		filter["status"] = status
	}

	// 获取总数
	total, err = mongo.DB().Count(ctx, filter, "strategies")
	if err != nil {
		return
	}

	// 分页查询，使用common/mongo的Find方法，自动处理分页、租户等
	pageBean := &beans.Page{PageNum: int64(page), PageSize: int64(size)}
	total, err = mongo.DB().Find(ctx, filter, &list, "strategies", pageBean, nil)
	return
}