From 504f3e9bc274bcd6c7221e738802e0e5d4aef36f Mon Sep 17 00:00:00 2001
From: lishihai <dslsh@dscom>
Date: 星期四, 20 六月 2024 17:07:02 +0800
Subject: [PATCH] 查询属性值和对象 通过主键ID查询

---
 controllers/product_controller.go |  149 ++++++++++++++++++++++++++++++++++++-------------
 1 files changed, 110 insertions(+), 39 deletions(-)

diff --git a/controllers/product_controller.go b/controllers/product_controller.go
index 606e381..cb0e7fc 100644
--- a/controllers/product_controller.go
+++ b/controllers/product_controller.go
@@ -18,6 +18,8 @@
 	"wms/pkg/mysqlx"
 	"wms/pkg/structx"
 	"wms/request"
+	"wms/response"
+	"wms/service"
 )
 
 type ProductController struct {
@@ -53,22 +55,40 @@
 		return
 	}
 	//params.ID = utils.GetUUID()
-	err := models.NewMaterialSearch().Create(&params)
+
+	err := models.WithTransaction(func(tx *gorm.DB) error {
+		if err := models.NewMaterialSearch().SetOrm(tx).Create(&params); err != nil {
+			return err
+		}
+		materialAttachmentList := []*models.MaterialAttachment{}
+		for _, v := range params.AttachmentIDs {
+			ma := &models.MaterialAttachment{MaterialID: params.ID, AttachmentID: v}
+			materialAttachmentList = append(materialAttachmentList, ma)
+		}
+		if len(materialAttachmentList) > 0 {
+			if err := models.NewMaterialAttachmentSearch().SetOrm(tx).CreateBatch(materialAttachmentList); err != nil {
+				return err
+			}
+		}
+		avs := make([]*models.AttributeValue, 0)
+		for _, v := range params.Attributes {
+			av := models.AttributeValue{
+				Model:       gorm.Model{},
+				EntityID:    params.ID,
+				AttributeID: v.ID,
+				Value:       v.Value,
+			}
+			avs = append(avs, &av)
+
+		}
+		if err := models.NewAttributeValueSearch().SetOrm(tx).CreateBatch(avs); err != nil {
+			return err
+		}
+		return nil
+	})
 	if err != nil {
 		util.ResponseFormat(c, code.RequestParamError, "浜у搧淇℃伅淇濆瓨澶辫触")
 		return
-	}
-
-	materialAttachmentList := []*models.MaterialAttachment{}
-	for _, v := range params.AttachmentIDs {
-		ma := &models.MaterialAttachment{MaterialID: params.ID, AttachmentID: v}
-		materialAttachmentList = append(materialAttachmentList, ma)
-	}
-	if len(materialAttachmentList) > 0 {
-		if err := models.NewMaterialAttachmentSearch().CreateBatch(materialAttachmentList); err != nil {
-			util.ResponseFormat(c, code.SaveFail, "闄勪欢淇濆瓨澶辫触")
-			return
-		}
 	}
 
 	util.ResponseFormat(c, code.Success, "淇濆瓨鎴愬姛")
@@ -216,38 +236,64 @@
 		util.ResponseFormat(c, code.RequestParamError, "浜у搧淇℃伅鏇存柊澶辫触")
 		return
 	}
