初始化项目

controller/ad_source_controller.go

@@ -0,0 +1,111 @@
+package controller
+
+import (
+	"context"
+	"strconv"
+
+	"cidService/model/dto"
+	"cidService/service"
+	"github.com/gogf/gf/v2/errors/gerror"
+)
+
+var (
+	AdSource = cAdSource{}
+)
+
+type cAdSource struct{}
+
+// Create 创建广告源
+func (c *cAdSource) Create(ctx context.Context, req *dto.CreateAdSourceReq) (res *dto.GetAdSourceRes, err error) {
+	id, err := service.AdSource.CreateAdSource(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+
+	adSource, err := service.AdSource.GetAdSourceByID(ctx, id)
+	if err != nil {
+		return nil, err
+	}
+
+	return &dto.GetAdSourceRes{
+		AdSource: adSource,
+	}, nil
+}
+
+// Update 更新广告源
+func (c *cAdSource) Update(ctx context.Context, req *dto.UpdateAdSourceReq) (res *dto.GetAdSourceRes, err error) {
+	id, err := strconv.ParseInt(req.Id, 10, 64)
+	if err != nil {
+		return nil, gerror.New("无效的广告源ID")
+	}
+
+	affected, err := service.AdSource.UpdateAdSource(ctx, id, req)
+	if err != nil {
+		return nil, err
+	}
+	if affected == 0 {
+		return nil, gerror.New("广告源更新失败")
+	}
+
+	adSource, err := service.AdSource.GetAdSourceByID(ctx, id)
+	if err != nil {
+		return nil, err
+	}
+
+	return &dto.GetAdSourceRes{
+		AdSource: adSource,
+	}, nil
+}
+
+// Delete 删除广告源
+func (c *cAdSource) Delete(ctx context.Context, req *dto.DeleteAdSourceReq) (res *dto.DeleteAdSourceRes, err error) {
+	id, err := strconv.ParseInt(req.Id, 10, 64)
+	if err != nil {
+		return nil, gerror.New("无效的广告源ID")
+	}
+
+	affected, err := service.AdSource.DeleteAdSource(ctx, id)
+	if err != nil {
+		return nil, err
+	}
+	if affected == 0 {
+		return nil, gerror.New("广告源删除失败")
+	}
+
+	return &dto.DeleteAdSourceRes{
+		Success: true,
+	}, nil
+}
+
+// GetByID 根据ID获取广告源
+func (c *cAdSource) GetByID(ctx context.Context, req *dto.GetAdSourceReq) (res *dto.GetAdSourceRes, err error) {
+	id, err := strconv.ParseInt(req.Id, 10, 64)
+	if err != nil {
+		return nil, gerror.New("无效的广告源ID")
+	}
+
+	adSource, err := service.AdSource.GetAdSourceByID(ctx, id)
+	if err != nil {
+		return nil, err
+	}
+	if adSource == nil {
+		return nil, gerror.New("广告源不存在")
+	}
+
+	return &dto.GetAdSourceRes{
+		AdSource: adSource,
+	}, nil
+}
+
+// GetList 获取广告源列表
+func (c *cAdSource) GetList(ctx context.Context, req *dto.ListAdSourceReq) (res *dto.ListAdSourceRes, err error) {
+	adSources, err := service.AdSource.GetAvailableSources(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	return &dto.ListAdSourceRes{
+		List:  adSources,
+		Total: len(adSources),
+	}, nil
+}

controller/cid_controller.go

@@ -0,0 +1,62 @@
+package controller
+
+import (
+	"context"
+
+	"cidService/model/dto"
+	"cidService/service"
+	"github.com/gogf/gf/v2/errors/gerror"
+)
+
+var (
+	CID = cCID{}
+)
+
+type cCID struct{}
+
+// GenerateCID 生成CID广告
+func (c *cCID) GenerateCID(ctx context.Context, req *dto.GenerateCIDReq) (res *dto.GenerateCIDRes, err error) {
+	if req == nil {
+		return nil, gerror.New("请求参数不能为空")
+	}
+
+	if req.RequestType == "" {
+		req.RequestType = "default" // 默认请求类型
+	}
+
+	result, err := service.CID.GenerateCID(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+
+	return result, nil
+}
+
+// GetStatistics 获取CID统计信息
+func (c *cCID) GetStatistics(ctx context.Context, req *dto.GetCIDStatisticsReq) (res *dto.GetCIDStatisticsRes, err error) {
+	if req == nil {
+		return nil, gerror.New("请求参数不能为空")
+	}
+
+	result, err := service.CID.GetCIDStatistics(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+
+	return result, nil
+}
+
+// GetCIDHistory 获取CID历史记录
+func (c *cCID) GetCIDHistory(ctx context.Context, req *dto.GetCIDHistoryReq) (res *dto.GetCIDHistoryRes, err error) {
+	if req == nil {
+		return nil, gerror.New("请求参数不能为空")
+	}
+
+	// 查询历史记录
+	history, err := service.CID.GetCIDHistory(ctx, 1, req.Page, req.Size) // 临时使用固定用户ID
+	if err != nil {
+		return nil, err
+	}
+
+	return history, nil
+}

controller/strategy_controller.go

@@ -0,0 +1,78 @@
+package controller
+
+import (
+	"context"
+
+	"cidService/model/dto"
+	"cidService/service"
+	"github.com/gogf/gf/v2/errors/gerror"
+)
+
+var (
+	Strategy = cStrategy{}
+)
+
+type cStrategy struct{}
+
+// Create 创建策略
+func (c *cStrategy) Create(ctx context.Context, req *dto.CreateStrategyReq) (res *dto.StrategyRes, err error) {
+	id, err := service.Strategy.CreateStrategy(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+
+	strategy, err := service.Strategy.GetStrategyByID(ctx, id)
+	if err != nil {
+		return nil, err
+	}
+
+	return strategy, nil
+}
+
+// Update 更新策略
+func (c *cStrategy) Update(ctx context.Context, req *dto.UpdateStrategyReq) (res *dto.StrategyRes, err error) {
+	affected, err := service.Strategy.UpdateStrategy(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+	if affected == 0 {
+		return nil, gerror.New("策略更新失败")
+	}
+
+	strategy, err := service.Strategy.GetStrategyByID(ctx, req.Id)
+	if err != nil {
+		return nil, err
+	}
+
+	return strategy, nil
+}
+
+// Delete 删除策略
+func (c *cStrategy) Delete(ctx context.Context, req *dto.DeleteStrategyReq) (res *dto.DeleteStrategyRes, err error) {
+	affected, err := service.Strategy.DeleteStrategy(ctx, req.Id)
+	if err != nil {
+		return nil, err
+	}
+	if affected == 0 {
+		return nil, gerror.New("策略删除失败")
+	}
+
+	return &dto.DeleteStrategyRes{
+		Success: true,
+	}, nil
+}
+
+// GetByID 根据ID获取策略
+func (c *cStrategy) GetByID(ctx context.Context, req *dto.GetStrategyReq) (res *dto.StrategyRes, err error) {
+	strategy, err := service.Strategy.GetStrategyByID(ctx, req.Id)
+	if err != nil {
+		return nil, err
+	}
+
+	return strategy, nil
+}
+
+// GetList 获取策略列表
+func (c *cStrategy) GetList(ctx context.Context, req *dto.GetStrategyListReq) (res *dto.GetStrategyListRes, err error) {
+	return service.Strategy.GetStrategyList(ctx, req)
+}