|
|
@@ -0,0 +1,230 @@
|
|
|
+package crontask
|
|
|
+
|
|
|
+//
|
|
|
+//import (
|
|
|
+// "fmt"
|
|
|
+// "github.com/zeromicro/go-zero/core/logx"
|
|
|
+// "strconv"
|
|
|
+// "time"
|
|
|
+// "wechat-api/ent"
|
|
|
+// "wechat-api/ent/usagedetail"
|
|
|
+// "wechat-api/ent/usagestatisticday"
|
|
|
+// "wechat-api/ent/usagestatistichour"
|
|
|
+//)
|
|
|
+//
|
|
|
+//type OrgBot struct {
|
|
|
+// OrganizationID uint64 `json:"organization_id"`
|
|
|
+// BotID string `json:"bot_id"`
|
|
|
+//}
|
|
|
+//
|
|
|
+//func (l *CronTask) computeHistoricalStatistic() {
|
|
|
+// // 获取所有机器人信息
|
|
|
+// //wxbots, err := l.svcCtx.DB.Wx.Query().Select(wx.FieldWxid, wx.FieldID, wx.FieldOrganizationID).All(l.ctx)
|
|
|
+// //if err != nil {
|
|
|
+// // l.Errorf("fetch wxids error:%v\n", err)
|
|
|
+// // return
|
|
|
+// //}
|
|
|
+// //
|
|
|
+// //wxbotsSet := make(map[uint64][]*ent.Wx)
|
|
|
+// //for _, bot := range wxbots {
|
|
|
+// // if !strings.HasPrefix(bot.Wxid, "temp-") {
|
|
|
+// // wxbotsSet[bot.OrganizationID] = append(wxbotsSet[bot.OrganizationID], bot)
|
|
|
+// // }
|
|
|
+// //}
|
|
|
+// 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(2025, 5, 5, 0, 0, 0, 0, time.Local)
|
|
|
+// end := time.Date(2025, 5, 8, 0, 0, 0, 0, time.Local)
|
|
|
+//
|
|
|
+// //start := time.Date(2024, 11, 24, 0, 0, 0, 0, time.Local)
|
|
|
+// //end := time.Date(2025, 5, 9, 0, 0, 0, 0, time.Local)
|
|
|
+//
|
|
|
+// for now := start; !now.After(end); now = now.Add(time.Hour) {
|
|
|
+// fmt.Println(now.Format("2006-01-02 15:00:00"))
|
|
|
+//
|
|
|
+// // 获取本小时的第一分钟
|
|
|
+// currentHour := time.Date(now.Year(), now.Month(), now.Day(), now.Hour(), 0, 0, 0, now.Location()).Add(-8 * time.Hour)
|
|
|
+// currentHourInt, _ := strconv.Atoi(currentHour.Format("2006010215"))
|
|
|
+// currentHourLocation := time.Date(now.Year(), now.Month(), now.Day(), now.Hour(), 0, 0, 0, now.Location())
|
|
|
+// currentHourIntLocation, _ := strconv.Atoi(currentHourLocation.Format("2006010215"))
|
|
|
+//
|
|
|
+// // 上一个小时的起始时间
|
|
|
+// lastHour := currentHour.Add(-time.Hour * 1)
|
|
|
+// lastHourLocation := currentHourLocation.Add(-time.Hour * 1)
|
|
|
+// lastHourInt, _ := strconv.Atoi(lastHour.Format("2006010215"))
|
|
|
+// lastHourIntLocation, _ := strconv.Atoi(lastHourLocation.Format("2006010215"))
|
|
|
+//
|
|
|
+// l.Logger.Infof("currentHour:%d\n", currentHour)
|
|
|
+// l.Logger.Infof("currentHourInt:%d\n", currentHourInt)
|
|
|
+// l.Logger.Infof("currentHourLocation:%d\n", currentHourLocation)
|
|
|
+// l.Logger.Infof("currentHourIntLocation:%d\n", currentHourIntLocation)
|
|
|
+// l.Logger.Infof("lastHour:%d\n", lastHour)
|
|
|
+// l.Logger.Infof("lastHourLocation:%d\n", lastHourLocation)
|
|
|
+// l.Logger.Infof("lastHourInt:%d\n", lastHourInt)
|
|
|
+// l.Logger.Infof("lastHourIntLocation:%d\n", lastHourIntLocation)
|
|
|
+//
|
|
|
+// var allHourConsumeCoinFloat float64
|
|
|
+// for orgID, wxinfos := range wxbotsSet {
|
|
|
+// var orgHourConsumeCoinFloat float64
|
|
|
+// for _, wxinfo := range wxinfos {
|
|
|
+// l.Logger.Infof("开始计算小时数据:%d\n", lastHourInt)
|
|
|
+//
|
|
|
+// // 先判断该账号是否已经统计了小时数据，如果已经统计了，就不需要再统计了
|
|
|
+// var consumeCoinFloat float64
|
|
|
+//
|
|
|
+// // 计算积分消耗
|
|
|
+// consumeCoinFloat, _ = l.svcCtx.DB.UsageDetail.Query().Where(
|
|
|
+// usagedetail.BotID(wxinfo),
|
|
|
+// usagedetail.OrganizationIDEQ(orgID),
|
|
|
+// usagedetail.CreatedAtGTE(lastHour),
|
|
|
+// usagedetail.CreatedAtLT(currentHour),
|
|
|
+// ).Aggregate(ent.Sum("credits")).Float64(l.ctx)
|
|
|
+//
|
|
|
+// ush, _ := l.svcCtx.DB.UsageStatisticHour.Query().Where(usagestatistichour.AddtimeEQ(uint64(lastHourIntLocation)), usagestatistichour.BotIDEQ(wxinfo), usagestatistichour.OrganizationIDEQ(orgID)).Only(l.ctx)
|
|
|
+// if ush == nil {
|
|
|
+// continue
|
|
|
+// }
|
|
|
+// allHourConsumeCoinFloat += consumeCoinFloat
|
|
|
+// orgHourConsumeCoinFloat += consumeCoinFloat
|
|
|
+//
|
|
|
+// logx.Info("hour consumeCoinFloat: ", consumeCoinFloat)
|
|
|
+// logx.Info("hour allHourConsumeCoinFloat: ", allHourConsumeCoinFloat)
|
|
|
+// logx.Info("hour orgHourConsumeCoinFloat: ", orgHourConsumeCoinFloat)
|
|
|
+//
|
|
|
+// _, err := l.svcCtx.DB.UsageStatisticHour.Update().
|
|
|
+// Where(usagestatistichour.AddtimeEQ(uint64(lastHourIntLocation)), usagestatistichour.BotIDEQ(wxinfo), usagestatistichour.OrganizationIDEQ(orgID)).
|
|
|
+// SetConsumeCoin(consumeCoinFloat).
|
|
|
+// Save(l.ctx)
|
|
|
+// if err != nil {
|
|
|
+// l.Errorf("save hour data error:%v \n", err)
|
|
|
+// }
|
|
|
+// }
|
|
|
+//
|
|
|
+// _, err = l.svcCtx.DB.UsageStatisticHour.Update().
|
|
|
+// Where(usagestatistichour.AddtimeEQ(uint64(lastHourIntLocation)), usagestatistichour.OrganizationIDEQ(orgID), usagestatistichour.BotIDEQ("")).
|
|
|
+// SetConsumeCoin(orgHourConsumeCoinFloat).
|
|
|
+// Save(l.ctx)
|
|
|
+// if err != nil {
|
|
|
+// l.Errorf("save hour data error:%v \n", err)
|
|
|
+// }
|
|
|
+// }
|
|
|
+//
|
|
|
+// // 先判断该租户是否已经统计了小时数据，如果已经统计了，就不需要再统计了
|
|
|
+// _, err = l.svcCtx.DB.UsageStatisticHour.Update().
|
|
|
+// Where(usagestatistichour.AddtimeEQ(uint64(lastHourIntLocation)), usagestatistichour.OrganizationIDEQ(0), usagestatistichour.BotIDEQ("")).
|
|
|
+// SetConsumeCoin(allHourConsumeCoinFloat).
|
|
|
+// Save(l.ctx)
|
|
|
+// if err != nil {
|
|
|
+// l.Errorf("save hour data error:%v \n", err)
|
|
|
+// }
|
|
|
+//
|
|
|
+// /*
|
|
|
+// 计算日数据
|
|
|
+// ----------------------------------------------------------------------------------------------------------
|
|
|
+// */
|
|
|
+// //dayStr := time.Now().Format("20060102")
|
|
|
+// //day, _ := strconv.Atoi(dayStr)
|
|
|
+//
|
|
|
+// // 获取昨天的第一小时
|
|
|
+// yesterday := now.AddDate(0, 0, -1)
|
|
|
+// yesterdayFirstHour := time.Date(yesterday.Year(), yesterday.Month(), yesterday.Day(), 0, 0, 0, 0, now.Location())
|
|
|
+// yesterdayInt, _ := strconv.Atoi(yesterdayFirstHour.Format("20060102"))
|
|
|
+// yesterdayFirstHourInt, _ := strconv.Atoi(yesterdayFirstHour.Format("2006010215"))
|
|
|
+//
|
|
|
+// // 获取昨天的最后一小时
|
|
|
+// yesterdayLastHour := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())
|
|
|
+// yesterdayLastHourInt, _ := strconv.Atoi(yesterdayLastHour.Format("2006010215"))
|
|
|
+//
|
|
|
+// var allDayConsumeCoinFloat float64
|
|
|
+// for orgID, wxinfos := range wxbotsSet {
|
|
|
+// var orgDayConsumeCoinFloat float64
|
|
|
+// for _, wxinfo := range wxinfos {
|
|
|
+// l.Logger.Infof("开始计算日数据:%d\n", yesterdayInt)
|
|
|
+// l.Logger.Infof("yesterdayFirstHourInt:%d\n", yesterdayFirstHourInt)
|
|
|
+// l.Logger.Infof("yesterdayLastHourInt:%d\n", yesterdayLastHourInt)
|
|
|
+//
|
|
|
+// hourDataBatch, _ := l.svcCtx.DB.UsageStatisticHour.Query().Where(
|
|
|
+// usagestatistichour.Type(1),
|
|
|
+// usagestatistichour.BotID(wxinfo),
|
|
|
+// usagestatistichour.OrganizationIDEQ(orgID),
|
|
|
+// usagestatistichour.AddtimeGTE(uint64(yesterdayFirstHourInt)),
|
|
|
+// usagestatistichour.AddtimeLT(uint64(yesterdayLastHourInt)),
|
|
|
+// ).All(l.ctx)
|
|
|
+//
|
|
|
+// if hourDataBatch == nil {
|
|
|
+// continue
|
|
|
+// }
|
|
|
+//
|
|
|
+// var consumeCoin float64
|
|
|
+// for _, hourData := range hourDataBatch {
|
|
|
+// consumeCoin += hourData.ConsumeCoin
|
|
|
+// }
|
|
|
+//
|
|
|
+// orgDayConsumeCoinFloat += consumeCoin
|
|
|
+//
|
|
|
+// allDayConsumeCoinFloat += consumeCoin
|
|
|
+//
|
|
|
+// // 如果添加过了就略过
|
|
|
+// if yesterdayLastHourInt <= currentHourInt {
|
|
|
+// _, err := l.svcCtx.DB.UsageStatisticDay.Update().
|
|
|
+// Where(usagestatisticday.AddtimeEQ(uint64(yesterdayInt)), usagestatisticday.BotID(wxinfo), usagestatisticday.OrganizationIDEQ(orgID)).
|
|
|
+// SetConsumeCoin(consumeCoin).
|
|
|
+// Save(l.ctx)
|
|
|
+// if err != nil {
|
|
|
+// l.Errorf("create day data error:%v \n", err)
|
|
|
+// continue
|
|
|
+// }
|
|
|
+// }
|
|
|
+// }
|
|
|
+// // 如果添加过了就略过
|
|
|
+// if yesterdayLastHourInt <= currentHourInt {
|
|
|
+// _, err := l.svcCtx.DB.UsageStatisticDay.Update().
|
|
|
+// Where(usagestatisticday.AddtimeEQ(uint64(yesterdayInt)), usagestatisticday.OrganizationIDEQ(orgID), usagestatisticday.BotIDEQ("")).
|
|
|
+// SetConsumeCoin(orgDayConsumeCoinFloat).
|
|
|
+// Save(l.ctx)
|
|
|
+// if err != nil {
|
|
|
+// l.Errorf("create day data error:%v \n", err)
|
|
|
+// continue
|
|
|
+// }
|
|
|
+// }
|
|
|
+// }
|
|
|
+//
|
|
|
+// // 如果添加过了就略过
|
|
|
+// if yesterdayLastHourInt <= currentHourInt {
|
|
|
+// _, err = l.svcCtx.DB.UsageStatisticDay.Update().
|
|
|
+// Where(usagestatisticday.AddtimeEQ(uint64(yesterdayInt)), usagestatisticday.OrganizationIDEQ(0), usagestatisticday.BotIDEQ("")).
|
|
|
+// SetConsumeCoin(allDayConsumeCoinFloat).
|
|
|
+// Save(l.ctx)
|
|
|
+// if err != nil {
|
|
|
+// l.Errorf("create day data error:%v \n", err)
|
|
|
+// }
|
|
|
+// }
|
|
|
+// }
|
|
|
+//
|
|
|
+// return
|
|
|
+//}
|
