优化

src/WMS.Web.Api/Controllers/InStockTaskController.cs

@@ -31,14 +31,17 @@ namespace WMS.Web.Api.Controllers
         private readonly IInStockService _inStockService;
         private readonly IInStockTaskService _inStockTaskService;
         private readonly IInStockTaskRepositories _inStockTaskRepositories;
+        private readonly IInventoryDetailsRepositories _inventoryDetailsRepositories;
         public InStockTaskController(
             IOptions<QiniuOptions> option,
             IExportExcelService exportExcelService,
             ILoginService loginService,
             IInStockTaskService inStockTaskService,
             IInStockTaskRepositories inStockTaskRepositories,
+            IInventoryDetailsRepositories inventoryDetailsRepositories,
             IInStockService inStockService)
         {
+            _inventoryDetailsRepositories = inventoryDetailsRepositories;
             this._option = option?.Value;
             this._exportExcelService = exportExcelService;
             this._loginService = loginService;
@@ -170,6 +173,16 @@ namespace WMS.Web.Api.Controllers
                 return ResultList<SourceBillNoQueryResponse>.ReFailure(ResultCodes.Token_Invalid_Error);
 
             var list= await _inStockTaskRepositories.GetListBy(new InStockTaskBillNoQueryRequest() { SourceBillNo = sourceBillNo});
+
+            //获取物料对应的仓位
+            var materialNumbers = list.GroupBy(x => x.MaterialNumber).Select(x => x.Key).ToList();
+            var stockCodes = list.GroupBy(x => x.StockCode).Select(x => x.Key).ToList();
+            var materialSubStocks= await _inventoryDetailsRepositories.GetMaterialSubStocks(materialNumbers, stockCodes, loginInfo.UserInfo.CompanyId);
+            list.ForEach(x => 
+            {
+                x.MaterialSubStocks = materialSubStocks.Where(t => t.MaterialNumber == x.MaterialNumber).ToList();
+            });
+
             return ResultList<SourceBillNoQueryResponse>.ReSuccess(list);
         }