增加token验证和数据权限过滤

zhangqian

2024-03-23 b68185aed1a86536b01fdfbca6b2cca7bd71a50f

 controllers/order.go

@@ -13,6 +13,7 @@
   "outsourcing/request"
   "outsourcing/service"
   "outsourcing/service/outsourcing"
   "outsourcing/utils/jwt"
)

type OrderController struct{}
@@ -30,7 +31,14 @@
      util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
      return
   }
   list, total, err := models.NewOutsourcingOrderSearch().SetPage(params.Page, params.PageSize).SetKeyword(params.Keyword).SetOrder("id desc").Find()

   companyId := jwt.GetCompanyID(c)
   if companyId == 0 {
      util.ResponseFormat(c, code.InternalError, "内部错误")
      return
   }

   list, total, err := models.NewOutsourcingOrderSearch().SetEnterpriseID(companyId).SetPage(params.Page, params.PageSize).SetKeyword(params.Keyword).SetOrder("id desc").Find()
   if err != nil {
      util.ResponseFormat(c, code.InternalError, "查询错误")
      return
@@ -45,7 +53,7 @@
// @Success   200   {object}  util.ResponseList{data=request.OutsourcingOrderOverview}  "成功"
// @Router    /api-outsourcing/v1/order/overview [get]
func (slf *OrderController) OrderOverview(c *gin.Context) {
   result, err := models.NewOutsourcingOrderSearch().CountGroupByStatus()
   result, err := models.NewOutsourcingOrderSearch().SetEnterpriseID(jwt.GetCompanyID(c)).CountGroupByStatus()
   if err != nil {
      util.ResponseFormat(c, code.InternalError, "查询错误")
      return
@@ -80,6 +88,13 @@
      util.ResponseFormat(c, code.RequestParamError, "参数缺失")
      return
   }

   _, err := models.NewOutsourcingOrderSearch().SetEnterpriseID(jwt.GetCompanyID(c)).SetID(params.OutsourcingOrderId).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "订单不存在")
      return
   }

   list, total, err := models.NewOutsourcingOrderProductSearch().SetPage(params.Page, params.PageSize).SetOutsourcingOrderID(params.OutsourcingOrderId).SetOrder("id desc").Find()
   if err != nil {
      util.ResponseFormat(c, code.InternalError, "查询错误")
@@ -105,8 +120,15 @@
      util.ResponseFormat(c, code.RequestParamError, "物料申请不能为空")
      return
   }

   _, err := models.NewOutsourcingOrderSearch().SetEnterpriseID(jwt.GetCompanyID(c)).SetNumber(params.ApplyList[0].OutsourcingOrderNumber).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "订单不存在")
      return
   }

   var apply []*models.OutsourcingMaterialApply
   err := structx.AssignTo(params.ApplyList, &apply)
   err = structx.AssignTo(params.ApplyList, &apply)
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "数据转换失败")
      return
@@ -141,6 +163,13 @@
      util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
      return
   }

   _, err := models.NewOutsourcingOrderSearch().SetEnterpriseID(jwt.GetCompanyID(c)).SetNumber(params.Number).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "订单不存在")
      return
   }

   find, err := models.NewOutsourcingMaterialApplySearch().SetOutsourcingOrderNumber(params.Number).FindNotTotal()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "查询失败")
@@ -162,6 +191,13 @@
      util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
      return
   }

   _, err := models.NewOutsourcingOrderSearch().SetEnterpriseID(jwt.GetCompanyID(c)).SetNumber(params.OutsourcingOrderNumber).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "订单不存在")
      return
   }

   //限定状态参数
   if params.Status == constvar.OutsourcingOrderStatusCreate || //取消确认
      params.Status == constvar.OutsourcingOrderStatusWaitProduce || //确认接受
@@ -196,6 +232,12 @@
   }
   if params.OutsourcingOrderID == 0 {
      util.ResponseFormat(c, code.RequestParamError, "参数缺失")
      return
   }

   _, err := models.NewOutsourcingOrderSearch().SetEnterpriseID(jwt.GetCompanyID(c)).SetID(params.OutsourcingOrderID).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "订单不存在")
      return
   }

@@ -236,6 +278,13 @@
      util.ResponseFormat(c, code.RequestParamError, "参数缺失")
      return
   }

   _, err := models.NewOutsourcingOrderSearch().SetEnterpriseID(jwt.GetCompanyID(c)).SetID(params.OutsourcingOrderID).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "订单不存在")
      return
   }

   if len(params.DeliveryList) == 0 {
      util.ResponseFormat(c, code.RequestParamError, "发货列表不能为空")
      return
@@ -248,8 +297,8 @@
      WaybillNumber:      params.WaybillNumber,
   }

   err := models.WithTransaction(func(db *gorm.DB) error {
      err := models.NewOutsourcingOrderDeliverySearch().SetOrm(db).Create(delivery)
   err = models.WithTransaction(func(db *gorm.DB) error {
      err = models.NewOutsourcingOrderDeliverySearch().SetOrm(db).Create(delivery)
      if err != nil {
         return err
      }
@@ -285,6 +334,13 @@
      util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
      return
   }

   _, err := models.NewOutsourcingOrderSearch().SetEnterpriseID(jwt.GetCompanyID(c)).SetID(params.OutsourcingOrderID).First()
   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "订单不存在")
      return
   }

   list, err := models.NewOutsourcingOrderDeliveryDetailsSearch().
      SetOutsourcingOrderID(params.OutsourcingOrderID).
      SetPreload(true).