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
}