From 2bd9ddd56697e202a443d04139f854afa2559e95 Mon Sep 17 00:00:00 2001
From: WangHan <wwh_work@126.com>
Date: 星期一, 29 四月 2024 14:26:58 +0800
Subject: [PATCH] 漏洞修改

---
 consum-base/src/main/java/com/consum/base/controller/LWhFormTransferController.java |   74 +++++++++++++++++++++----------------
 1 files changed, 42 insertions(+), 32 deletions(-)

diff --git a/consum-base/src/main/java/com/consum/base/controller/LWhFormTransferController.java b/consum-base/src/main/java/com/consum/base/controller/LWhFormTransferController.java
index 6508add..a28a846 100644
--- a/consum-base/src/main/java/com/consum/base/controller/LWhFormTransferController.java
+++ b/consum-base/src/main/java/com/consum/base/controller/LWhFormTransferController.java
@@ -13,6 +13,7 @@
 import javax.servlet.http.HttpServletResponse;
 
 import org.apache.commons.compress.utils.Lists;
+import org.apache.commons.lang3.ObjectUtils;
 import org.apache.poi.ss.usermodel.Workbook;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -23,6 +24,8 @@
 
 import com.consum.base.BaseController;
 import com.consum.base.core.WhBusinessEnum;
+import com.consum.base.core.type.StatesType;
+import com.consum.base.core.type.TransferStatesType;
 import com.consum.base.core.utils.CommonUtil;
 import com.consum.base.core.utils.IdUtil;
 import com.consum.base.pojo.GoodsUseRecordVO;
@@ -47,7 +50,8 @@
 import com.consum.base.service.LWhProcureModelService;
 import com.consum.base.service.LWhProcureModelUserRecordService;
 import com.consum.base.service.LWhProcureModelUserService;
-import com.consum.base.service.impl.LWhFormTransferCoreService;
+import com.consum.base.service.core.LWhFormTransferCoreService;
+import com.consum.model.po.BaseWarehouse;
 import com.consum.model.po.FinSysTenantUser;
 import com.consum.model.po.LWhFormTransfer;
 import com.consum.model.po.LWhProcureModel;
@@ -88,11 +92,11 @@
     @Autowired
     private LWhProcureModelUserRecordService lWhProcureModelUserRecordService;
     @Autowired
-    private BaseWarehouseService baseWarehouseService;
-    @Autowired
     private LGoodsUserRecordCoreService lGoodsUserRecordCoreService;
     @Autowired
     private LWhGoodsService lWhGoodsService;
