From b84057ed929c035b6f70ae6a0079e5c440bc93da Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期六, 25 十一月 2023 17:57:01 +0800
Subject: [PATCH] srm插入入库操作修改

---
 proto/purchase_wms/server.go |   20 ++++++++++++++++++--
 1 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/proto/purchase_wms/server.go b/proto/purchase_wms/server.go
index 173e7db..bdae0cd 100644
--- a/proto/purchase_wms/server.go
+++ b/proto/purchase_wms/server.go
@@ -3,6 +3,7 @@
 import (
 	"context"
 	"github.com/shopspring/decimal"
+	"gorm.io/gorm"
 	"strconv"
 	"time"
 	"wms/constvar"
@@ -49,6 +50,21 @@
 		detail.Amount = decimal.NewFromInt(product.Amount)
 		details = append(details, &detail)
 	}
-	err = models.NewOperationSearch().Create(&operation)
-	return new(PurchaseToWmsResponse), err
+	err = models.WithTransaction(func(db *gorm.DB) error {
+		err := models.NewOperationSearch().SetOrm(db).Create(&operation)
+		if err != nil {
+			return err
+		}
+		for _, detail := range details {
+			detail.OperationID = operation.Id
+		}
+		err = models.NewOperationDetailsSearch().SetOrm(db).CreateBatch(details)
+		return err
+	})
+	if err != nil {
+		return nil, err
+	}
+	resp := new(PurchaseToWmsResponse)
+	resp.Warehouse = warehouse.Name
+	return resp, nil
 }

--
Gitblit v1.8.0