增加月度统计手动执行接口

zhangqian

2024-05-06 8d3f2400195b0ae746e51ed4871e2b0ab621c928

 task/month_stats.go

@@ -2,28 +2,35 @@

import (
   "encoding/json"
   "fmt"
   "github.com/shopspring/decimal"
   "gorm.io/gorm"
   "time"
   "wms/constvar"
   "wms/models"
   "wms/pkg/logx"
   "wms/service"
)

func MonthStats() {
func MonthStats() (err error) {
   //加锁，只需要一个进程运行此任务
   var (
      lockName  = "monthStats"
      serviceID = "wms"
   )
   err := models.NewLockSearch().AcquireLock(lockName, serviceID)
   err = models.NewLockSearch().AcquireLock(lockName, serviceID)
   if err != nil {
      logx.Errorf("MonthStats AcquireLock err:%v", err)
      return
   }
   defer func() {
      err := models.NewLockSearch().ReleaseLock(lockName, serviceID)
      if err != nil {
         service.SendAlarm("月度统计执行失败", "报错: "+err.Error())
      }
      err = models.NewLockSearch().ReleaseLock(lockName, serviceID)
      if err != nil {
         logx.Errorf("MonthStats ReleaseLock err:%v", err)
         service.SendAlarm("月度统计执行成功解锁失败", err.Error())
      }
   }()

@@ -65,6 +72,12 @@
      return
   }
   var record models.MonthStats

   err = models.NewMonthStatsSearch().SetDate(date).Delete()
   if err != nil {
      return
   }

   for _, groupSum := range groupSumList {
      productId := groupSum.Class
      if productMap[productId] == nil {
@@ -86,7 +99,7 @@
         inputMoreUnits  string
         outputMoreUnits string
      )
      if product.MoreUnit {
      if *product.MoreUnit {
         moreValueArr := make([]models.UnitItems, 0, len(product.MoreUnitList))
         inputMoreValueArr := make([]models.UnitItems, 0, len(product.MoreUnitList))
         outputMoreValueArr := make([]models.UnitItems, 0, len(product.MoreUnitList))
@@ -124,29 +137,37 @@
      }

      record.BeginMoreUnits = moreUnits
      err = models.NewMonthStatsSearch().Create(&record)
      if err != nil {
         logx.Errorf("NewMonthStatsSearch Create err:%v, record: %+v", err, record)
      }

      if oldRecordsMap[productId] != nil && (!inputMap[productId].IsZero() || !outputMap[productId].IsZero()) {
         record.InputAmount = inputMap[productId]
         record.InputMoreUnits = inputMoreUnits
         record.OutputAmount = outputMap[productId]
         record.OutputMoreUnits = outputMoreUnits
         m := map[string]interface{}{
            "input_amount":      inputMap[productId],
            "input_more_units":  inputMoreUnits,
            "output_amount":     outputMap[productId],
            "output_more_units": outputMoreUnits,
            "end_more_units":    moreUnits,
            "end_amount":        amount,
         }
         err = models.NewMonthStatsSearch().SetID(oldRecordsMap[productId].Id).UpdateByMap(m)
      err = models.WithTransaction(func(db *gorm.DB) error {
         err = models.NewMonthStatsSearch().SetOrm(db).Create(&record)
         if err != nil {
            logx.Errorf("NewMonthStatsSearch UpdateByMap err:%v, id:%v, m:%+v", err, oldRecordsMap[productId].ID, m)
            logx.Errorf("NewMonthStatsSearch Create err:%v, record: %+v", err, record)
            service.SendAlarm("月度统计创建本月失败", fmt.Sprintf("NewMonthStatsSearch Create err:%v, record: %+v", err, record))
         }
      }

         if oldRecordsMap[productId] != nil && (!inputMap[productId].IsZero() || !outputMap[productId].IsZero()) {
            record.InputAmount = inputMap[productId]
            record.InputMoreUnits = inputMoreUnits
            record.OutputAmount = outputMap[productId]
            record.OutputMoreUnits = outputMoreUnits
            m := map[string]interface{}{
               "input_amount":      inputMap[productId],
               "input_more_units":  inputMoreUnits,
               "output_amount":     outputMap[productId],
               "output_more_units": outputMoreUnits,
               "end_more_units":    moreUnits,
               "end_amount":        amount,
            }
            err = models.NewMonthStatsSearch().SetOrm(db).SetID(oldRecordsMap[productId].Id).UpdateByMap(m)
            if err != nil {
               logx.Errorf("NewMonthStatsSearch UpdateByMap err:%v, id:%v, m:%+v", err, oldRecordsMap[productId].ID, m)
               service.SendAlarm("月度统计更改上月失败", fmt.Sprintf("NewMonthStatsSearch Create err:%v, record: %+v", err, record))
            }
         }
         return nil
      })
   }
   if err == nil {
      service.SendAlarm("月度统计执行成功", "")
   }
   return
}