From b1fc20d02bb883b88365cc81ad90c8b1b45ef6ae Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期六, 29 六月 2024 21:52:08 +0800
Subject: [PATCH] fix
---
task/month_stats.go | 186 ++++++++++------------------------------------
1 files changed, 41 insertions(+), 145 deletions(-)
diff --git a/task/month_stats.go b/task/month_stats.go
index ee539ea..b2de4fb 100644
--- a/task/month_stats.go
+++ b/task/month_stats.go
@@ -1,31 +1,33 @@
package task
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())
}
}()
@@ -40,154 +42,48 @@
oldRecordsMap := models.MonthStatsMap(oldRecords)
//鏈湀鏈熷垵鏁伴噺/涓婃湀缁撲綑鏁伴噺
- groupSumList, err := models.NewLocationProductAmountSearch().GroupSum("product_id", "amount")
-
- productIds := make([]string, 0, len(groupSumList))
-
- for _, groupSum := range groupSumList {
- productIds = append(productIds, groupSum.Class)
- }
- products, err := models.NewMaterialSearch().SetFields("id, name, unit, weight, more_unit, more_unit_value").SetIDs(productIds).FindNotTotal()
+ statsRecords, err := service.GetCurrentStats(date, nil)
if err != nil {
- logx.Errorf("MonthStats get products err:%v", err)
- return
+ return err
}
- productMap := models.MaterialMap(products)
- beginTime, endTime := GetLastMonthPeriod()
- inputMap, err := GetStatsByOperationType(beginTime, endTime, constvar.BaseOperationTypeIncoming)
+ err = models.NewMonthStatsSearch().SetDate(date).Delete()
if err != nil {
- logx.Errorf("MonthStats GetStatsByOperationType input err:%v", err)
return
}
- outputMap, err := GetStatsByOperationType(beginTime, endTime, constvar.BaseOperationTypeOutgoing)
- if err != nil {
- logx.Errorf("MonthStats GetStatsByOperationType output err:%v", err)
- return
- }
- var record models.MonthStats
- for _, groupSum := range groupSumList {
- productId := groupSum.Class
- if productMap[productId] == nil {
- continue
- }
- product := productMap[productId]
- amount := groupSum.Sum
- record = models.MonthStats{
- ProductId: productId,
- ProductName: product.Name,
- BeginAmount: amount,
- Unit: product.Unit,
- Weight: product.Weight.Mul(amount),
- Date: date,
- }
-
- var (
- moreUnits string
- inputMoreUnits string
- outputMoreUnits string
- )
- 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))
- for _, unitItem := range product.MoreUnitList {
- moreValueArr = append(moreValueArr, models.UnitItems{
- Amount: amount.Mul(unitItem.Amount),
- Unit: unitItem.Unit,
- Floating: unitItem.Floating,
- })
- bys, _ := json.Marshal(moreValueArr)
- moreUnits = string(bys)
-
- if !inputMap[productId].IsZero() {
- inputMoreValueArr = append(inputMoreValueArr, models.UnitItems{
- Amount: inputMap[productId].Mul(unitItem.Amount),
- Unit: unitItem.Unit,
- Floating: unitItem.Floating,
- })
- bys, _ = json.Marshal(inputMoreValueArr)
- inputMoreUnits = string(bys)
-
- }
-
- if !outputMap[productId].IsZero() {
- outputMoreValueArr = append(outputMoreValueArr, models.UnitItems{
- Amount: outputMap[productId].Mul(unitItem.Amount),
- Unit: unitItem.Unit,
- Floating: unitItem.Floating,
- })
- bys, _ = json.Marshal(outputMoreValueArr)
- outputMoreUnits = string(bys)
-
- }
- }
- }
-
- record.BeginMoreUnits = moreUnits
- err = models.NewMonthStatsSearch().Create(&record)
- if err != nil {
- 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().SetID(oldRecordsMap[productId].Id).UpdateByMap(m)
+ for _, record := range statsRecords {
+ record.BeginAmount = record.EndAmount
+ record.BeginMoreUnits = record.EndMoreUnits
+ 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)
- service.SendAlarm("鏈堝害缁熻鏇存敼涓婃湀澶辫触", fmt.Sprintf("NewMonthStatsSearch Create err:%v, record: %+v", err, record))
+ logx.Errorf("NewMonthStatsSearch Create err:%v, record: %+v", err, record)
+ service.SendAlarm("鏈堝害缁熻鍒涘缓鏈湀澶辫触", fmt.Sprintf("NewMonthStatsSearch Create err:%v, record: %+v", err, record))
}
- }
+
+ if oldRecordsMap[record.ProductId] != nil && oldRecordsMap[record.ProductId].Id != 0 {
+ m := map[string]interface{}{
+ "end_more_units": record.BeginMoreUnits,
+ "end_amount": record.BeginAmount,
+ }
+ if !record.InputAmount.IsZero() || !record.InputAmount.IsZero() {
+ m["input_amount"] = record.InputAmount
+ m["input_more_units"] = record.InputMoreUnits
+ m["output_amount"] = record.InputAmount
+ m["output_more_units"] = record.OutputMoreUnits
+ }
+ err = models.NewMonthStatsSearch().SetOrm(db).SetID(oldRecordsMap[record.ProductId].Id).UpdateByMap(m)
+ if err != nil {
+ logx.Errorf("NewMonthStatsSearch UpdateByMap err:%v, id:%v, m:%+v", err, oldRecordsMap[record.ProductId].Id, m)
+ service.SendAlarm("鏈堝害缁熻鏇存敼涓婃湀澶辫触", fmt.Sprintf("NewMonthStatsSearch Create err:%v, record: %+v", err, record))
+ }
+ }
+ return nil
+ })
+ }
+ if err == nil {
+ service.SendAlarm("鏈堝害缁熻鎵ц鎴愬姛", "")
}
return
-}
-
-func GetStatsByOperationType(beginTime, endTime time.Time, operationType constvar.BaseOperationType) (m map[string]decimal.Decimal, err error) {
- operationIds, err := models.NewOperationSearch().SetBaseOperationType(operationType).SetFields("id").SetTimeBetween(beginTime, endTime).FindIds()
- if err != nil {
- return
- }
- groupSumList, err := models.NewOperationDetailsSearch().SetOperationIds(operationIds).SetFields("product_id, amount").GroupSum("product_id", "amount")
- if err != nil {
- return
- }
- m = make(map[string]decimal.Decimal, len(groupSumList))
- for _, v := range groupSumList {
- m[v.Class] = v.Sum
- }
- return
-}
-
-// GetLastMonthPeriod 杩斿洖涓婁釜鏈堢殑鏈堝垵鏃堕棿鍜屾湀鏈椂闂�
-func GetLastMonthPeriod() (time.Time, time.Time) {
- // 鑾峰彇褰撳墠鏃堕棿
- now := time.Now()
-
- // 璁＄畻涓婁釜鏈堢殑骞翠唤鍜屾湀浠�
- lastMonth := now.AddDate(0, -1, 0)
- lastYear, lastMonthNum, _ := lastMonth.Date()
-
- // 鑾峰彇涓婁釜鏈堢殑绗竴澶╃殑鏃ユ湡锛堝嵆涓婁釜鏈堟湀鍒濓級
- firstDayOfLastMonth := time.Date(lastYear, lastMonthNum, 1, 0, 0, 0, 0, now.Location())
-
- // 鑾峰彇鏈湀绗竴澶╃殑鏃ユ湡锛堝嵆鏈湀鏈堝垵锛�
- firstDayOfThisMonth := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location())
-
- // 涓婁釜鏈堟湀鏈椂闂村嵆涓烘湰鏈堟湀鍒濆噺鍘讳竴绉�
- lastDayOfLastMonth := firstDayOfThisMonth.Add(-time.Second)
-
- return firstDayOfLastMonth, lastDayOfLastMonth
}
--
Gitblit v1.8.0