Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>

skd/job/stat/kwrank/method.go

@@ -0,0 +1,29 @@
+package kwrank
+
+import (
+	"time"
+
+	"91porn-server/models/s/kwrankmod"
+	"91porn-server/models/s/kwstatmod"
+)
+
+type RankSetDoc = kwrankmod.RankSetDoc
+
+func GetStatSetDocList(sumDate time.Time) ([]RankSetDoc, error) {
+	keywordCountMap, err := kwstatmod.KeywordCountMapBySumDate(sumDate)
+	if err != nil {
+		return nil, err
+	}
+	rankDocList := make([]RankSetDoc, len(keywordCountMap))
+	i := 0
+	for w, c := range keywordCountMap {
+		word := w
+		count := c
+		rankDocList[i] = RankSetDoc{
+			Word:  &word,
+			Count: &count,
+		}
+		i++
+	}
+	return rankDocList, nil
+}

skd/job/stat/kwrank/stat.go

@@ -0,0 +1,52 @@
+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)
+}