From 57c1d91e82ec4085ccedf099063cd14e609a301f Mon Sep 17 00:00:00 2001
From: jiangshuai <291802688@qq.com>
Date: 星期三, 15 十一月 2023 16:36:50 +0800
Subject: [PATCH] Merge branch 'master' of http://192.168.5.5:10010/r/aps/WMS

---
 controllers/operation.go |   32 ++++++++++++++++++++++++++++----
 1 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/controllers/operation.go b/controllers/operation.go
index 72eaadd..d7912aa 100644
--- a/controllers/operation.go
+++ b/controllers/operation.go
@@ -23,6 +23,7 @@
 	"wms/pkg/logx"
 	"wms/pkg/structx"
 	"wms/proto/product_inventory"
+	"wms/proto/purchase_wms"
 	"wms/request"
 )
 
@@ -488,6 +489,9 @@
 					return err
 				}
 			}
+			if operation.SourceNumber != "" {
+				go UpdatePurchaseStatus(operation.SourceNumber)
+			}
 
 		}
 
@@ -530,6 +534,9 @@
 				if err := models.NewLocationProductAmountSearch().SetID(locAmount.Id).Update(locAmount); err != nil {
 					return err
 				}
+			}
+			if operation.SourceNumber != "" {
+				go UpdateSalesDetailStatus(operation.SourceNumber)
 			}
 		}
 
@@ -684,17 +691,23 @@
 		util.ResponseFormat(c, code.RequestError, err.Error())
 		return
 	}
-	if operation.SourceNumber != "" {
-		go UpdateSalesDetailStatus(operation.SourceNumber)
-	}
+
 	util.ResponseFormat(c, code.Success, "鎿嶄綔鎴愬姛")
 }
 
-var ProductInventoryServiceConn *grpc.ClientConn
+var (
+	ProductInventoryServiceConn *grpc.ClientConn
+	PurchaseServiceConn         *grpc.ClientConn
+)
 
 func InitProductInventoryServiceConn() {
 	var err error
 	ProductInventoryServiceConn, err = grpc.Dial(conf.GrpcServerConf.CrmAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
+	if err != nil {
+		logx.Errorf("grpc dial product service error: %v", err.Error())
+		return
+	}
+	PurchaseServiceConn, err = grpc.Dial(conf.GrpcServerConf.SrmAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
 	if err != nil {
 		logx.Errorf("grpc dial product service error: %v", err.Error())
 		return
@@ -704,6 +717,9 @@
 func CloseProductInventoryServiceConn() {
 	if ProductInventoryServiceConn != nil {
 		ProductInventoryServiceConn.Close()
+	}
+	if PurchaseServiceConn != nil {
+		PurchaseServiceConn.Close()
 	}
 }
 
@@ -718,6 +734,14 @@
 	}
 }
 
+func UpdatePurchaseStatus(number string) {
+	client := purchase_wms.NewPurchaseServiceClient(PurchaseServiceConn)
+	_, err := client.UpdatePurchaseStatus(context.Background(), &purchase_wms.UpdatePurchaseStatusRequest{Number: number})
+	if err != nil {
+		logx.Errorf("grpc dial UpdatePurchaseStatus service error: %v", err)
+	}
+}
+
 // ListTransfer
 // @Tags      鍏ュ簱/鍑哄簱
 // @Summary   搴撳瓨璋冩嫧鍒楄〃

--
Gitblit v1.8.0