package kwrank

import (
	"fmt"
	"time"

	"91porn-server/common/db"
	"91porn-server/common/log"
	"91porn-server/common/timeutil/timerange"
	"91porn-server/models/s/kwrankmod"
	"91porn-server/models/s/statrecordmod"
	"91porn-server/skd/skdg"
)

// RunStat
func RunStat(defaultStatDate time.Time) {
	log.Info("keyword rank stat running...")
	defer log.Info("keyword rank stat finished...")
	now := time.Now()
	sumDate := timerange.LocDayRange(now).Head //获取零日点
	//装配基础统计数据
	setList, err := GetStatSetDocList(sumDate)
	if err != nil {
		log.Error("kwrank RunStat fillStatSetDoc faild!", log.E(err))
		return
	}
	//提交数据
	log.Info("keyword rank stat committing...", log.Any("sumDate", sumDate), log.Any("location", sumDate.Location().String()))
	recentMinute := timerange.RecentMinute(now, statrecordmod.FiveMinuteScale) //对齐本次统计时间 Minute % frequency  == 0
	if err = commit(sumDate, recentMinute, setList); err != nil {
		log.Error("kwrank RunStat commit faild!", log.E(err))
		return
	}
	log.Info("keyword rank stat committed.", log.Any("sumDate", sumDate), log.Any("location", sumDate.Location().String()))
}

// commit 提交统计数据
func commit(sumDate time.Time, recordTime time.Time, setDocList []RankSetDoc) error {
	opt := (&db.TransOpts{}).SetReEntry(10)
	return skdg.StatDB.Trans(func(trans *db.MongoTool) (err error) {
		if len(setDocList) != 0 {
			if err = kwrankmod.ChangeStatTrans(trans, sumDate, recordTime, setDocList); err != nil {
				return fmt.Errorf("kwrank commit ChangeStatTrans faild!, err:%+v\n", err)
			}
		}
		//提交本次修改记录
		if err = statrecordmod.UpsertOneTrans(trans, statrecordmod.KeywordRankStatJob, recordTime); err != nil {
			return fmt.Errorf("kwrank commit UpsertOneTrans faild!, err:%+v\n", err)
		}
		return
	}, opt)
}