From 44b1064dcfc3d88c0eba7dca09c420ee64483f5a Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 25 六月 2024 14:37:20 +0800
Subject: [PATCH] fix transaction

---
 service/operation.go |   20 ++++++++++----------
 1 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/service/operation.go b/service/operation.go
index bdce573..d276fe6 100644
--- a/service/operation.go
+++ b/service/operation.go
@@ -139,11 +139,11 @@
 
 			if locAmount, aok := mapLocAmount[strconv.Itoa(roleProduct.LocationId)+v.ProductId]; aok {
 				locAmount.Amount = locAmount.Amount.Add(v.Amount)
-				if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
-					return res.Error
+				if err := models.NewLocationProductAmountSearch().SetOrm(tx).SetID(int(locAmount.ID)).Save(locAmount); err != nil {
+					return err
 				}
 			} else {
-				if err := models.NewLocationProductAmountSearch().Create(&models.LocationProductAmount{
+				if err := models.NewLocationProductAmountSearch().SetOrm(tx).Create(&models.LocationProductAmount{
 					LocationId:        roleProduct.LocationId,
 					ProductCategoryID: v.Product.CategoryId,
 					ProductId:         v.ProductId,
@@ -165,11 +165,11 @@
 
 				if locAmount, aok := mapLocAmount[strconv.Itoa(roleCategory.LocationId)+v.ProductId]; aok {
 					locAmount.Amount = locAmount.Amount.Add(v.Amount)
-					if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
-						return res.Error
+					if err := models.NewLocationProductAmountSearch().SetOrm(tx).SetID(int(locAmount.ID)).Save(locAmount); err != nil {
+						return err
 					}
 				} else {
-					if err := models.NewLocationProductAmountSearch().Create(&models.LocationProductAmount{
+					if err := models.NewLocationProductAmountSearch().SetOrm(tx).Create(&models.LocationProductAmount{
 						LocationId:        roleCategory.LocationId,
 						ProductCategoryID: v.Product.CategoryId,
 						ProductId:         v.ProductId,
@@ -182,11 +182,11 @@
 			} else {
 				if locAmount, aok := mapLocAmount[strconv.Itoa(operation.LocationID)+v.ProductId]; aok {
 					locAmount.Amount = locAmount.Amount.Add(v.Amount)
-					if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
-						return res.Error
+					if err := models.NewLocationProductAmountSearch().SetOrm(tx).SetID(int(locAmount.ID)).Save(locAmount); err != nil {
+						return err
 					}
 				} else {
-					if err := models.NewLocationProductAmountSearch().Create(&models.LocationProductAmount{
+					if err := models.NewLocationProductAmountSearch().SetOrm(tx).Create(&models.LocationProductAmount{
 						LocationId:        operation.LocationID,
 						ProductCategoryID: v.Product.CategoryId,
 						ProductId:         v.ProductId,
@@ -240,7 +240,7 @@
 				return errors.New(fmt.Sprintf("浜у搧锛�%v,搴撳瓨锛�%v,鍑哄簱锛�%v,鏁伴噺涓嶅锛屾棤娉曞畬鎴愬嚭搴撴搷浣�", v.Product.Name, locAmount.Amount.String(), v.Amount.String()))
 			}
 			locAmount.Amount = locAmount.Amount.Sub(v.Amount)
-			if err := models.NewLocationProductAmountSearch().SetID(locAmount.Id).Update(locAmount); err != nil {
+			if err := models.NewLocationProductAmountSearch().SetOrm(tx).SetID(locAmount.Id).Save(locAmount); err != nil {
 				return err
 			}
 		} else {

--
Gitblit v1.8.0