fix

zhangqian

2024-07-04 f0786e779666d67a50cd57d82cfe1e278b47d397

 controllers/product_controller.go

@@ -7,6 +7,8 @@
   "github.com/shopspring/decimal"
   "github.com/spf13/cast"
   "gorm.io/gorm"
   "io"
   "net/url"
   "strconv"
   "time"
   "wms/constvar"
@@ -20,6 +22,7 @@
   "wms/request"
   "wms/response"
   "wms/service"
   "wms/utils/http"
)

type ProductController struct {
@@ -446,7 +449,16 @@
      util.ResponseFormat(c, code.RequestParamError, "无效id")
      return
   }
   err := models.NewMaterialSearch().SetID(id).Delete()
   err := models.WithTransaction(func(tx *gorm.DB) error {
      if err := models.NewMaterialSearch().SetOrm(tx).SetID(id).Delete(); err != nil {
         return err
      }
      if err := models.NewAttributeValueSearch().SetOrm(tx).SetEntityID(id).Delete(); err != nil { //删除物料对应的动态属性
         return err
      }
      return nil
   })

   if err != nil {
      util.ResponseFormat(c, code.RequestParamError, "删除失败")
      return
@@ -1036,8 +1048,8 @@

// InputProduct
//
//   @Tags      物料管理
//   @Summary   导入物料
//   @Tags      产品
//   @Summary   导入物料/产品
//
// @Accept multipart/form-data
// @Param file formData file true "file"
@@ -1063,3 +1075,40 @@
   resp.InputCount = insertCount
   util.ResponseFormat(c, code.Success, resp)
}

// DownloadInputFormat
//
//   @Tags      产品
//   @Summary   下载导入物料/产品模板
//
// @Param     Authorization   header string true "token"
//
//   @Success   200      {object}   util.Response    "成功"
//   @Router      /api-wms/v1/product/downloadInputFormat [get]
func (slf ProductController) DownloadInputFormat(c *gin.Context) {
   template, err := models.NewFileTemplateAttachmentSearch().SetCategory(constvar.FileWarehouseCategory_JialianInput3).First() //物料导入模板
   if err != nil {
      util.ResponseFormat(c, code.NoTemplateError, "获取模版记录失败:"+err.Error())
      return
   }
   readerCloser, err := http.HttpGetWithReadCloser(template.FileUrl)
   if err != nil {
      util.ResponseFormat(c, code.NoTemplateError, "获取模版记录失败:"+err.Error())
      return
   }

   fileContentDisposition := "attachment;filename=\"" + url.QueryEscape("物料导入.xlsx") + "\""
   w := c.Writer
   w.Header().Set("Content-Type", "application/octet-stream")
   w.Header().Set("Content-Disposition", fileContentDisposition)
   w.Header().Set("Content-Transfer-Encoding", "binary")
   w.Header().Set("Cache-Control", "no-cache")
   _, err = io.Copy(w, readerCloser)
   if err != nil {
      util.ResponseFormat(c, code.NoTemplateError, "下载失败:"+err.Error())
      return
   }
   w.Flush()
   _ = readerCloser.Close()
   util.ResponseFormat(c, code.Success, "")
}