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