wechat-api/crontask/compute_historical_credit.go

package crontask

import (
	"github.com/zeromicro/go-zero/core/logx"
	"time"
	"wechat-api/ent"
	"wechat-api/ent/creditbalance"
	"wechat-api/ent/creditusage"
	"wechat-api/ent/usagedetail"
	"wechat-api/ent/usagetotal"
)

func (l *CronTask) computeHistoricalCredit() {
	orgBots := []OrgBot{}
	err := l.svcCtx.DB.UsageDetail.Query().
		GroupBy(usagedetail.FieldOrganizationID, usagedetail.FieldBotID).
		Aggregate().
		Scan(l.ctx, &orgBots)

	if err != nil {
		l.Errorf("group usage_detail error: %v", err)
		return
	}

	wxbotsSet := make(map[uint64][]string)
	for _, ob := range orgBots {
		wxbotsSet[ob.OrganizationID] = append(wxbotsSet[ob.OrganizationID], ob.BotID)
	}
	logx.Info("wxbotsSet: ", wxbotsSet)

	/*
		计算本小时的数据
		1. 查询出上小时里所有 usagedetail 内容
		2. 挨个遍历他的 bot_id ,再查询他的 bot_id 相关的参数
		3. 遍历的时候可能有重复，所以要先检查是否生成了数据，如果有就忽略，没有再生成
		----------------------------------------------------------------------------------------------------------
	*/

	// 获取当前时间
	//now := time.Now()
	//start := time.Date(2024, 12, 25, 0, 0, 0, 0, time.Local)
	start := time.Date(2025, 4, 1, 0, 0, 0, 0, time.Local)
	end := time.Date(2025, 4, 24, 23, 0, 0, 0, time.Local)
	//start := time.Date(2025, 3, 18, 0, 0, 0, 0, time.Local)
	//end := time.Date(2025, 3, 19, 23, 0, 0, 0, time.Local)

	balanceOrgSet := make(map[uint64]float64)
	balanceBotSet := make(map[string]float64)

	for orgID, wxinfos := range wxbotsSet {
		if _, ok := balanceOrgSet[orgID]; !ok {
			balanceOrgSet[orgID] = 0
		}
		for _, wxinfo := range wxinfos {
			if _, ok := balanceBotSet[wxinfo]; !ok {
				balanceBotSet[wxinfo] = 0
			}
			l.Logger.Infof("开始计算小时数据:%d\n", start)

			// 计算积分消耗
			usageDetails, _ := l.svcCtx.DB.UsageDetail.Query().Where(
				usagedetail.BotID(wxinfo),
				usagedetail.CreatedAtGTE(start),
				usagedetail.CreatedAtLT(end),
			).Order(ent.Desc(usagedetail.FieldCreatedAt)).All(l.ctx)

			for _, usageDetail := range usageDetails {
				balanceBotSet[wxinfo] += usageDetail.Credits
				// 更改积分明细表
				hourDataCount, _ := l.svcCtx.DB.CreditUsage.Query().Where(
					creditusage.TableEQ("usage_detail"),
					creditusage.NidEQ(usageDetail.ID),
				).Count(l.ctx)
				if hourDataCount == 0 {
					balanceOrgSet[orgID] += usageDetail.Credits
					_, err = l.svcCtx.DB.CreditUsage.Create().
						SetNotNilNumber(&usageDetail.Credits).
						SetNtype(1).
						SetTable("usage_detail").
						SetOrganizationID(orgID).
						SetNid(usageDetail.ID).
						Save(l.ctx)
					l.Errorf("save hour data error:%v \n", err)
				}
			}
		}
	}

	for orgID, balance := range balanceOrgSet {
		creditBalance, _ := l.svcCtx.DB.CreditBalance.Query().Where(
			creditbalance.OrganizationIDEQ(orgID),
		).First(l.ctx)
		if creditBalance == nil {
			_, err = l.svcCtx.DB.CreditBalance.Create().
				SetBalance(balance).
				SetOrganizationID(orgID).
				Save(l.ctx)
			l.Errorf("save hour data error:%v \n", err)
		} else {
			b := creditBalance.Balance - balance
			_, err = l.svcCtx.DB.CreditBalance.Update().
				Where(creditbalance.OrganizationIDEQ(orgID)).
				SetBalance(b).
				Save(l.ctx)
			l.Errorf("save hour data error:%v \n", err)
		}
	}

	for botID, balance := range balanceBotSet {
		_, err = l.svcCtx.DB.UsageTotal.Update().
			Where(usagetotal.BotIDEQ(botID)).
			SetCredits(balance).
			Save(l.ctx)
		l.Errorf("save hour data error:%v \n", err)
	}
	return
}