Files
rootandClaude Opus 5 8679200f41 Initial commit
Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-09-15 13:57:10 +08:00

64 lines
1.6 KiB
Go

package useradverstatmod
import (
"91porn-server/common/log"
"91porn-server/models/commod"
"fmt"
"time"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo/options"
)
type DailyRetentionSummary struct {
ID DailyRetentionSummaryID `bson:"_id"`
Retained int64 `bson:"retained"`
AdClick int64 `bson:"adClick"`
TotalClick int64 `bson:"totalClick"`
PayCount int64 `bson:"payCount"`
PayTotal int64 `bson:"payTotal"`
}
type DailyRetentionSummaryID struct {
RegDay time.Time `bson:"regDay"`
AdGroup commod.AdGroup `bson:"adGroup"`
}
func AggregateDailyRetention(date, regDayStart, regDayEnd time.Time) (out []*DailyRetentionSummary, err error) {
pipeline := []bson.M{
{
"$match": bson.M{
"date": date,
"regDay": bson.M{
"$gte": regDayStart,
"$lte": regDayEnd,
},
},
},
{
"$group": bson.M{
"_id": bson.M{
"regDay": "$regDay",
"adGroup": "$adGroup",
},
"retained": bson.M{"$sum": 1},
"adClick": bson.M{"$sum": "$adClick"},
"totalClick": bson.M{"$sum": "$totalClick"},
"payCount": bson.M{"$sum": "$payCount"},
"payTotal": bson.M{"$sum": "$payTotal"},
},
},
}
opts := options.Aggregate().SetAllowDiskUse(true)
if err = coll(nil).Aggregate(&out, pipeline, opts); err != nil {
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "AggregateDailyRetention", table, "Aggregate", err),
log.Any("date", date),
log.Any("regDayStart", regDayStart),
log.Any("regDayEnd", regDayEnd),
)
return nil, err
}
return
}