92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
|
|
"cidservice/model/entity"
|
|
"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()
|
|
}
|