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
}