forked from all_user/stat_center
rootandClaude Opus 5
01d9f7bc86
- 新增 advitise v2: info/location/bind/click-stat/day-click-stat 模型与 对应的 service、api、router、proto 定义 - models/index.go 注册 5 个新模型的索引初始化 - 鉴权白名单: /rsdownload/ 改为 /download/,放行 advitise v2 的 get/click 接口 - statg/config: proxyUrl 增加 fileRes1 - common/genId: lid 生成逻辑补充 - 新增 91fans / huangguo 的 mobileconfig 描述文件 - 新增 deploy/nginx/cpcdownload.mfcg.me.conf 网关配置 Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
100 lines
2.9 KiB
Go
100 lines
2.9 KiB
Go
package advitiseClickStatMod
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"stat/common/log"
|
|
"stat/common/mongos"
|
|
"time"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
func Init(ctx context.Context) {
|
|
db = mongos.New(mongos.MgoManager, TN.AdvitiseClickStat)
|
|
indexes := []mongo.IndexModel{
|
|
{
|
|
Keys: bson.D{
|
|
{Key: FN.AdvitiseClickStat.AppId, Value: 1},
|
|
{Key: FN.AdvitiseClickStat.AdvitiseId, Value: 1},
|
|
{Key: FN.AdvitiseClickStat.LocationId, Value: 1},
|
|
{Key: FN.AdvitiseClickStat.StatDate, Value: -1},
|
|
},
|
|
},
|
|
{Keys: bson.D{
|
|
{Key: FN.AdvitiseClickStat.AdvitiseId, Value: 1},
|
|
{Key: FN.AdvitiseClickStat.StatDate, Value: -1},
|
|
}},
|
|
{Keys: bson.D{
|
|
{Key: FN.AdvitiseClickStat.StatDate, Value: -1},
|
|
}},
|
|
}
|
|
if _, err := db.CreateIndex(ctx, indexes); err != nil {
|
|
panic(fmt.Sprintf("(advitiseClickStatMod:Init) CreateIndex failed: %+v", err))
|
|
}
|
|
}
|
|
|
|
// InsertOrUpdate 按 (appId, advitiseId, locationId, statDate) upsert +1
|
|
// 对齐 media/biadsclickmod.InsertOrUpdateBIAdsClickModel
|
|
func InsertOrUpdate(ctx context.Context, appId int32, advitiseId int64, locationId int32, statDate time.Time) error {
|
|
filter := bson.M{
|
|
FN.AdvitiseClickStat.AppId: appId,
|
|
FN.AdvitiseClickStat.AdvitiseId: advitiseId,
|
|
FN.AdvitiseClickStat.LocationId: locationId,
|
|
FN.AdvitiseClickStat.StatDate: statDate,
|
|
}
|
|
update := bson.M{
|
|
"$inc": bson.M{FN.AdvitiseClickStat.Clicks: 1},
|
|
"$setOnInsert": bson.M{
|
|
FN.AdvitiseClickStat.CreatedAt: time.Now(),
|
|
},
|
|
}
|
|
if _, err := db.UpsertOneEx(ctx, filter, update); err != nil {
|
|
log.Warn("advitiseClickStatMod InsertOrUpdate upsert failed",
|
|
log.E(err), log.Any("filter", filter))
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func FindMany(ctx context.Context, filter bson.M, opts ...*options.FindOptions) (AdvitiseClickStatList, error) {
|
|
list := make(AdvitiseClickStatList, 0)
|
|
if err := db.Find(ctx, &list, filter, opts...); err != nil {
|
|
log.Warn("advitiseClickStatMod Find failed", log.E(err), log.Any("filter", filter))
|
|
return nil, err
|
|
}
|
|
return list, nil
|
|
}
|
|
|
|
func Count(ctx context.Context, filter bson.M) (int64, error) {
|
|
c, err := db.Count(ctx, filter)
|
|
if err != nil {
|
|
log.Warn("advitiseClickStatMod Count failed", log.E(err), log.Any("filter", filter))
|
|
return 0, err
|
|
}
|
|
return c, nil
|
|
}
|
|
|
|
// Aggregate 通用聚合(留给上层写 pipeline)
|
|
func Aggregate(ctx context.Context, bind interface{}, pipeline []bson.M) error {
|
|
return db.Aggregate(ctx, bind, pipeline)
|
|
}
|
|
|
|
// GetAppClicks 按 appId + statDate 汇总点击(对齐 media/biadsclickmod.GetAppClicks)
|
|
func GetAppClicks(ctx context.Context, appId int32, statDate time.Time) (int64, error) {
|
|
list, err := FindMany(ctx, bson.M{
|
|
FN.AdvitiseClickStat.AppId: appId,
|
|
FN.AdvitiseClickStat.StatDate: statDate,
|
|
})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
total := int64(0)
|
|
for _, i := range list {
|
|
total += i.Clicks
|
|
}
|
|
return total, nil
|
|
}
|