liujiandao
2023-09-23 ea02a2a9a8219d44b103f0a31cf9fc81ff8495e1
controllers/operation.go
@@ -4,6 +4,10 @@
   "errors"
   "fmt"
   "github.com/gin-gonic/gin"
   "github.com/shopspring/decimal"
   "gorm.io/gorm"
   "strconv"
   "wms/constvar"
   "wms/extend/code"
   "wms/extend/util"
   "wms/models"
@@ -27,19 +31,64 @@
   var params models.Operation
   if err := c.BindJSON(&reqParams); err != nil {
      util.ResponseFormat(c, code.RequestParamError, "参数解析失败,数据类型错误")
      return
   }
   if err := structx.AssignTo(reqParams, params); err != nil {
      util.ResponseFormat(c, code.RequestParamError, "数据转换错误")
   if err := structx.AssignTo(reqParams, &params); err != nil {
      util.ResponseFormat(c, code.RequestParamError, "数据转换错误"+err.Error())
      return
   }
   if err := slf.CheckParams(params); err != nil {
      util.ResponseFormat(c, code.RequestParamError, err.Error())
      return
   }
   if err := slf.FormatLocation(&params); err != nil {
      util.ResponseFormat(c, code.RequestParamError, err.Error())
      return
   }
   params.Status = constvar.OperationStatus_Ready
   if err := models.NewOperationSearch().Create(&params); err != nil {
      logx.Errorf("Operation create err: %v", err)
      util.ResponseFormat(c, code.SaveFail, "添加失败:"+err.Error())
      return
   }
   util.ResponseFormat(c, code.Success, "添加成功")
}
func (slf OperationController) FormatLocation(params *models.Operation) error {
   operationType, err := models.NewOperationTypeSearch().SetID(uint(params.OperationTypeId)).First()
   if err != nil {
      return err
   }
   if operationType.BaseOperationType == constvar.BaseOperationTypeIncoming {
      if location, err := models.NewLocationSearch().SetType(int(constvar.LocationTypeVendor)).First(); err != nil {
         return err
      } else {
         params.FromLocationID = location.Id
      }
      if params.ToLocationID == 0 {
         return errors.New("请选择目标位置")
      }
   }
   if operationType.BaseOperationType == constvar.BaseOperationTypeOutgoing {
      if location, err := models.NewLocationSearch().SetType(int(constvar.LocationTypeCustomer)).First(); err != nil {
         return err
      } else {
         params.ToLocationID = location.Id
      }
      if params.FromLocationID == 0 {
         return errors.New("请选择源位置")
      }
   }
   if operationType.BaseOperationType == constvar.BaseOperationTypeInternal {
      if params.ToLocationID == 0 {
         return errors.New("请选择目标位置")
      }
      if params.FromLocationID == 0 {
         return errors.New("请选择源位置")
      }
   }
   return nil
}
func (slf OperationController) CheckParams(params models.Operation) error {
@@ -51,15 +100,7 @@
      return errors.New("operationTypeId为0")
   }
   if params.FromLocationId == 0 {
      return errors.New("请选择源位置")
   }
   if params.ToLocationId == 0 {
      return errors.New("请选择目标位置")
   }
   if params.OperationDate.IsZero() {
   if params.OperationDate == "" {
      return errors.New("请选择安排日期")
   }
@@ -69,20 +110,225 @@
   //检查明细部分
   for _, v := range params.Details {
      if v.ProductId == 0 {
         return errors.New("productID为0")
      if v.ProductId == "" {
         return errors.New("productID为空")
      }
      if v.ProductName == "" {
         return errors.New("产品名称异常")
      }
      if v.Quantity.IsNegative() {
         return errors.New("产品数量出错")
      }
      if v.FinishQuantity.IsNegative() {
      if v.Amount.IsNegative() {
         return errors.New("产品数量出错")
      }
   }
   fmt.Println(111111)
   return nil
}
// List
// @Tags      入库/出库
// @Summary   入库/出库列表
// @Produce   application/json
// @Accept     json
// @Param     object  query  request.OperationList true  "参数"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/operation/operation [get]
func (slf OperationController) List(c *gin.Context) {
   var params request.OperationList
   if err := c.ShouldBindQuery(&params); err != nil {
      util.ResponseFormat(c, code.RequestParamError, "参数解析失败,数据类型错误:"+err.Error())
      return
   }
   if err := slf.CheckListParams(&params); err != nil {
      util.ResponseFormat(c, code.RequestParamError, err.Error())
      return
   }
   search := models.NewOperationSearch()
   search.SetPage(params.Page, params.PageSize)
   list, total, err := search.SetOperationTypeId(params.OperationTypeId).SetPreload(true).SetOrder("created_at desc").Find()
   if err != nil {
      util.ResponseFormat(c, code.RequestError, "查找失败:"+err.Error())
      return
   }
   util.ResponseFormatList(c, code.Success, list, int(total))
}
func (slf OperationController) CheckListParams(params *request.OperationList) error {
   if !params.PageInfo.Check() {
      return errors.New("数据分页信息错误")
   }
   if params.OperationTypeId == 0 {
      return errors.New("operationTypeId为0")
   }
   return nil
}
// Update
// @Tags      入库/出库
// @Summary   修改入库/出库信息
// @Produce   application/json
// @Param     object  body request.UpdateOperation true  "入库信息"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/operation/update [post]
func (slf OperationController) Update(c *gin.Context) {
   var reqParams request.UpdateOperation
   var params models.Operation
   if err := c.BindJSON(&reqParams); err != nil {
      util.ResponseFormat(c, code.RequestParamError, "参数解析失败,数据类型错误:"+err.Error())
      return
   }
   if reqParams.Status != constvar.OperationStatus_Ready {
      util.ResponseFormat(c, code.RequestParamError, "该入库信息已完成,无法进行修改")
      return
   }
   if err := structx.AssignTo(reqParams, &params); err != nil {
      util.ResponseFormat(c, code.RequestParamError, "数据转换错误"+err.Error())
      return
   }
   if err := slf.CheckParams(params); err != nil {
      util.ResponseFormat(c, code.RequestParamError, err.Error())
      return
   }
   if err := slf.FormatLocation(&params); err != nil {
      util.ResponseFormat(c, code.RequestParamError, err.Error())
      return
   }
   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 {
         return err
      }
      return nil
   }); err != nil {
      util.ResponseFormat(c, code.RequestParamError, "修改失败:"+err.Error())
      return
   }
   util.ResponseFormat(c, code.Success, "修改成功")
}
// DeleteDevice
//
//   @Tags      入库/出库
//   @Summary   删除入库/出库信息
//   @Produce   application/json
//   @Param      id   path      int         true   "id"
//   @Success   200   {object}   util.Response   "成功"
//   @Router      /api-wms/v1/operation/operation/{id} [delete]
func (slf OperationController) Delete(c *gin.Context) {
   id, err := strconv.Atoi(c.Param("id"))
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "错误的id值")
      return
   }
   if id == 0 {
      util.ResponseFormat(c, code.RequestParamError, "id为0")
      return
   }
   operation, err := models.NewOperationSearch().SetID(id).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "未找到相关出入库信息:"+err.Error())
      return
   }
   if operation.Status != constvar.OperationStatus_Ready {
      util.ResponseFormat(c, code.RequestError, "该入库信息无法进行删除")
      return
   }
   if err := models.WithTransaction(func(tx *gorm.DB) error {
      if err := models.NewOperationDetailsSearch().SetOrm(tx).SetOperationId(id).Delete(); err != nil {
         return err
      }
      if err := models.NewOperationSearch().SetOrm(tx).SetID(id).Delete(); err != nil {
         return err
      }
      return nil
   }); err != nil {
      util.ResponseFormat(c, code.RequestParamError, "修改失败:"+err.Error())
      return
   }
   util.ResponseFormat(c, code.Success, "删除成功")
}
// DeleteDevice
//
//   @Tags      入库/出库
//   @Summary   更改记录状态
//   @Produce   application/json
//   @Param      id   path      int         true   "id"
//   @Success   200   {object}   util.Response   "成功"
//   @Router      /api-wms/v1/operation/finish/{id} [put]
func (slf OperationController) Finish(c *gin.Context) {
   id, err := strconv.Atoi(c.Param("id"))
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "错误的id值")
      return
   }
   if id == 0 {
      util.ResponseFormat(c, code.RequestParamError, "id为0")
      return
   }
   operation, err := models.NewOperationSearch().SetID(id).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "未找到相关出入库信息:"+err.Error())
      return
   }
   if operation.Status != constvar.OperationStatus_Ready {
      util.ResponseFormat(c, code.RequestError, "该出入库信息无法完成")
      return
   }
   operationType, err := models.NewOperationTypeSearch().SetID(uint(operation.OperationTypeId)).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestError, err.Error())
      return
   }
   if err := models.WithTransaction(func(tx *gorm.DB) error {
      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 {
         var listProdtId []string
         var listProdt []*models.Material
         mapProdt := make(map[string]decimal.Decimal)
         listDetails, err := models.NewOperationDetailsSearch().SetOperationId(operation.Id).FindAll()
         if err != nil {
            return err
         }
         for _, v := range listDetails {
            listProdtId = append(listProdtId, v.ProductId)
            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 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
         }
      }
      return nil
   }); err != nil {
      util.ResponseFormat(c, code.RequestError, err.Error())
      return
   }
   util.ResponseFormat(c, code.Success, "操作成功")
}