forked from all_user/stat_center
90 lines
2.5 KiB
Go
90 lines
2.5 KiB
Go
package blocstatdistrict
|
|
|
|
import (
|
|
"context"
|
|
"stat/common/log"
|
|
"stat/common/statutils"
|
|
"stat/common/utils"
|
|
"stat/models/appInfoMod"
|
|
statrecordmod "stat/models/statrecord"
|
|
"time"
|
|
)
|
|
|
|
/*
|
|
* 主要统计任务:
|
|
* 1、渠道每天的：直推访问数、裂变访问数、次日留存率、3日留存率、7日留存率。
|
|
*/
|
|
func RunStat(defaultStatDate time.Time) error {
|
|
log.Info("statdistrict stat running...")
|
|
ctx := context.Background()
|
|
recordMap := defaultTimeMap(defaultStatDate)
|
|
|
|
// 1.获取统计项上次统计的时候(每天执行)
|
|
recoredResult, err := statrecordmod.LastRecordTimeMap(ctx, statrecordmod.BlocDiscStatJob) //第一次从默认记录时间开始统计
|
|
if err != nil {
|
|
log.Error("statdistrict RunStat lastRecordTime failed!", log.E(err))
|
|
return err
|
|
}
|
|
for _, record := range recoredResult {
|
|
recordMap[record.Item] = record.RecordTime
|
|
}
|
|
|
|
appIds, err := appInfoMod.GetAppIds(ctx)
|
|
if err != nil {
|
|
log.Error("statdistrict RunStat GetAppIds failed!", log.E(err))
|
|
return err
|
|
}
|
|
|
|
statFrq := 24 * time.Hour // 统计频率
|
|
curStatTime := utils.BeginOfToday() // 本次统计时间
|
|
|
|
for item, lastStatAt := range recordMap {
|
|
_lastStatAt := lastStatAt
|
|
for curStatTime.After(_lastStatAt) {
|
|
end := _lastStatAt.Add(statFrq)
|
|
|
|
// 超过本次统计结束时间，则结束
|
|
if end.After(curStatTime) {
|
|
end = curStatTime
|
|
}
|
|
// 根据时间段统计数据
|
|
err := runStatMethod(ctx, item, _lastStatAt, end, appIds)
|
|
if err != nil {
|
|
log.Error("statdistrict statByTimeRange failed!", log.E(err), log.Any("item", item))
|
|
return err
|
|
}
|
|
_lastStatAt = end
|
|
//fmt.Println(_lastStatAt)
|
|
}
|
|
recordMap[item] = _lastStatAt
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// runStatMethod 调用统计项方法
|
|
func runStatMethod(ctx context.Context, item string, start, end time.Time, appIds []int32) (err error) {
|
|
// 1.获取统计方法
|
|
statMethod, ok := statMethodMap[item]
|
|
if !ok {
|
|
return nil
|
|
}
|
|
//fmt.Println(item)
|
|
|
|
// 调用统计方法，获取数据
|
|
err = statutils.ServerMon.RunWithName(func() error {
|
|
return statMethod(ctx, start, end, appIds)
|
|
}, statutils.GetFuncName(statMethod))
|
|
|
|
if err != nil {
|
|
log.Error("statdistrict stat statByTimeRange statMethod failed!", log.Any("statItem", item), log.E(err))
|
|
return err
|
|
}
|
|
|
|
// 3.更新记录
|
|
if err := statrecordmod.UpsertItemRecord(ctx, statrecordmod.BlocDiscStatJob, item, end); err != nil {
|
|
log.Error("statdistrict stat commit UpsertItemRecord failed!", log.E(err))
|
|
return err
|
|
}
|
|
return nil
|
|
}
|