From 84d2953e7b641600779e7aecc80bdd35e96d31d6 Mon Sep 17 00:00:00 2001
From: jiangshuai <291802688@qq.com>
Date: 星期二, 06 二月 2024 10:13:12 +0800
Subject: [PATCH] 拦截器

---
 api/v1/purchase/purchase.go |   11 +++++------
 1 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/api/v1/purchase/purchase.go b/api/v1/purchase/purchase.go
index bd0a9d1..e723875 100644
--- a/api/v1/purchase/purchase.go
+++ b/api/v1/purchase/purchase.go
@@ -2,7 +2,6 @@
 
 import (
 	"context"
-	"fmt"
 	"github.com/gin-gonic/gin"
 	"go.uber.org/zap"
 	"gorm.io/gorm"
@@ -13,8 +12,6 @@
 	"srm/proto/purchase_wms"
 	"strconv"
 	"strings"
-	"time"
-
 	//"srm/model/purchase"
 
 	//"srm/model/purchase"
@@ -54,7 +51,6 @@
 	purchaseRecord.Status = purchase.OrderStatusConfirmed
 	purchaseRecord.HandledBy = "admin"
 	purchaseRecord.Creator = "admin"
-	purchaseRecord.Number = fmt.Sprintf("CG%v", time.Now().Unix())
 	purchaseRecord.Principal = "admin"
 	purchaseRecord.OrderType = "閲囪喘璁㈠崟"
 
@@ -241,6 +237,7 @@
 		response.FailWithMessage(err.Error(), c)
 		return
 	}
+	warehouse := ""
 	if params.Status == purchase.OrderStatusReceived {
 		data, err := service.NewPurchaseService().GetPurchase(uint(params.Id))
 		if err != nil {
@@ -262,19 +259,21 @@
 			product = append(product, &p)
 		}
 		client := purchase_wms.NewPurchaseServiceClient(purchase_wms.PurchaseConn)
-		_, err = client.PurchaseToWms(context.Background(), &purchase_wms.PurchaseToWmsRequest{
+		resp, err := client.PurchaseToWms(context.Background(), &purchase_wms.PurchaseToWmsRequest{
 			Number:       data.Number,
 			SupplierName: data.Supplier.Name,
 			Product:      product,
+			Source:       "SRM_PURCHASE",
 		})
 		if err != nil {
 			global.GVA_LOG.Error("grpc璋冪敤澶辫触!", zap.Error(err))
 			response.FailWithMessage("grpc璋冪敤澶辫触", c)
 			return
 		}
+		warehouse = resp.Warehouse
 	}
 
-	err = service.NewPurchaseService().Submit(params.Id, params.Status)
+	err = service.NewPurchaseService().Submit(params.Id, params.Status, warehouse)
 	if err != nil {
 		global.GVA_LOG.Error("鏇存柊澶辫触!", zap.Error(err))
 		response.FailWithMessage("鏇存柊澶辫触", c)

--
Gitblit v1.8.0