初始化项目

2025-12-06 10:38:48 +08:00
parent c9fcfc761e
commit 88a2753211
20 changed files with 2552 additions and 74 deletions
--- a/dao/ad_source_dao.go
+++ b/dao/ad_source_dao.go
@@ -0,0 +1,91 @@
+package dao
+
+import (
+	"cidService/model/entity"
+	"context"
+
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+var (
+	AdSource = adSourceDao{}
+)
+
+type adSourceDao struct{}
+
+// GetByName 根据名称获取广告源
+func (d *adSourceDao) GetByName(ctx context.Context, name string) (adSource *entity.AdSource, err error) {
+	err = g.DB().Model("ad_sources").
+		Where("name = ?", name).
+		Scan(&adSource)
+	return
+}
+
+// GetAvailableSources 获取可用的广告源
+func (d *adSourceDao) GetAvailableSources(ctx context.Context) (list []*entity.AdSource, err error) {
+	err = g.DB().Model("ad_sources").
+		Where("status = ?", "active").
+		Order("priority DESC, created_at ASC").
+		Scan(&list)
+	return
+}
+
+// GetSourcesByProvider 根据提供商获取广告源
+func (d *adSourceDao) GetSourcesByProvider(ctx context.Context, provider string) (list []*entity.AdSource, err error) {
+	err = g.DB().Model("ad_sources").
+		Where("provider = ? AND status = ?", provider, "active").
+		Order("priority DESC").
+		Scan(&list)
+	return
+}
+
+// Create 创建广告源
+func (d *adSourceDao) Create(ctx context.Context, adSource *entity.AdSource) (id int64, err error) {
+	result, err := g.DB().Model("ad_sources").Insert(adSource)
+	if err != nil {
+		return 0, err
+	}
+	id, err = result.LastInsertId()
+	return
+}
+
+// Update 更新广告源
+func (d *adSourceDao) Update(ctx context.Context, adSource *entity.AdSource) (affected int64, err error) {
+	result, err := g.DB().Model("ad_sources").
+		Where("id = ?", adSource.Id).
+		Update(adSource)
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected()
+}
+
+// Delete 删除广告源
+func (d *adSourceDao) Delete(ctx context.Context, id int64) (affected int64, err error) {
+	result, err := g.DB().Model("ad_sources").
+		Where("id = ?", id).
+		Delete()
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected()
+}
+
+// GetByID 根据ID获取广告源
+func (d *adSourceDao) GetByID(ctx context.Context, id int64) (adSource *entity.AdSource, err error) {
+	err = g.DB().Model("ad_sources").
+		Where("id = ?", id).
+		Scan(&adSource)
+	return
+}
+
+// UpdateFields 更新广告源部分字段
+func (d *adSourceDao) UpdateFields(ctx context.Context, id int64, data *entity.AdSource) (affected int64, err error) {
+	result, err := g.DB().Model("ad_sources").
+		Where("id = ?", id).
+		Update(data)
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected()
+}
--- a/dao/cid_request_dao.go
+++ b/dao/cid_request_dao.go
@@ -0,0 +1,83 @@
+package dao
+
+import (
+	"cidService/model/entity"
+	"context"
+
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+var (
+	CIDRequest = cidRequestDao{}
+)
+
+type cidRequestDao struct{}
+
+// Create 创建CID请求记录
+func (d *cidRequestDao) Create(ctx context.Context, request *entity.CidRequest) (id int64, err error) {
+	result, err := g.DB().Model("cid_requests").Insert(request)
+	if err != nil {
+		return 0, err
+	}
+	id, err = result.LastInsertId()
+	return
+}
+
+// GetHistory 获取CID请求历史
+func (d *cidRequestDao) GetHistory(ctx context.Context, userId int64, page, size int) (list []*entity.CidRequest, total int64, err error) {
+	model := g.DB().Model("cid_requests")
+
+	// 根据用户ID筛选
+	model = model.Where("user_id = ?", userId)
+
+	// 获取总数
+	count, err := model.Count()
+	if err != nil {
+		return
+	}
+	total = int64(count)
+
+	// 分页查询
+	offset := (page - 1) * size
+	err = model.Order("created_at DESC").
+		Offset(offset).
+		Limit(size).
+		Scan(&list)
+
+	return
+}
+
+// GetStatistics 获取统计信息
+func (d *cidRequestDao) GetStatistics(ctx context.Context, userId int64) (stats map[string]interface{}, err error) {
+	stats = make(map[string]interface{})
+
+	// 总请求数
+	totalRequests, err := g.DB().Model("cid_requests").
+		Where("user_id = ?", userId).
+		Count()
+	if err != nil {
+		return nil, err
+	}
+	stats["total_requests"] = totalRequests
+
+	// 成功请求数
+	successfulRequests, err := g.DB().Model("cid_requests").
+		Where("user_id = ? AND status = ?", userId, "completed").
+		Count()
+	if err != nil {
+		return nil, err
+	}
+	stats["successful_requests"] = successfulRequests
+
+	// 平均处理时间
+	avgProcessTime, err := g.DB().Model("cid_requests").
+		Fields("AVG(process_time)").
+		Where("user_id = ?", userId).
+		Value()
+	if err != nil {
+		return nil, err
+	}
+	stats["average_process_time"] = avgProcessTime
+
+	return
+}
--- a/dao/strategy_dao.go
+++ b/dao/strategy_dao.go
@@ -0,0 +1,100 @@
+package dao
+
+import (
+	"cidService/model/entity"
+	"context"
+
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+var (
+	Strategy = strategyDao{}
+)
+
+type strategyDao struct{}
+
+// GetByName 根据名称获取策略
+func (d *strategyDao) GetByName(ctx context.Context, name string) (strategy *entity.Strategy, err error) {
+	err = g.DB().Model("strategies").
+		Where("name = ?", name).
+		Scan(&strategy)
+	return
+}
+
+// GetByID 根据ID获取策略
+func (d *strategyDao) GetByID(ctx context.Context, id int64) (strategy *entity.Strategy, err error) {
+	err = g.DB().Model("strategies").
+		Where("id = ?", id).
+		Scan(&strategy)
+	return
+}
+
+// GetByTenantLevel 根据租户级别获取策略
+func (d *strategyDao) GetByTenantLevel(ctx context.Context, tenantLevel string) (strategy *entity.Strategy, err error) {
+	err = g.DB().Model("strategies").
+		Where("tenant_level = ? AND status = ?", tenantLevel, "active").
+		Order("priority DESC, created_at ASC").
+		Scan(&strategy)
+	return
+}
+
+// Create 创建策略
+func (d *strategyDao) Create(ctx context.Context, strategy *entity.Strategy) (id int64, err error) {
+	result, err := g.DB().Model("strategies").Insert(strategy)
+	if err != nil {
+		return 0, err
+	}
+	id, err = result.LastInsertId()
+	return
+}
+
+// Update 更新策略
+func (d *strategyDao) Update(ctx context.Context, strategy *entity.Strategy) (affected int64, err error) {
+	result, err := g.DB().Model("strategies").
+		Where("id = ?", strategy.Id).
+		Update(strategy)
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected()
+}
+
+// Delete 删除策略
+func (d *strategyDao) Delete(ctx context.Context, id int64) (affected int64, err error) {
+	result, err := g.DB().Model("strategies").
+		Where("id = ?", id).
+		Delete()
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected()
+}
+
+// GetList 获取策略列表
+func (d *strategyDao) GetList(ctx context.Context, page, size int, tenantLevel, status string) (list []*entity.Strategy, total int64, err error) {
+	model := g.DB().Model("strategies")
+
+	// 筛选条件
+	if tenantLevel != "" {
+		model = model.Where("tenant_level = ?", tenantLevel)
+	}
+	if status != "" {
+		model = model.Where("status = ?", status)
+	}
+
+	// 获取总数
+	count, err := model.Count()
+	if err != nil {
+		return
+	}
+	total = int64(count)
+
+	// 分页查询
+	offset := (page - 1) * size
+	err = model.Order("priority DESC, created_at DESC").
+		Offset(offset).
+		Limit(size).
+		Scan(&list)
+
+	return
+}