From d7d23ea22c8ab9a2756e307cc6aa4cdff3731da7 Mon Sep 17 00:00:00 2001
From: "JinLu.Yin" <411641505@qq.com>
Date: Thu, 15 Sep 2022 23:28:07 +0800
Subject: [PATCH] =?UTF-8?q?=E7=94=9F=E4=BA=A7=E9=80=80=E6=96=99=EF=BC=8C?=
 =?UTF-8?q?=E4=BB=93=E5=BA=93=E9=BB=98=E8=AE=A4=E5=80=BC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../wm/controller/WmRtIssueController.java    | 51 ++++++++++++++++++-
 1 file changed, 49 insertions(+), 2 deletions(-)

diff --git a/ktg-mes/src/main/java/com/ktg/mes/wm/controller/WmRtIssueController.java b/ktg-mes/src/main/java/com/ktg/mes/wm/controller/WmRtIssueController.java
index 36c7005..e69edba 100644
--- a/ktg-mes/src/main/java/com/ktg/mes/wm/controller/WmRtIssueController.java
+++ b/ktg-mes/src/main/java/com/ktg/mes/wm/controller/WmRtIssueController.java
@@ -4,7 +4,11 @@ import java.util.List;
 import javax.servlet.http.HttpServletResponse;
 
 import com.ktg.common.constant.UserConstants;
-import com.ktg.mes.wm.service.IWmRtIssueLineService;
+import com.ktg.common.utils.StringUtils;
+import com.ktg.mes.wm.domain.WmStorageArea;
+import com.ktg.mes.wm.domain.WmStorageLocation;
+import com.ktg.mes.wm.domain.WmWarehouse;
+import com.ktg.mes.wm.service.*;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.transaction.annotation.Transactional;
@@ -21,7 +25,6 @@ import com.ktg.common.core.controller.BaseController;
 import com.ktg.common.core.domain.AjaxResult;
 import com.ktg.common.enums.BusinessType;
 import com.ktg.mes.wm.domain.WmRtIssue;
-import com.ktg.mes.wm.service.IWmRtIssueService;
 import com.ktg.common.utils.poi.ExcelUtil;
 import com.ktg.common.core.page.TableDataInfo;
 
@@ -41,6 +44,18 @@ public class WmRtIssueController extends BaseController
     @Autowired
     private IWmRtIssueLineService wmRtIssueLineService;
 
+    @Autowired
+    private IWmWarehouseService wmWarehouseService;
+
+    @Autowired
+    private IWmStorageLocationService wmStorageLocationService;
+
+    @Autowired
+    private IWmStorageAreaService wmStorageAreaService;
+
+    @Autowired
+    private IStorageCoreService storageCoreService;
+
     /**
      * 查询生产退料单头列表
      */
@@ -87,6 +102,22 @@ public class WmRtIssueController extends BaseController
         if(UserConstants.NOT_UNIQUE.equals(wmRtIssueService.checkUnique(wmRtIssue))){
             return AjaxResult.error("退料单编号已存在");
         }
+
+        if(StringUtils.isNotNull(wmRtIssue.getWarehouseId())){
+            WmWarehouse warehouse = wmWarehouseService.selectWmWarehouseByWarehouseId(wmRtIssue.getWarehouseId());
+            wmRtIssue.setWarehouseCode(warehouse.getWarehouseCode());
+            wmRtIssue.setWarehouseName(warehouse.getWarehouseName());
+        }
+        if(StringUtils.isNotNull(wmRtIssue.getLocationId())){
+            WmStorageLocation location = wmStorageLocationService.selectWmStorageLocationByLocationId(wmRtIssue.getLocationId());
+            wmRtIssue.setLocationCode(location.getLocationCode());
+            wmRtIssue.setLocationName(location.getLocationName());
+        }
+        if(StringUtils.isNotNull(wmRtIssue.getAreaId())){
+            WmStorageArea area = wmStorageAreaService.selectWmStorageAreaByAreaId(wmRtIssue.getAreaId());
+            wmRtIssue.setAreaCode(area.getAreaCode());
+            wmRtIssue.setAreaName(area.getAreaName());
+        }
         return toAjax(wmRtIssueService.insertWmRtIssue(wmRtIssue));
     }
 
@@ -101,6 +132,22 @@ public class WmRtIssueController extends BaseController
         if(UserConstants.NOT_UNIQUE.equals(wmRtIssueService.checkUnique(wmRtIssue))){
             return AjaxResult.error("退料单编号已存在");
         }
+
+        if(StringUtils.isNotNull(wmRtIssue.getWarehouseId())){
+            WmWarehouse warehouse = wmWarehouseService.selectWmWarehouseByWarehouseId(wmRtIssue.getWarehouseId());
+            wmRtIssue.setWarehouseCode(warehouse.getWarehouseCode());
+            wmRtIssue.setWarehouseName(warehouse.getWarehouseName());
+        }
+        if(StringUtils.isNotNull(wmRtIssue.getLocationId())){
+            WmStorageLocation location = wmStorageLocationService.selectWmStorageLocationByLocationId(wmRtIssue.getLocationId());
+            wmRtIssue.setLocationCode(location.getLocationCode());
+            wmRtIssue.setLocationName(location.getLocationName());
+        }
+        if(StringUtils.isNotNull(wmRtIssue.getAreaId())){
+            WmStorageArea area = wmStorageAreaService.selectWmStorageAreaByAreaId(wmRtIssue.getAreaId());
+            wmRtIssue.setAreaCode(area.getAreaCode());
+            wmRtIssue.setAreaName(area.getAreaName());
+        }
         return toAjax(wmRtIssueService.updateWmRtIssue(wmRtIssue));
     }