+    @Autowired
+    private BaseWarehouseService baseWarehouseService;
 
     /**
      * @Description 鏂板
@@ -121,23 +125,21 @@
         if (transferBusinessType == 1) {
             // 鍏堝嚭搴擄紝鍐嶅叆搴�
             // 鍑哄簱鍓� 璁剧疆鍑哄簱浠撳簱
-            List<LWhProcureModel> modelByForm = lWhProcureModelService.getModelByForm(WhBusinessEnum.DIAOBO, id);
+            List<LWhProcureModel> modelByForm = lWhProcureModelService.getModelByForm(WhBusinessEnum.BUMENFENFA, id);
             Set<Long> baseModelIds =
                 modelByForm.stream().map(LWhProcureModel::getBaseGoodsModelsId).collect(Collectors.toSet());
 
-            // TODO 涓嶅悓搴撳鐞�
+            FinSysTenantUser sysTenantUser = this.getSysInfo();
+            String agencyId = sysTenantUser.getTenantId();
+            List<BaseWarehouse> baseWarehouseList =
+                baseWarehouseService.getBaseWareHouseList(Long.valueOf(agencyId), StatesType.NORMAL.getValue());
+            Set<Long> wareHouseIds = baseWarehouseList.stream().map(BaseWarehouse::getId).collect(Collectors.toSet());
             // 閫氳繃璋冩嫧鍗曚腑鐨勫瀷鍙穒d鏌ヨ鍑鸿鍨嬪彿鐗╁搧鎵�鍦ㄧ殑浠撳簱浣嶇疆
-            List<GoodsInfoDTO> goodsInfoDTOS = lWhGoodsService.queryGoodsInfo(baseModelIds);
+            List<GoodsInfoDTO> goodsInfoDTOS = lWhGoodsService.queryGoodsInfo(baseModelIds, wareHouseIds,null);
             GoodsInfoDTO goodsInfoDTO = goodsInfoDTOS.stream().findFirst().orElse(null);
-
-            // FinSysTenantUser sysTenantUser = this.getSysInfo();
-            // String agencyId = sysTenantUser.getTenantId();
-            // List<BaseWarehouse> baseWarehouseList =
-            // baseWarehouseService.getBaseWareHouseList(Long.valueOf(agencyId), (short)1, (short)1);
-            // if (CollectionUtils.isEmpty(baseWarehouseList)) {
-            // return ResponseValue.error("鏈烘瀯鏃犻粯璁や粨搴擄紒");
-            // }
-            // BaseWarehouse baseWarehouse = baseWarehouseList.get(0);
+            if (ObjectUtils.isEmpty(goodsInfoDTO)) {
+                return ResponseValue.error("璇ュ瀷鍙锋病鏈夊簱瀛樺彲浣跨敤");
+            }
 
             Long wareHouseId = goodsInfoDTO.getWarehouseId();
             String warehouseName = goodsInfoDTO.getWarehouseName();
@@ -145,7 +147,7 @@
             lWhFormTransfer.setOutWarehouseId(wareHouseId);
             lWhFormTransfer.setOutWarehouseName(warehouseName);
             lWhFormTransferService.update(lWhFormTransfer);
-            lWhFormTransferCoreService.doTransferOutPut(id, getCurrentUser());
+            lWhFormTransferCoreService.doTransferOutPutNew(id, getCurrentUser(), WhBusinessEnum.BUMENFENFA);
 
             lWhFormTransferCoreService.doTransferInPut(id, getCurrentUser(), param.getOperatorName());
         }
@@ -235,6 +237,10 @@
         if (sysInfo == null) {
             return ResponseValue.error("鐧诲綍鐢ㄦ埛淇℃伅涓嶅瓨鍦�");
         }
+        String tenantId = sysInfo.getTenantId();
+        if (param.getInAgencyId() == null) {
+            param.setInAgencyId(Long.valueOf(tenantId));
+        }
         GenericPager<Map<String, Object>> mapGenericPager = lWhFormTransferService.queryFormTransferDetailList(param);
 
         return ResponseValue.success(mapGenericPager);
@@ -253,6 +259,10 @@
         if (id == null) {
             return ResponseValue.error("鍙傛暟閿欒");
         }
+        LWhFormTransfer lWhFormTransfer = lWhFormTransferService.selectById(id);
+        if (lWhFormTransfer.getStates() != TransferStatesType.OUT_PENDING.getValue()) {
+            return ResponseValue.error("鐘舵�侀敊璇紝涓嶈兘鎾ら攢");
+        }
 
         int num = this.lWhFormTransferService.updateStatus(id);
         return num > 0 ? ResponseValue.success(1) : ResponseValue.error("淇敼澶辫触锛�");
@@ -266,7 +276,7 @@
     @ApiOperation(value = "璋冩嫧鍏ュ簱", notes = "璋冩嫧鍏ュ簱")
     @ApiImplicitParams({@ApiImplicitParam(name = "id", value = "璋冩嫧鍗昳d", required = true, dataType = "Long"),})
     @PostMapping("/income")
-    public ResponseValue income(Long id) {
+    public ResponseValue income(Long id) throws Exception {
         lWhFormTransferCoreService.doTransferInPut(id, getCurrentUser(), null);
         return ResponseValue.success();
     }
@@ -279,25 +289,24 @@
     @ApiOperation(value = "璋冩嫧鍑哄簱", notes = "璋冩嫧鍑哄簱")
     @ApiImplicitParams({@ApiImplicitParam(name = "id", value = "璋冩嫧鍗昳d", required = true, dataType = "Long"),})
     @PostMapping("/output")
-    public ResponseValue output(Long id) {
+    public ResponseValue output(Long id) throws Exception {
 
         List<LWhProcureModel> modelByForm = lWhProcureModelService.getModelByForm(WhBusinessEnum.DIAOBO, id);
         Set<Long> baseModelIds =
             modelByForm.stream().map(LWhProcureModel::getBaseGoodsModelsId).collect(Collectors.toSet());
 
-        // TODO 涓嶅悓搴撳鐞�
-        // 閫氳繃璋冩嫧鍗曚腑鐨勫瀷鍙穒d鏌ヨ鍑鸿鍨嬪彿鐗╁搧鎵�鍦ㄧ殑浠撳簱浣嶇疆
-        List<GoodsInfoDTO> goodsInfoDTOS = lWhGoodsService.queryGoodsInfo(baseModelIds);
-        GoodsInfoDTO goodsInfoDTO = goodsInfoDTOS.stream().findFirst().orElse(null);
-        // 鍑哄簱鍓� 璁剧疆鍑哄簱浠撳簱
-        // FinSysTenantUser sysTenantUser = this.getSysInfo();
-        // String agencyId = sysTenantUser.getTenantId();
-        // List<BaseWarehouse> baseWarehouseList =
-        // baseWarehouseService.getBaseWareHouseList(Long.valueOf(agencyId), (short)1, (short)1);
-        // if (CollectionUtils.isEmpty(baseWarehouseList)) {
-        // return ResponseValue.error("鏈烘瀯鏃犻粯璁や粨搴擄紒");
-        // }
+        FinSysTenantUser sysTenantUser = this.getSysInfo();
+        String agencyId = sysTenantUser.getTenantId();
+        List<BaseWarehouse> baseWarehouseList =
+            baseWarehouseService.getBaseWareHouseList(Long.valueOf(agencyId), StatesType.NORMAL.getValue());
+        Set<Long> wareHouseIds = baseWarehouseList.stream().map(BaseWarehouse::getId).collect(Collectors.toSet());
 
+        // 閫氳繃璋冩嫧鍗曚腑鐨勫瀷鍙穒d鏌ヨ鍑鸿鍨嬪彿鐗╁搧鎵�鍦ㄧ殑浠撳簱浣嶇疆
+        List<GoodsInfoDTO> goodsInfoDTOS = lWhGoodsService.queryGoodsInfo(baseModelIds, wareHouseIds,1);
+        GoodsInfoDTO goodsInfoDTO = goodsInfoDTOS.stream().findFirst().orElse(null);
+        if (ObjectUtils.isEmpty(goodsInfoDTO)) {
+            return ResponseValue.error("璇ュ瀷鍙锋病鏈夊簱瀛樺彲浣跨敤");
+        }
         Long wareHouseId = goodsInfoDTO.getWarehouseId();
         String warehouseName = goodsInfoDTO.getWarehouseName();
         LWhFormTransfer lWhFormTransfer = new LWhFormTransfer(id);
@@ -305,7 +314,7 @@
         lWhFormTransfer.setOutWarehouseName(warehouseName);
         lWhFormTransferService.update(lWhFormTransfer);
         // lWhFormTransferCoreService.doTransferOutPut(id, getCurrentUser());
-        lWhFormTransferCoreService.doTransferOutPutNew(id, getCurrentUser());
+        lWhFormTransferCoreService.doTransferOutPutNew(id, getCurrentUser(), WhBusinessEnum.DIAOBO);
         return ResponseValue.success();
     }
 
@@ -347,7 +356,7 @@
         map.put("totalAmount", totalAmount);
 
         Workbook workbook = ExcelExportUtil.exportExcel(params, TransferExcelTemplate.class, export, map);
-        String filePath = downLoadExcel(fileName, response, workbook);
+        String filePath = downLoadExcel(fileName, workbook);
         return ResponseValue.success("瀵煎嚭鎴愬姛", filePath);
 
     }
@@ -502,6 +511,7 @@
 
                 goodsUseRecordVO.setRecordSkuDtoList(useRecordSkuList);
                 return goodsUseRecordVO;
+
             }).sorted(Comparator.comparing(GoodsUseRecordVO::getUpdateTime, Comparator.nullsFirst(Long::compareTo))
                 .reversed())
             .collect(Collectors.toList());

--
Gitblit v1.9.1