jiangshuai
2023-09-22 34e1e9dbc7a750048c0825b8f91b07ed526ae577
controllers/operation.go
@@ -5,7 +5,6 @@
   "fmt"
   "github.com/gin-gonic/gin"
   "github.com/shopspring/decimal"
   "github.com/spf13/cast"
   "gorm.io/gorm"
   "strconv"
   "wms/constvar"
@@ -65,9 +64,9 @@
      if location, err := models.NewLocationSearch().SetType(int(constvar.LocationTypeVendor)).First(); err != nil {
         return err
      } else {
         params.FromLocationId = location.Id
         params.FromLocationID = location.Id
      }
      if params.ToLocationId == 0 {
      if params.ToLocationID == 0 {
         return errors.New("请选择目标位置")
      }
   }
@@ -75,17 +74,17 @@
      if location, err := models.NewLocationSearch().SetType(int(constvar.LocationTypeCustomer)).First(); err != nil {
         return err
      } else {
         params.ToLocationId = location.Id
         params.ToLocationID = location.Id
      }
      if params.FromLocationId == 0 {
      if params.FromLocationID == 0 {
         return errors.New("请选择源位置")
      }
   }
   if operationType.BaseOperationType == constvar.BaseOperationTypeInternal {
      if params.ToLocationId == 0 {
      if params.ToLocationID == 0 {
         return errors.New("请选择目标位置")
      }
      if params.FromLocationId == 0 {
      if params.FromLocationID == 0 {
         return errors.New("请选择源位置")
      }
   }
@@ -112,12 +111,12 @@
   //检查明细部分
   for _, v := range params.Details {
      if v.ProductId == "" {
         return errors.New("productID为0")
         return errors.New("productID为空")
      }
      if v.ProductName == "" {
         return errors.New("产品名称异常")
      }
      if v.Quantity.IsNegative() {
      if v.Amount.IsNegative() {
         return errors.New("产品数量出错")
      }
   }
@@ -170,15 +169,9 @@
// @Summary   修改入库/出库信息
// @Produce   application/json
// @Param     object  body request.UpdateOperation true  "入库信息"
// @Param     id  path int true  "入库信息id"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/operation/operation/{id} [put]
// @Router    /api-wms/v1/operation/update [post]
func (slf OperationController) Update(c *gin.Context) {
   id := cast.ToUint(c.Param("id"))
   if id == 0 {
      util.ResponseFormat(c, code.RequestParamError, "空的记录id")
      return
   }
   var reqParams request.UpdateOperation
   var params models.Operation
   if err := c.BindJSON(&reqParams); err != nil {
@@ -203,6 +196,10 @@
   }
   if err := models.WithTransaction(func(tx *gorm.DB) error {
      if err := models.NewOperationDetailsSearch().SetOrm(tx).SetOperationId(params.Id).Delete(); err != nil {
         return err
      }
      operationSearch := models.NewOperationSearch().SetOrm(tx)
      if err := operationSearch.Orm.Model(&params).Association("Details").Replace(params.Details); err != nil {
         return err
      }
      if err := models.NewOperationSearch().SetOrm(tx).SetID(params.Id).Save(&params); err != nil {
@@ -296,11 +293,7 @@
      if err := models.NewOperationSearch().SetOrm(tx).SetID(id).Update(&models.Operation{Status: constvar.OperationStatus_Finish}); err != nil {
         return err
      }
      if operationType.BaseOperationType == constvar.BaseOperationTypeIncoming {
         if err := models.NewMaterialSearch().Orm.Exec("update material INNER JOIN wms_operation_details on wms_operation_details.product_id=material.id INNER JOIN wms_operation on wms_operation.id=wms_operation_details.operation_id set material.amount=material.amount + wms_operation_details.quantity where wms_operation.id=?", id).Error; err != nil {
            return err
         }
      } else if operationType.BaseOperationType == constvar.BaseOperationTypeOutgoing {
      if operationType.BaseOperationType != constvar.BaseOperationTypeInternal {
         var listProdtId []string
         var listProdt []*models.Material
         mapProdt := make(map[string]decimal.Decimal)
@@ -310,23 +303,39 @@
         }
         for _, v := range listDetails {
            listProdtId = append(listProdtId, v.ProductId)
            mapProdt[v.ProductId] = v.Quantity
            mapProdt[v.ProductId] = v.Amount
         }
         if err := models.NewMaterialSearch().Orm.Where("id IN ?", listProdtId).Find(&listProdt).Error; err != nil {
            return err
         }
         for _, v := range listProdt {
            if value, ok := mapProdt[v.ID]; !ok {
               return errors.New("产品种类异常")
            } else {
               if v.Amount.LessThan(value) {
                  return errors.New(fmt.Sprintf("产品:%v,库存:%v,出库:%v,数量不够,无法完成出库操作", v.Name, v.Amount.String(), value.String()))
         if operationType.BaseOperationType == constvar.BaseOperationTypeIncoming {
            for k, v := range listProdt {
               if value, ok := mapProdt[v.ID]; !ok {
                  return errors.New("产品种类异常")
               } else {
                  listProdt[k].Amount.Add(value)
                  if err := tx.Save(listProdt[k]).Error; err != nil {
                     return err
                  }
               }
            }
         }
         if err := models.NewMaterialSearch().Orm.Exec("update material INNER JOIN wms_operation_details on wms_operation_details.product_id=material.id INNER JOIN wms_operation on wms_operation.id=wms_operation_details.operation_id set material.amount=material.amount - wms_operation_details.quantity where wms_operation.id=?", id).Error; err != nil {
            return err
         if operationType.BaseOperationType == constvar.BaseOperationTypeOutgoing {
            for k, v := range listProdt {
               if value, ok := mapProdt[v.ID]; !ok {
                  return errors.New("产品种类异常")
               } else {
                  if v.Amount.LessThan(value) {
                     return errors.New(fmt.Sprintf("产品:%v,库存:%v,出库:%v,数量不够,无法完成出库操作", v.Name, v.Amount.String(), value.String()))
                  }
                  listProdt[k].Amount.Sub(value)
                  if err := tx.Save(listProdt[k]).Error; err != nil {
                     return err
                  }
               }
            }
         }
      }
      return nil
   }); err != nil {