weedfs包放到项目里

zhangqian

2024-06-12 e5df488268e29b272932e6cc1d2b1e7034590ba0

 controllers/warehouse.go

@@ -6,6 +6,9 @@
   "github.com/gin-gonic/gin"
   "github.com/spf13/cast"
   "gorm.io/gorm"
   "strconv"
   "strings"
   "wms/constvar"
   "wms/extend/code"
   "wms/extend/util"
   "wms/models"
@@ -39,7 +42,61 @@
      util.ResponseFormat(c, code.RequestParamError, err.Error())
      return
   }
   if err := models.NewWarehouseSearch().Create(&params); err != nil {
   //创建视图
   view := &models.Location{
      Name:      params.Code,
      JointName: params.Code,
      Type:      constvar.LocationTypeView,
   }
   id, err := models.NewLocationSearch().CreateReturnId(view)
   if err != nil {
      util.ResponseFormat(c, code.SaveFail, "视图创建失败")
      return
   }
   //创建默认位置
   location := &models.Location{
      Name:              "默认位置",
      JointName:         params.Code + "/默认位置",
      Type:              constvar.LocationTypeInternal,
      ReplenishLocation: true,
      ParentId:          id,
   }
   locationId, err := models.NewLocationSearch().CreateReturnId(location)
   if err != nil {
      util.ResponseFormat(c, code.SaveFail, "位置创建失败")
      return
   }
   params.LocationId = locationId
   err = models.WithTransaction(func(tx *gorm.DB) error {
      err := models.NewWarehouseSearch().SetOrm(tx).Create(&params)
      if err != nil {
         return err
      }
      //创建三个默认操作类型
      var types []*models.OperationType
      inType := &models.OperationType{
         Name:              params.Name + "-入库",
         BaseOperationType: constvar.BaseOperationTypeIncoming,
         WarehouseId:       params.Id,
      }
      types = append(types, inType)
      outType := &models.OperationType{
         Name:              params.Name + "-出库",
         BaseOperationType: constvar.BaseOperationTypeOutgoing,
         WarehouseId:       params.Id,
      }
      types = append(types, outType)
      internalType := &models.OperationType{
         Name:              params.Name + "-内部调拨",
         BaseOperationType: constvar.BaseOperationTypeInternal,
         WarehouseId:       params.Id,
      }
      types = append(types, internalType)
      err = models.NewOperationTypeSearch().SetOrm(tx).CreateBatch(types)
      return err
   })
   if err != nil {
      _ = models.NewLocationSearch().SetID(locationId).Delete()
      logx.Errorf("warehouse create err: %v", err)
      util.ResponseFormat(c, code.SaveFail, "插入失败")
      return
@@ -48,39 +105,51 @@
   util.ResponseFormat(c, code.Success, "添加成功")
}

// Update
// UpdateWarehouse
// @Tags      仓库
// @Summary   编辑仓库
// @Produce   application/json
// @Param     object  body request.UpdateWarehouse true  "仓库信息"
// @Param     id  path string true  "仓库id"
// @Param     object  body  models.Warehouse true  "仓库信息"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/warehouse/warehouse/{id} [put]
func (slf WarehouseController) Update(c *gin.Context) {
   id := cast.ToUint(c.Param("id"))
   if id == 0 {
      util.ResponseFormat(c, code.RequestParamError, "空的记录id")
      return
   }
   var (
      reqParams request.UpdateWarehouse
      params    models.Warehouse
   )
   if err := c.BindJSON(&reqParams); err != nil {
// @Router    /api-wms/v1/warehouse/updateWarehouse [post]
func (slf WarehouseController) UpdateWarehouse(c *gin.Context) {
   var params models.Warehouse
   if err := c.BindJSON(&params); err != nil {
      util.ResponseFormat(c, code.RequestParamError, fmt.Sprintf("参数解析失败: %v"+err.Error()))
      return
   }
   if err := structx.AssignTo(reqParams, &params); err != nil {
      util.ResponseFormat(c, code.RequestParamError, fmt.Sprintf("数据转换错误: %v", err.Error()))
      return
   }
   params.ID = id
   if err := slf.ParamsCheck(params); err != nil {
      util.ResponseFormat(c, code.RequestParamError, err.Error())
      return
   }

   err := models.NewWarehouseSearch().SetID(params.ID).Update(&params)
   warehouse, err := models.NewWarehouseSearch().SetID(params.Id).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "仓库不存在")
      return
   }
   err = models.WithTransaction(func(db *gorm.DB) error {
      //更新作业类型名称
      if params.Name != warehouse.Name {
         m := make(map[string]interface{})
         m["name"] = params.Name + "-入库"
         err := models.NewOperationTypeSearch().SetOrm(db).SetBaseOperationType(constvar.BaseOperationTypeIncoming).SetWarehouseId(params.Id).UpdateByMap(m)
         if err != nil {
            return err
         }
         m["name"] = params.Name + "-出库"
         err = models.NewOperationTypeSearch().SetOrm(db).SetBaseOperationType(constvar.BaseOperationTypeOutgoing).SetWarehouseId(params.Id).UpdateByMap(m)
         if err != nil {
            return err
         }
         m["name"] = params.Name + "-内部调拨"
         err = models.NewOperationTypeSearch().SetOrm(db).SetBaseOperationType(constvar.BaseOperationTypeInternal).SetWarehouseId(params.Id).UpdateByMap(m)
         if err != nil {
            return err
         }
      }
      err = models.NewWarehouseSearch().SetID(params.Id).Update(&params)
      return err
   })

   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "修改失败")
@@ -92,19 +161,49 @@

func (slf WarehouseController) ParamsCheck(params models.Warehouse) (err error) {
   var oldRecord *models.Warehouse
   if params.ID != 0 {
      oldRecord, err = models.NewWarehouseSearch().SetID(params.ID).First()
   if params.Id != 0 {
      oldRecord, err = models.NewWarehouseSearch().SetID(params.Id).First()
      if err == gorm.ErrRecordNotFound {
         return errors.New("记录不存在")
      }
   }
   if oldRecord == nil || params.Code != oldRecord.Code {
      _, err = models.NewWarehouseSearch().SetCode(params.Code).First()
      if err != gorm.ErrRecordNotFound {
   //更新位置信息
   if oldRecord != nil && params.Code != oldRecord.Code {
      locations, err := models.NewLocationSearch().SetJointName(oldRecord.Code).FindNotTotal()
      if err != nil {
         return errors.New("查询位置信息失败")
      }

      err = models.WithTransaction(func(db *gorm.DB) error {
         for _, location := range locations {
            m := make(map[string]interface{})
            index := strings.Index(location.JointName, "/")
            if index > 0 {
               m["joint_name"] = params.Code + location.JointName[index:]
            } else {
               m["joint_name"] = params.Code
            }
            err = models.NewLocationSearch().SetOrm(db).SetID(location.Id).UpdateByMap(m)
            if err != nil {
               return err
            }
         }
         return nil
      })
      if err != nil {
         return errors.New("更新位置信息失败")
      }
   }
   if oldRecord == nil {
      record, err := models.NewWarehouseSearch().SetCode(params.Code).First()
      if record != nil && record.Code == params.Code {
         fmt.Println(err)
         return errors.New("仓库编号重复")
      }
   }

   if params.Code == "" {
      return errors.New("缩写不能为空")
   }
   return nil
}

@@ -126,6 +225,24 @@
      util.ResponseFormat(c, code.RequestParamError, "查找失败")
      return
   }
   //获取位置信息
   ids := make([]int, 0)
   for _, warehouse := range list {
      ids = append(ids, warehouse.LocationId)
   }
   locations, err := models.NewLocationSearch().SetIds(ids).FindNotTotal()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "位置信息查找失败")
      return
   }
   for _, warehouse := range list {
      for _, location := range locations {
         if warehouse.LocationId == location.Id {
            warehouse.WarehouseLocation = warehouse.Code + "/" + location.Name
            break
         }
      }
   }

   util.ResponseFormatList(c, code.Success, list, cast.ToInt(total))
}
@@ -138,16 +255,58 @@
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/warehouse/warehouse/{id} [delete]
func (slf WarehouseController) Delete(c *gin.Context) {
   id := cast.ToUint(c.Param("id"))
   id, _ := strconv.Atoi(c.Param("id"))
   if id == 0 {
      util.ResponseFormat(c, code.RequestParamError, "空的记录id")
      return
   }

   err := models.NewWarehouseSearch().SetID(id).Delete()
   first, err := models.NewWarehouseSearch().SetID(id).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "删除失败")
      util.ResponseFormat(c, code.RequestParamError, "获取仓库信息失败")
      return
   }
   err = models.WithTransaction(func(tx *gorm.DB) error {
      //删除位置信息
      err = models.NewLocationSearch().SetOrm(tx).SetJointName(first.Code).Delete()
      if err != nil {
         return err
      }
      //删除操作类型
      err = models.NewOperationTypeSearch().SetOrm(tx).SetWarehouseId(first.Id).Delete()
      if err != nil {
         return err
      }
      //删除仓库
      err = models.NewWarehouseSearch().SetOrm(tx).SetID(id).Delete()
      return err
   })
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "删除仓库失败")
      return
   }

   util.ResponseFormat(c, code.UpdateSuccess, "删除成功")
}

// GetWarehouseDetails
// @Tags      仓库
// @Summary   获取仓库详情
// @Produce   application/json
// @Param     id  path string true  "仓库id"
// @Success   200 {object} util.Response{data=models.Warehouse}   "成功"
// @Router    /api-wms/v1/warehouse/getWarehouseDetails/{id} [get]
func (slf WarehouseController) GetWarehouseDetails(c *gin.Context) {
   id, _ := strconv.Atoi(c.Param("id"))
   if id == 0 {
      util.ResponseFormat(c, code.RequestParamError, "空的记录id")
      return
   }

   first, err := models.NewWarehouseSearch().SetID(id).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "获取仓库信息失败")
      return
   }
   util.ResponseFormat(c, code.UpdateSuccess, first)
}