From 357321d3e35ee729d297b96f5168dd7b6f41b10c Mon Sep 17 00:00:00 2001
From: liuchen864 <23082234@qq.com>
Date: Mon, 18 Dec 2023 13:53:44 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9Bug=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/transaction/TransactionServiceImpl.java  | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/win-module-wms/win-module-wms-biz/src/main/java/com/win/module/wms/service/transaction/TransactionServiceImpl.java b/win-module-wms/win-module-wms-biz/src/main/java/com/win/module/wms/service/transaction/TransactionServiceImpl.java
index ac74b50c..5888af49 100644
--- a/win-module-wms/win-module-wms-biz/src/main/java/com/win/module/wms/service/transaction/TransactionServiceImpl.java
+++ b/win-module-wms/win-module-wms-biz/src/main/java/com/win/module/wms/service/transaction/TransactionServiceImpl.java
@@ -12,9 +12,11 @@ import com.win.module.wms.controller.transaction.vo.TransactionPageReqVO;
 import com.win.module.wms.controller.transferlog.vo.TransferlogCreateReqVO;
 import com.win.module.wms.convert.transaction.TransactionConvert;
 import com.win.module.wms.convert.transferlog.TransferlogConvert;
+import com.win.module.wms.dal.dataobject.location.LocationDO;
 import com.win.module.wms.dal.dataobject.transaction.TransactionDO;
 import com.win.module.wms.dal.dataobject.transferlog.TransferlogDO;
 import com.win.module.wms.dal.mysql.balance.BalanceMapper;
+import com.win.module.wms.dal.mysql.location.LocationMapper;
 import com.win.module.wms.dal.mysql.transaction.TransactionMapper;
 import com.win.module.wms.dal.mysql.transferlog.TransferlogMapper;
 import com.win.module.wms.service.rule.RuleService;
@@ -37,16 +39,14 @@ public class TransactionServiceImpl implements TransactionService {
 
     @Resource
     private TransactionMapper transactionMapper;
-
     @Resource
     private BalanceMapper balanceMapper;
-
     @Resource
     private TransferlogMapper transferlogMapper;
-
+    @Resource
+    private LocationMapper locationMapper;
     @Resource
     private RuleService ruleService;
-
     @Resource
     private SerialNumberApi serialNumberApi;
 
@@ -62,6 +62,10 @@ public class TransactionServiceImpl implements TransactionService {
             result += transactionMapper.insert(transactionDO);
             BalanceCreateReqVO balanceCreateReqVO = new BalanceCreateReqVO();
             BeanUtils.copyProperties(createReqVO, balanceCreateReqVO);
+            LocationDO locationDO = locationMapper.selectByCode(balanceCreateReqVO.getLocationCode());
+            balanceCreateReqVO.setWarehouseCode(locationDO.getWarehouseCode());
+            balanceCreateReqVO.setAreaCode(locationDO.getAreaCode());
+            balanceCreateReqVO.setLocationGroupCode(locationDO.getLocationGroupCode());
             balanceCreateReqVO.setLastTransNumber(number);
             RuleRespVO ruleRespVO = ruleService.warehouseStorage(createReqVO.getLocationCode());
             balanceMapper.updateBalanceQty(balanceCreateReqVO, ruleRespVO, createReqVO.getInventoryAction());