zhangqian
2024-07-01 73b6baf6af3d88cdcb0e2df7932a9bd96b0b85c5
controllers/operation_type.go
@@ -7,6 +7,7 @@
   "github.com/spf13/cast"
   "gorm.io/gorm"
   "strconv"
   "strings"
   "wms/constvar"
   "wms/extend/code"
   "wms/extend/util"
@@ -116,7 +117,22 @@
      util.ResponseFormat(c, code.RequestParamError, err.Error())
      return
   }
   list, total, err := models.NewOperationTypeSearch().SetPage(params.Page, params.PageSize).SetKeyword(params.Keyword).SetOrder("id desc").SetPreload(true).Find()
   warehouseIds := make([]int, 0)
   if "" != params.WarehouseIds {
      for _, strId := range strings.Split(params.WarehouseIds, ",") {
         intId, err := strconv.Atoi(strId)
         if err != nil {
            util.ResponseFormat(c, code.RequestParamError, err.Error())
         }
         warehouseIds = append(warehouseIds, intId)
      }
   }
   list, total, err := models.NewOperationTypeSearch().
      SetPage(params.Page, params.PageSize).
      SetKeyword(params.Keyword).
      SetOrder("id desc").
      SetWarehouseIds(warehouseIds).
      SetPreload(true).Find()
   if err != nil {
      util.ResponseFormat(c, code.RequestError, "查找失败")
      return
@@ -140,6 +156,9 @@
      }
      if value, ok := mapStatistics[strconv.Itoa(v.Id)+string(constvar.OperationStatus_Finish)]; ok {
         list[k].FinishCount = value.Count
      }
      if value, ok := mapStatistics[strconv.Itoa(v.Id)+string(constvar.OperationStatus_Cancel)]; ok {
         list[k].CancelCount = value.Count
      }
   }
   util.ResponseFormatListWithPage(c, code.Success, list, cast.ToInt(total), params.Page, params.PageSize)
@@ -205,6 +224,9 @@
      if value, ok := mapStatistics[strconv.Itoa(v.Id)+string(constvar.OperationStatus_Finish)]; ok {
         list[k].FinishCount = value.Count
      }
      if value, ok := mapStatistics[strconv.Itoa(v.Id)+string(constvar.OperationStatus_Cancel)]; ok {
         list[k].FinishCount = value.Count
      }
   }
   util.ResponseFormatListWithPage(c, code.Success, list, cast.ToInt(total), params.Page, params.PageSize)
}