cid/service/ad_source_service.go

package service

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

	"github.com/gogf/gf/v2/errors/gerror"
)

type adSource struct{}

// AdSource 广告源服务
var AdSource = new(adSource)

// GetAvailableSources 获取可用的广告源列表
func (s *adSource) GetAvailableSources(ctx context.Context) (list []*entity.AdSource, err error) {
	return dao.AdSource.GetAvailableSources(ctx)
}

// GetSourcesByProvider 根据提供商获取广告源
func (s *adSource) GetSourcesByProvider(ctx context.Context, provider string) (list []*entity.AdSource, err error) {
	return dao.AdSource.GetSourcesByProvider(ctx, provider)
}

// CreateAdSource 创建广告源
func (s *adSource) CreateAdSource(ctx context.Context, req *dto.CreateAdSourceReq) (id string, err error) {
	// 检查广告源名称是否已存在
	existingSource, err := dao.AdSource.GetByName(ctx, req.Name)
	if err != nil {
		return "", err
	}
	if existingSource != nil {
		return "", gerror.New("广告源名称已存在")
	}

	adSource := &entity.AdSource{
		Name:     req.Name,
		Code:     req.Code,
		Provider: req.Provider,
		Type:     req.Type,
		APIConfig: config.APIConfig{
			Endpoint: req.APIEndpoint,
		},
	}

	// 设置状态
	adSource.Status = "active" // 默认状态

	return dao.AdSource.Create(ctx, adSource)
}

// UpdateAdSource 更新广告源
func (s *adSource) UpdateAdSource(ctx context.Context, id string, req *dto.UpdateAdSourceReq) (affected int64, err error) {

	// 检查广告源是否存在
	existingSource, err := dao.AdSource.GetByID(ctx, id)
	if err != nil {
		return 0, err
	}
	if existingSource == nil {
		return 0, gerror.New("广告源不存在")
	}

	// 如果更新名称，检查是否与其他广告源冲突
	if req.Name != "" && req.Name != existingSource.Name {
		conflictSource, err := dao.AdSource.GetByName(ctx, req.Name)
		if err != nil {
			return 0, err
		}
		if conflictSource != nil {
			return 0, gerror.New("广告源名称已存在")
		}
	}

	// 构建更新数据
	updateData := &entity.AdSource{}
	if req.Name != "" {
		updateData.Name = req.Name
	}
	if req.APIEndpoint != "" {
		updateData.APIConfig.Endpoint = req.APIEndpoint
	}

	return dao.AdSource.UpdateFields(ctx, id, updateData)
}

// DeleteAdSource 删除广告源
func (s *adSource) DeleteAdSource(ctx context.Context, id string) (affected int64, err error) {
	// 检查广告源是否存在
	existingSource, err := dao.AdSource.GetByID(ctx, id)
	if err != nil {
		return 0, err
	}
	if existingSource == nil {
		return 0, gerror.New("广告源不存在")
	}

	return dao.AdSource.Delete(ctx, id)
}

// GetAdSourceByID 根据ID获取广告源
func (s *adSource) GetAdSourceByID(ctx context.Context, id string) (adSource *entity.AdSource, err error) {
	return dao.AdSource.GetByID(ctx, id)
}