初始化项目

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
+}