80 lines
2.4 KiB
Go
80 lines
2.4 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
|
|
"cid/model/entity"
|
|
"gitee.com/red-future---jilin-g/common/mongo"
|
|
"go.mongodb.org/mongo-driver/v2/bson"
|
|
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
|
)
|
|
|
|
var AdSource = &adSourceDao{}
|
|
|
|
type adSourceDao struct{}
|
|
|
|
// GetByName 根据名称获取广告源
|
|
func (d *adSourceDao) GetByName(ctx context.Context, name string) (adSource *entity.AdSource, err error) {
|
|
err = mongo.FindOne(ctx, bson.M{"name": name}, &adSource, "ad_sources")
|
|
return
|
|
}
|
|
|
|
// GetAvailableSources 获取可用的广告源
|
|
func (d *adSourceDao) GetAvailableSources(ctx context.Context) (list []*entity.AdSource, err error) {
|
|
err = mongo.Find(ctx, bson.M{"status": "active"}, &list, "ad_sources",
|
|
options.Find().SetSort(bson.M{"priority": -1, "createdAt": 1}))
|
|
return
|
|
}
|
|
|
|
// GetSourcesByProvider 根据提供商获取广告源
|
|
func (d *adSourceDao) GetSourcesByProvider(ctx context.Context, provider string) (list []*entity.AdSource, err error) {
|
|
err = mongo.Find(ctx, bson.M{"provider": provider, "status": "active"}, &list, "ad_sources",
|
|
options.Find().SetSort(bson.M{"priority": -1}))
|
|
return
|
|
}
|
|
|
|
// Create 创建广告源
|
|
func (d *adSourceDao) Create(ctx context.Context, adSource *entity.AdSource) (id string, err error) {
|
|
ids, err := mongo.Insert(ctx, []interface{}{adSource}, "ad_sources")
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if len(ids) > 0 {
|
|
id = ids[0].(string)
|
|
}
|
|
return
|
|
}
|
|
|
|
// Update 更新广告源
|
|
func (d *adSourceDao) Update(ctx context.Context, adSource *entity.AdSource) (affected int64, err error) {
|
|
result, err := mongo.Update(ctx, bson.M{"_id": adSource.Id}, bson.M{"$set": adSource}, "ad_sources")
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return result.ModifiedCount, nil
|
|
}
|
|
|
|
// Delete 删除广告源
|
|
func (d *adSourceDao) Delete(ctx context.Context, id string) (affected int64, err error) {
|
|
count, err := mongo.Delete(ctx, bson.M{"_id": id}, "ad_sources")
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
// GetByID 根据ID获取广告源
|
|
func (d *adSourceDao) GetByID(ctx context.Context, id string) (adSource *entity.AdSource, err error) {
|
|
err = mongo.FindOne(ctx, bson.M{"_id": id}, &adSource, "ad_sources")
|
|
return
|
|
}
|
|
|
|
// UpdateFields 更新广告源部分字段
|
|
func (d *adSourceDao) UpdateFields(ctx context.Context, id string, data *entity.AdSource) (affected int64, err error) {
|
|
result, err := mongo.Update(ctx, bson.M{"_id": id}, bson.M{"$set": data}, "ad_sources")
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return result.ModifiedCount, nil
|
|
}
|