finish operation code refactor

zhangqian

2024-06-14 72b4fcd52bd82427b3932a5a5c01ced2c0dee201

 service/operation.go

@@ -1,9 +1,15 @@
package service

import (
   "errors"
   "fmt"
   "github.com/gin-gonic/gin"
   "github.com/shopspring/decimal"
   "gorm.io/gorm"
   "strconv"
   "time"
   "wms/constvar"
   "wms/middleware"
   "wms/models"
)

@@ -92,3 +98,213 @@

   return models.NewOperationSearch().CreateBatch(operations)
}

// FinishOperationInput 完成入库
func FinishOperationInput(c *gin.Context, tx *gorm.DB, operation *models.Operation, listDetails []*models.OperationDetails, mapLocAmount map[string]*models.LocationProductAmount) (err error) {
   userInfo := middleware.GetUserInfo(c)
   locationRoleList, err := models.NewLocationProductSearch().Find()
   if err != nil {
      return errors.New("获取上架规则信息失败")
   }
   var mapLocationRoleProduct, mapLocationRoleCategory map[string]*models.LocationProduct
   if len(locationRoleList) > 0 {
      mapLocationRoleProduct = make(map[string]*models.LocationProduct)
      mapLocationRoleCategory = make(map[string]*models.LocationProduct)
      for _, v := range locationRoleList {
         if v.RuleType == constvar.RuleType_Product {
            mapLocationRoleProduct[strconv.Itoa(v.AreaId)+v.ProductId] = v
         }
         if v.RuleType == constvar.RuleType_ProductCategory {
            mapLocationRoleCategory[strconv.Itoa(v.AreaId)+strconv.Itoa(v.ProductCategoryID)] = v
         }
      }
   }

   var details []*models.OperationDetails
   for k, v := range listDetails {
      listDetails[k].Product.Amount = listDetails[k].Product.Amount.Add(v.Amount)
      if err := tx.Save(&listDetails[k].Product).Error; err != nil {
         return err
      }

      if roleProduct, ok := mapLocationRoleProduct[strconv.Itoa(operation.LocationID)+v.ProductId]; ok {
         detail := &models.OperationDetails{
            ProductId:      v.ProductId,
            Amount:         v.Amount,
            FromLocationID: roleProduct.AreaId,
            ToLocationID:   roleProduct.LocationId,
         }
         details = append(details, detail)

         if locAmount, aok := mapLocAmount[strconv.Itoa(roleProduct.LocationId)+v.ProductId]; aok {
            locAmount.Amount = locAmount.Amount.Add(v.Amount)
            if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
               return res.Error
            }
         } else {
            if err := models.NewLocationProductAmountSearch().Create(&models.LocationProductAmount{
               LocationId:        roleProduct.LocationId,
               ProductCategoryID: v.Product.CategoryId,
               ProductId:         v.ProductId,
               Amount:            v.Amount,
               CreateDate:        time.Now().Format("2006-01-02 15:04:05"),
            }); err != nil {
               return err
            }
         }
      } else {
         if roleCategory, cok := mapLocationRoleCategory[strconv.Itoa(operation.LocationID)+strconv.Itoa(v.Product.CategoryId)]; cok {
            detail := &models.OperationDetails{
               ProductId:      v.ProductId,
               Amount:         v.Amount,
               FromLocationID: roleCategory.AreaId,
               ToLocationID:   roleCategory.LocationId,
            }
            details = append(details, detail)

            if locAmount, aok := mapLocAmount[strconv.Itoa(roleCategory.LocationId)+v.ProductId]; aok {
               locAmount.Amount = locAmount.Amount.Add(v.Amount)
               if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
                  return res.Error
               }
            } else {
               if err := models.NewLocationProductAmountSearch().Create(&models.LocationProductAmount{
                  LocationId:        roleCategory.LocationId,
                  ProductCategoryID: v.Product.CategoryId,
                  ProductId:         v.ProductId,
                  Amount:            v.Amount,
                  CreateDate:        time.Now().Format("2006-01-02 15:04:05"),
               }); err != nil {
                  return err
               }
            }
         } else {
            if locAmount, aok := mapLocAmount[strconv.Itoa(operation.LocationID)+v.ProductId]; aok {
               locAmount.Amount = locAmount.Amount.Add(v.Amount)
               if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
                  return res.Error
               }
            } else {
               if err := models.NewLocationProductAmountSearch().Create(&models.LocationProductAmount{
                  LocationId:        operation.LocationID,
                  ProductCategoryID: v.Product.CategoryId,
                  ProductId:         v.ProductId,
                  Amount:            v.Amount,
                  CreateDate:        time.Now().Format("2006-01-02 15:04:05"),
               }); err != nil {
                  return err
               }
            }
         }
      }
   }
   if len(details) > 0 {
      if err := tx.Create(&models.Operation{
         Number:            operation.Number,
         SourceNumber:      operation.SourceNumber,
         OperationTypeId:   0,
         OperationTypeName: operation.OperationTypeName,
         Status:            constvar.OperationStatus_Finish,
         OperationDate:     operation.OperationDate,
         ContacterID:       operation.ContacterID,
         ContacterName:     operation.ContacterName,
         CompanyID:         operation.CompanyID,
         CompanyName:       operation.CompanyName,
         Comment:           operation.Comment,
         BaseOperationType: constvar.BaseOperationTypeInternal,
         Details:           details,
         CheckedBy:         userInfo.Username,
      }).Error; err != nil {
         return err
      }
   }
   return nil

}

