Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-09-15 13:57:10 +08:00
commit 8679200f41
1897 changed files with 257900 additions and 0 deletions
@@ -0,0 +1,139 @@
+package withorder
+
+import (
+	"time"
+
+	"91porn-server/common/timeutil/timerange"
+	"91porn-server/models/s/statordermod"
+	"91porn-server/models/v/wdordmod"
+)
+
+type WithDrawOrderStateDoc = statordermod.WithDrawOrderStateDoc
+type WitOrderStateMap = statordermod.WitOrderStateMap
+
+func fillWitOrderStatDocMap(sumDate time.Time, timeRange timerange.TimeRange, docMap *WitOrderStateMap) error {
+	//订单数 OrderCount
+	if err := fillWitOrderCount(timeRange, docMap); err != nil {
+		return err
+	}
+	//完成订单数 CompleteOrderCount
+	if err := fillWitCompleteOrderCount(timeRange, docMap); err != nil {
+		return err
+	}
+	//订单金额 OrderAmount
+	if err := fillWitOrderAmount(timeRange, docMap); err != nil {
+		return err
+	}
+	//实际兑换金额 CompleteOrderAmount
+	if err := fillWitCompleteOrderPayAmount(timeRange, docMap); err != nil {
+		return err
+	}
+	//三方已出款订单数 fillWitPaiedOrderCount
+	if err := fillWitPaiedOrderCount(timeRange, docMap); err != nil {
+		return err
+	}
+	//三方已出款金额 fillWitPaiedMoney
+	return fillWitPaiedMoney(timeRange, docMap)
+}
+
+// fillWitOrderCount fillWitOrderCount
+func fillWitOrderCount(timeRange timerange.TimeRange, docMap *WitOrderStateMap) error {
+	orderCountMap, err := wdordmod.ChannelOrderCountMap(timeRange.Head, timeRange.Tail)
+	if err != nil {
+		return err
+	}
+	docM := *docMap
+	for channel, v := range orderCountMap {
+		if _, existed := docM[channel]; !existed {
+			docM[channel] = &WithDrawOrderStateDoc{}
+		}
+		val := v
+		docM[channel].OrderCount = &val
+	}
+	return nil
+}
+
+// fillWitCompleteOrderCount fillWitCompleteOrderCount
+func fillWitCompleteOrderCount(timeRange timerange.TimeRange, docMap *WitOrderStateMap) error {
+	completeOrderCountMap, err := wdordmod.ChannelCompleteOrderCountMap(timeRange.Head, timeRange.Tail)
+	if err != nil {
+		return err
+	}
+	docM := *docMap
+	for channel, v := range completeOrderCountMap {
+		if _, existed := docM[channel]; !existed {
+			docM[channel] = &WithDrawOrderStateDoc{}
+		}
+		val := v
+		docM[channel].CompleteOrderCount = &val
+	}
+	return nil
+}
+
+// fillWitOrderAmount fillWitOrderAmount
+func fillWitOrderAmount(timeRange timerange.TimeRange, docMap *WitOrderStateMap) error {
+	orderAmountMap, err := wdordmod.ChannelOrderMoneyMap(timeRange.Head, timeRange.Tail)
+	if err != nil {
+		return err
+	}
+	docM := *docMap
+	for channel, v := range orderAmountMap {
+		if _, existed := docM[channel]; !existed {
+			docM[channel] = &WithDrawOrderStateDoc{}
+		}
+		val := v
+		docM[channel].OrderAmount = &val
+	}
+	return nil
+}
+
+// fillWitCompleteOrderPayAmount fillWitCompleteOrderPayAmount
+func fillWitCompleteOrderPayAmount(timeRange timerange.TimeRange, docMap *WitOrderStateMap) error {
+	completeOrderAmountMap, err := wdordmod.ChannelCompleteOrderPayMoneyMap(timeRange.Head, timeRange.Tail)
+	if err != nil {
+		return err
+	}
+	docM := *docMap
+	for channel, v := range completeOrderAmountMap {
+		if _, existed := docM[channel]; !existed {
+			docM[channel] = &WithDrawOrderStateDoc{}
+		}
+		val := v
+		docM[channel].CompleteOrderAmount = &val
+	}
+	return nil
+}
+
+// fillWitPaiedOrderCount fillWitPaiedOrderCount
+func fillWitPaiedOrderCount(timeRange timerange.TimeRange, docMap *WitOrderStateMap) error {
+	receivedOrderCountMap, err := wdordmod.ChannelReceivedOrderCountMap(timeRange.Head, timeRange.Tail)
+	if err != nil {
+		return err
+	}
+	docM := *docMap
+	for channel, v := range receivedOrderCountMap {
+		if _, existed := docM[channel]; !existed {
+			docM[channel] = &WithDrawOrderStateDoc{}
+		}
+		val := v
+		docM[channel].PaidOrderCount = &val
+	}
+	return nil
+}
+
+// fillWitPaiedMoney fillWitPaiedMoney
+func fillWitPaiedMoney(timeRange timerange.TimeRange, docMap *WitOrderStateMap) error {
+	receivedMoneyMap, err := wdordmod.ChannelReceivedMoneyMap(timeRange.Head, timeRange.Tail)
+	if err != nil {
+		return err
+	}
+	docM := *docMap
+	for channel, v := range receivedMoneyMap {
+		if _, existed := docM[channel]; !existed {
+			docM[channel] = &WithDrawOrderStateDoc{}
+		}
+		val := v
+		docM[channel].PaidAmount = &val
+	}
+	return nil
+}
@@ -0,0 +1,97 @@
+package withorder
+
+import (
+	"fmt"
+	"time"
+
+	"91porn-server/common/db"
+	"91porn-server/common/log"
+	"91porn-server/common/timeutil/timerange"
+	"91porn-server/models/s/statordermod"
+	"91porn-server/models/s/statrecordmod"
+	"91porn-server/skd/skdg"
+)
+
+func RunStat(defaultStatDate time.Time) {
+	log.Info("withorder stat running...")
+	defer log.Info("withorder stat finished...")
+	now := time.Now()
+	//从历史提交记录中获取上次记录时间
+	lastStatAt, err := statrecordmod.LastRecordTime(statrecordmod.WithdrawOrderStatJob, defaultStatDate) //第一次从默认记录时间开始统计
+	if err != nil {
+		log.Error("withorder RunStat lastRecordTime faild!", log.E(err))
+		return
+	}
+	//过去几天的数据按一日为单位统计，帮助数据恢复
+	dayHead := timerange.LocDayRange(now).Head
+	for dayHead.After(lastStatAt) {
+		end := lastStatAt.AddDate(0, 0, 1)
+		//防止时间超出范围
+		if end.After(dayHead) {
+			end = dayHead
+		}
+		subTimeRange := timerange.TimeRange{ //连续子切片
+			Head: lastStatAt,
+			Tail: end,
+		}
+		if err = statByTimeRange(subTimeRange); err != nil {
+			log.Error("withorder RunStat statByTimeRange 1 faild!", log.E(err))
+			return
+		}
+		lastStatAt = end
+	}
+	//当天的数据按照5分钟统计
+	recentMinute := timerange.RecentMinute(now, statrecordmod.FiveMinuteScale) //对齐本次统计时间 Minute % frequency  == 0
+	//拆分时间区间并依次提交,控制数据库读写压力
+	for recentMinute.After(lastStatAt) {
+		end := lastStatAt.Add(statrecordmod.FiveMinuteScale * time.Minute)
+		//防止时间超出范围
+		if end.After(recentMinute) {
+			end = recentMinute
+		}
+		subTimeRange := timerange.TimeRange{ //连续子切片
+			Head: lastStatAt,
+			Tail: end,
+		}
+		if err = statByTimeRange(subTimeRange); err != nil {
+			log.Error("withorder RunStat statByTimeRange 2 faild!", log.E(err))
+			return
+		}
+		lastStatAt = end
+	}
+}
+
+func statByTimeRange(timeRange timerange.TimeRange) error {
+	sumDate := timerange.LocDayRange(timeRange.Head).Head //获取此时的零日点
+	//装配渠道统计数据
+	witOrderStateMap := make(WitOrderStateMap)
+	if err := fillWitOrderStatDocMap(sumDate, timeRange, &witOrderStateMap); err != nil {
+		log.Error("withorder RunStat fillWitOrderStatDocMap faild!", log.E(err))
+		return err
+	}
+	//提交渠道统计数据
+	log.Info("withorderStat withdraw committing...", log.Any("sumDate", sumDate), log.Any("location", sumDate.Location().String()))
+	if err := commitWit(sumDate, timeRange.Tail, witOrderStateMap); err != nil {
+		log.Error("withorder RunStat commitWit faild!", log.E(err))
+		return err
+	}
+	log.Info("withorderStat withdraw committed.", log.Any("sumDate", sumDate), log.Any("location", sumDate.Location().String()))
+	return nil
+}
+
+// commit 提交订单统计数据
+func commitWit(sumDate time.Time, recordTime time.Time, stateMap WitOrderStateMap) error {
+	opt := (&db.TransOpts{}).SetReEntry(10)
+	return skdg.StatDB.Trans(func(trans *db.MongoTool) error {
+		if len(stateMap) != 0 {
+			if err := statordermod.ChangeWithDrawStatTrans(trans, sumDate, recordTime, stateMap); err != nil {
+				return fmt.Errorf("stat commitWit ChangeWithDrawStatTrans faild!, err:%+v\n", err)
+			}
+		}
+		//提交本次修改记录
+		if err := statrecordmod.UpsertOneTrans(trans, statrecordmod.WithdrawOrderStatJob, recordTime); err != nil {
+			return fmt.Errorf("stat commitWit UpsertOneTrans faild!, err:%+v\n", err)
+		}
+		return nil
+	}, opt)
+}