cid/dao/ad_position_dao.go

package dao

import (
	"cid/model/dto"
	"cid/model/entity"
	"context"

	"gitea.com/red-future/common/beans"
	"gitea.com/red-future/common/db/mongo"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/util/gconv"
	"go.mongodb.org/mongo-driver/v2/bson"
)

var AdPosition = &adPosition{}

type adPosition struct {
}

// Insert 插入广告位
func (d *adPosition) Insert(ctx context.Context, req *dto.AddAdPositionReq) (ids []any, err error) {
	var result entity.AdPosition
	if err = gconv.Struct(req, &result); err != nil {
		return
	}
	ids, err = mongo.DB().Insert(ctx, []interface{}{&result}, entity.AdPositionCollection)
	return
}

// Update 更新广告位
func (d *adPosition) Update(ctx context.Context, id *bson.ObjectID, updateData *entity.AdPosition) (err error) {
	filter := bson.M{"_id": id}

	if !g.IsEmpty(updateData) {
		bsonm, err := mongo.EntityToBSONM(updateData)
		if err != nil {
			return err
		}
		update := bson.M{"$set": bsonm}
		_, err = mongo.DB().Update(ctx, filter, update, entity.AdPositionCollection)
	}
	return
}

// UpdateStatus 更新广告位状态
func (d *adPosition) UpdateStatus(ctx context.Context, id *bson.ObjectID, status string) (err error) {
	filter := bson.M{"_id": id}
	update := bson.M{"$set": bson.M{"status": status}}

	_, err = mongo.DB().Update(ctx, filter, update, entity.AdPositionCollection)
	return
}

// GetOne 获取单个广告位
func (d *adPosition) GetOne(ctx context.Context, id *bson.ObjectID) (adPosition *entity.AdPosition, err error) {
	filter := bson.M{"_id": id}

	adPosition = &entity.AdPosition{}
	err = mongo.DB().FindOne(ctx, filter, adPosition, entity.AdPositionCollection)
	return
}

// Delete 删除广告位
func (d *adPosition) Delete(ctx context.Context, id *bson.ObjectID) (err error) {
	filter := bson.M{"_id": id}
	_, err = mongo.DB().Delete(ctx, filter, entity.AdPositionCollection)
	return
}

// buildListFilter 构建列表查询的过滤条件
func (d *adPosition) buildListFilter(req *dto.ListAdPositionReq) bson.M {
	filter := bson.M{}

	if !g.IsEmpty(req.Name) {
		filter["name"] = bson.M{"$regex": req.Name, "$options": "i"}
	}
	if !g.IsEmpty(req.PositionCode) {
		filter["positionCode"] = req.PositionCode
	}
	if !g.IsEmpty(req.PageName) {
		filter["page"] = req.PageName
	}
	if !g.IsEmpty(req.Section) {
		filter["section"] = req.Section
	}
	if !g.IsEmpty(req.Status) {
		filter["status"] = req.Status
	}
	if !g.IsEmpty(req.AdFormat) {
		filter["adFormat"] = req.AdFormat
	}

	// 处理日期范围
	if len(req.DateRange) == 2 {
		startTime := gconv.Int64(req.DateRange[0])
		endTime := gconv.Int64(req.DateRange[1])
		filter["createdAt"] = bson.M{
			"$gte": startTime,
			"$lte": endTime,
		}
	}

	return filter
}

// List 获取广告位列表
func (d *adPosition) List(ctx context.Context, req *dto.ListAdPositionReq) (list []*entity.AdPosition, total int64, err error) {
	// 构建查询过滤条件
	filter := d.buildListFilter(req)

	// 使用common/mongo的Find方法，自动处理分页、租户等
	total, err = mongo.DB().Find(ctx, filter, &list, entity.AdPositionCollection, req.Page, nil)
	return
}

// GetAvailableAdPositions 获取可用的广告位列表
func (d *adPosition) GetAvailableAdPositions(ctx context.Context) (list []*entity.AdPosition, err error) {
	filter := bson.M{
		"status": "启用", // 只返回启用的广告位
	}

	// 使用空的Page参数获取所有数据
	page := &beans.Page{PageNum: 1, PageSize: -1} // -1表示不分页
	_, err = mongo.DB().Find(ctx, filter, &list, entity.AdPositionCollection, page, nil)
	return
}