// FinishOperationOutput 完成出库或报废
func FinishOperationOutput(tx *gorm.DB, listDetails []*models.OperationDetails, mapLocAmount map[string]*models.LocationProductAmount) (err error) {
   for k, v := range listDetails {
      if v.Product.Amount.LessThan(v.Amount) {
         return errors.New(fmt.Sprintf("产品：%v,库存：%v,出库：%v,数量不够，无法完成出库操作", v.Product.Name, v.Product.Amount.String(), v.Amount.String()))
      }
      listDetails[k].Product.Amount = listDetails[k].Product.Amount.Sub(v.Amount)
      if err := models.NewMaterialSearch().SetOrm(tx).Save(&listDetails[k].Product); err != nil {
         return err
      }

      if locAmount, aok := mapLocAmount[strconv.Itoa(v.FromLocationID)+v.ProductId]; aok {
         if locAmount.Amount.LessThan(v.Amount) {
            return errors.New(fmt.Sprintf("产品：%v,库存：%v,出库：%v,数量不够，无法完成出库操作", v.Product.Name, locAmount.Amount.String(), v.Amount.String()))
         }
         locAmount.Amount = locAmount.Amount.Sub(v.Amount)
         if err := models.NewLocationProductAmountSearch().SetID(locAmount.Id).Update(locAmount); err != nil {
            return err
         }
      } else {
         return errors.New("当前仓库没有该产品,请先入库")
      }
   }
   return nil

}

// FinishOperationInternal 完成内部调拨
func FinishOperationInternal(tx *gorm.DB, listDetails []*models.OperationDetails, mapLocAmount map[string]*models.LocationProductAmount) (err error) {
   for _, v := range listDetails {
      if fromLocAmount, aok := mapLocAmount[strconv.Itoa(v.FromLocationID)+v.ProductId]; aok {
         if fromLocAmount.Amount.LessThan(v.Amount) {
            return errors.New(fmt.Sprintf("产品：%v,库存：%v,出库：%v,数量不够，无法完成出库操作", v.Product.Name, fromLocAmount.Amount.String(), v.Amount.String()))
         }
         fromLocAmount.Amount = fromLocAmount.Amount.Sub(v.Amount)
         if err := models.NewLocationProductAmountSearch().SetOrm(tx).SetID(fromLocAmount.Id).Update(fromLocAmount); err != nil {
            return err
         }
      } else {
         return errors.New("当前仓库没有该产品,请先入库")
      }

      if toLocAmount, aok := mapLocAmount[strconv.Itoa(v.ToLocationID)+v.ProductId]; aok {
         toLocAmount.Amount = toLocAmount.Amount.Add(v.Amount)
         if err := models.NewLocationProductAmountSearch().SetOrm(tx).SetID(toLocAmount.Id).Update(toLocAmount); err != nil {
            return err
         }
      } else {
         if err := models.NewLocationProductAmountSearch().SetOrm(tx).Create(&models.LocationProductAmount{
            LocationId:        v.ToLocationID,
            ProductCategoryID: v.Product.CategoryId,
            ProductId:         v.ProductId,
            Amount:            v.Amount,
            CreateDate:        time.Now().Format("2006-01-02 15:04:05"),
         }); err != nil {
            return err
         }
      }
   }
   return nil

}

// FinishOperationAdjust 完成库存调整
func FinishOperationAdjust(tx *gorm.DB, listDetails []*models.OperationDetails, mapLocAmount map[string]*models.LocationProductAmount) (err error) {
   for _, v := range listDetails {
      if locAmount, aok := mapLocAmount[strconv.Itoa(v.ToLocationID)+v.ProductId]; aok {
         locAmount.Amount = v.Amount
         if err := models.NewLocationProductAmountSearch().SetOrm(tx).SetID(locAmount.Id).Save(locAmount); err != nil {
            return err
         }
      } else {
         if err := models.NewLocationProductAmountSearch().SetOrm(tx).Create(&models.LocationProductAmount{
            LocationId:        v.ToLocationID,
            ProductCategoryID: v.Product.CategoryId,
            ProductId:         v.ProductId,
            Amount:            v.Amount,
            CreateDate:        time.Now().Format("2006-01-02 15:04:05"),
         }); err != nil {
            return err
         }
      }
   }
   return nil

}