-
-	//鏇存柊鍦ㄥ簱鐨勪骇鍝佺被鍨�
-	if params.CategoryId > 0 {
-		find, _ := models.NewLocationProductAmountSearch().SetProductId(params.ID).Find()
-		if len(find) > 0 {
-			var ids []int
-			for _, f := range find {
-				if f.ProductCategoryID != params.CategoryId {
-					ids = append(ids, f.Id)
+	err = models.WithTransaction(func(tx *gorm.DB) error {
+		//鏇存柊鍦ㄥ簱鐨勪骇鍝佺被鍨�
+		if params.CategoryId > 0 {
+			find, _ := models.NewLocationProductAmountSearch().SetProductId(params.ID).Find()
+			if len(find) > 0 {
+				var ids []int
+				for _, f := range find {
+					if f.ProductCategoryID != params.CategoryId {
+						ids = append(ids, f.Id)
+					}
 				}
+				m := make(map[string]interface{})
+				m["productCategoryId"] = params.CategoryId
+				_ = models.NewLocationProductAmountSearch().SetOrm(tx).SetIds(ids).UpdateByMap(m)
 			}
-			m := make(map[string]interface{})
-			m["productCategoryId"] = params.CategoryId
-			_ = models.NewLocationProductAmountSearch().SetIds(ids).UpdateByMap(m)
 		}
-	}
+		materialAttachmentList := []*models.MaterialAttachment{}
+		for _, v := range params.AttachmentIDs {
+			ma := &models.MaterialAttachment{MaterialID: params.ID, AttachmentID: v}
+			materialAttachmentList = append(materialAttachmentList, ma)
+		}
+		if err := models.NewMaterialAttachmentSearch().SetOrm(tx).SetMaterialID(params.ID).Delete(); err != nil {
+			util.ResponseFormat(c, code.RequestParamError, "浜у搧闄勪欢娓呴櫎澶辫触")
+			return err
+		}
+		if len(materialAttachmentList) > 0 {
+			if err := models.NewMaterialAttachmentSearch().SetOrm(tx).CreateBatch(materialAttachmentList); err != nil {
+				util.ResponseFormat(c, code.RequestParamError, "浜у搧淇℃伅鏇存柊澶辫触")
+				return err
+			}
+		}
 
-	materialAttachmentList := []*models.MaterialAttachment{}
-	for _, v := range params.AttachmentIDs {
-		ma := &models.MaterialAttachment{MaterialID: params.ID, AttachmentID: v}
-		materialAttachmentList = append(materialAttachmentList, ma)
-	}
-	if err := models.NewMaterialAttachmentSearch().SetMaterialID(params.ID).Delete(); err != nil {
-		util.ResponseFormat(c, code.RequestParamError, "浜у搧闄勪欢娓呴櫎澶辫触")
+		attributeValueSearch := models.NewAttributeValueSearch()
+		//鍒犻櫎浠庢柊鎻掑叆
+		if err := attributeValueSearch.SetOrm(tx).SetEntityID(params.ID).Delete(); err != nil {
+			return err
+		}
+		avs := make([]*models.AttributeValue, 0)
+		for _, v := range params.Attributes {
+			av := models.AttributeValue{
+				Model:       gorm.Model{},
+				EntityID:    params.ID,
+				AttributeID: v.ID,
+				Value:       v.Value,
+			}
+			avs = append(avs, &av)
+
+		}
+		if err := models.NewAttributeValueSearch().SetOrm(tx).CreateBatch(avs); err != nil {
+			return err
+		}
+		return nil
+	})
+	if err != nil {
+		util.ResponseFormat(c, code.RequestParamError, "浜у搧淇℃伅鏇存柊澶辫触")
 		return
 	}
-	if len(materialAttachmentList) > 0 {
-		if err := models.NewMaterialAttachmentSearch().CreateBatch(materialAttachmentList); err != nil {
-			util.ResponseFormat(c, code.RequestParamError, "浜у搧淇℃伅鏇存柊澶辫触")
-			return
-		}
-	}
+
 	util.ResponseFormat(c, code.Success, "鏇存柊鎴愬姛")
 }
 
@@ -830,3 +876,28 @@
 
 	util.ResponseFormat(c, code.Success, "娣诲姞鎴愬姛")
 }
+
+// InputProduct
+//
+//	@Tags		鐗╂枡绠$悊
+//	@Summary	瀵煎叆鐗╂枡
+//	@Produce	application/xlsx
+//	@Success	200		{object}	util.Response 	"鎴愬姛"
+//	@Router		/api-wms/v1/product/inputProduct [post]
+func (slf ProductController) InputProduct(c *gin.Context) {
+	file, _, err := c.Request.FormFile("file")
+	if err != nil {
+		util.ResponseFormat(c, code.RequestParamError, err.Error())
+		return
+	}
+	defer file.Close()
+	resp := response.MaterialInputRes{InputCount: 0, ErrCount: 0, FileAddress: ""}
+	userInfo := middleware.GetUserInfo(c)
+	insertCount, err := service.InputMaterial(file, userInfo.Username)
+	if err != nil {
+		util.ResponseFormat(c, code.RequestParamError, err.Error())
+		return
+	}
+	resp.InputCount = insertCount
+	util.ResponseFormat(c, code.Success, resp)
+}

--
Gitblit v1.8.0