Initial commit

2026-05-20 16:13:58 +08:00
commit ceffbe382e
500 changed files with 157469 additions and 0 deletions
@@ -0,0 +1,107 @@
+package statdailyhour
+
+import (
+	"context"
+	"stat/common/log"
+	"stat/common/statutils"
+	"stat/common/utils"
+	"stat/models/appInfoMod"
+	statrecordmod "stat/models/statrecord"
+	"time"
+)
+
+func RunStat(defaultStatDate time.Time) error {
+	log.Info("daily stat running...")
+	ctx := context.Background()
+	recordMap := defaultTimeMap(defaultStatDate)
+
+	// 1.获取统计项上次统计的时候(每天执行)
+	recoredResult, err := statrecordmod.LastRecordTimeMap(ctx, statrecordmod.DailyHourStatDayJob) //第一次从默认记录时间开始统计
+	if err != nil {
+		log.Error("Daily 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("daily RunStat GetAppIds failed!", log.E(err))
+		return err
+	}
+
+	now := time.Now().Add(time.Minute * -5)
+	statFrq := 1 * time.Hour                   // 统计频率时间
+	curStatTime := utils.RecentMinute(now, 60) // 按60分钟对齐
+
+	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("daily statByTimeRange failed!", log.E(err), log.Any("item", item))
+				return err
+			}
+			_lastStatAt = end
+		}
+		recordMap[item] = _lastStatAt
+	}
+	return nil
+}
+
+// 实时统计每天的情况
+func RunStatToday() {
+	ctx := context.Background()
+	start := utils.BeginOfToday()
+	end := utils.EndOfToday()
+	log.Info("daily today stat running...")
+
+	appIds, err := appInfoMod.GetAppIds(ctx)
+	if err != nil {
+		log.Error("daily today RunStat GetAppIds failed!", log.E(err))
+		return
+	}
+
+	for item, statMethod := range statMethodMap {
+		// 调用统计方法，获取数据
+		err := statMethod(ctx, start, end, appIds)
+		if err != nil {
+			log.Error("Daily today stat statByTimeRange statMethod failed!", log.Any("statItem", item), log.E(err))
+		}
+	}
+
+}
+
+// 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
+	}
+
+	// 调用统计方法，获取数据
+	err = statutils.ServerMon.RunWithName(func() error {
+		return statMethod(ctx, start, end, appIds)
+	}, statutils.GetFuncName(statMethod))
+
+	if err != nil {
+		log.Error("Daily stat statByTimeRange statMethod failed!", log.Any("statItem", item), log.E(err))
+		return err
+	}
+
+	// 3.更新记录
+	if err := statrecordmod.UpsertItemRecord(ctx, statrecordmod.DailyHourStatDayJob, item, end); err != nil {
+		log.Error("Daily stat commit UpsertItemRecord failed!", log.E(err))
+		return err
+	}
+	return nil
+}