From a8914a16b23e93f6bfd12bcfd5cbe8b24cf7eb84 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期日, 07 四月 2024 21:49:01 +0800
Subject: [PATCH] 发货支持仓库传参

---
 proto/product_inventory/server.go |   12 ++++++++++++
 1 files changed, 12 insertions(+), 0 deletions(-)

diff --git a/proto/product_inventory/server.go b/proto/product_inventory/server.go
index 1113c11..4bf5368 100644
--- a/proto/product_inventory/server.go
+++ b/proto/product_inventory/server.go
@@ -351,6 +351,9 @@
 	inputLocationAmountMap := make(map[int64]map[string]decimal.Decimal)
 	outputLocationAmountMap := make(map[int64]map[string]decimal.Decimal)
 	for _, v := range outputList {
+		if req.WarehouseId != 0 && v.WareHouseID != req.WarehouseId {
+			continue
+		}
 		if outputProductMap[v.Number] == nil {
 			simpleInfo := &OutputSimpleInfo{
 				Number: v.Number,
@@ -364,6 +367,9 @@
 		}
 	}
 	for _, v := range inputList {
+		if req.WarehouseId != 0 && v.WareHouseID != req.WarehouseId {
+			continue
+		}
 		if inputProductMap[v.Number] == nil {
 			storeInfo := &StoreInfo{
 				Number: v.Number,
@@ -402,6 +408,9 @@
 
 	LocationIDWarehouseIDMap := make(map[int64]int64)
 	for _, output := range outputList {
+		if req.WarehouseId != 0 && output.WareHouseID != req.WarehouseId {
+			continue
+		}
 		if outputLocationAmountMap[output.LocationID] == nil {
 			outputLocationAmountMap[output.LocationID] = make(map[string]decimal.Decimal)
 		}
@@ -409,6 +418,9 @@
 		outputLocationAmountMap[output.LocationID][output.Number] = outputAmount
 	}
 	for _, input := range inputList {
+		if req.WarehouseId != 0 && input.WareHouseID != req.WarehouseId {
+			continue
+		}
 		LocationIDWarehouseIDMap[input.LocationID] = input.WareHouseID
 
 		if inputLocationAmountMap[input.LocationID] == nil {

--
Gitblit v1.8.0