From 3be684af887afe4c0e45f281cc53adfef6f12b78 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期一, 20 七月 2026 09:48:44 +0800
Subject: [PATCH] feat(erp): 添加销售订单生产状态跟踪功能

---
 yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/wm/returnissue/MesWmReturnIssueServiceImpl.java |  110 +++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 102 insertions(+), 8 deletions(-)

diff --git a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/wm/returnissue/MesWmReturnIssueServiceImpl.java b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/wm/returnissue/MesWmReturnIssueServiceImpl.java
index 41d05d5..c84bcc0 100644
--- a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/wm/returnissue/MesWmReturnIssueServiceImpl.java
+++ b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/wm/returnissue/MesWmReturnIssueServiceImpl.java
@@ -10,9 +10,12 @@
 import cn.iocoder.yudao.module.mes.controller.admin.wm.returnissue.vo.MesWmReturnIssuePageReqVO;
 import cn.iocoder.yudao.module.mes.controller.admin.wm.returnissue.vo.MesWmReturnIssueSaveReqVO;
 import cn.iocoder.yudao.module.mes.dal.dataobject.mdm.MesMdmItemDO;
+import cn.iocoder.yudao.module.mes.dal.dataobject.pro.task.MesProTaskDO;
+import cn.iocoder.yudao.module.mes.dal.dataobject.pro.task.MesProTaskIssueDO;
 import cn.iocoder.yudao.module.mes.dal.dataobject.wm.returnissue.MesWmReturnIssueDO;
 import cn.iocoder.yudao.module.mes.dal.dataobject.wm.returnissue.MesWmReturnIssueDetailDO;
 import cn.iocoder.yudao.module.mes.dal.dataobject.wm.returnissue.MesWmReturnIssueLineDO;
+import cn.iocoder.yudao.module.mes.dal.mysql.pro.task.MesProTaskIssueMapper;
 import cn.iocoder.yudao.module.mes.dal.mysql.wm.returnissue.MesWmReturnIssueMapper;
 import cn.iocoder.yudao.module.mes.enums.MesBizTypeConstants;
 import cn.iocoder.yudao.module.mes.enums.wm.MesWmQualityStatusEnum;
@@ -20,6 +23,7 @@
 import cn.iocoder.yudao.module.mes.enums.wm.MesWmTransactionTypeEnum;
 import cn.iocoder.yudao.module.mes.service.mdm.MesMdmItemService;
 import cn.iocoder.yudao.module.mes.service.md.workstation.MesMdWorkstationService;
+import cn.iocoder.yudao.module.mes.service.pro.task.MesProTaskService;
 import cn.iocoder.yudao.module.mes.service.pro.workorder.MesProWorkOrderService;
 import cn.iocoder.yudao.module.mes.service.wm.transaction.MesWmTransactionService;
 import cn.iocoder.yudao.module.mes.service.wm.transaction.dto.MesWmTransactionSaveReqDTO;
@@ -29,6 +33,8 @@
 import cn.iocoder.yudao.module.mes.dal.dataobject.wm.warehouse.MesWmWarehouseAreaDO;
 import cn.iocoder.yudao.module.mes.dal.dataobject.wm.warehouse.MesWmWarehouseDO;
 import cn.iocoder.yudao.module.mes.dal.dataobject.wm.warehouse.MesWmWarehouseLocationDO;
+import cn.iocoder.yudao.module.mes.dal.dataobject.wm.transaction.MesWmTransactionDO;
+import cn.iocoder.yudao.module.mes.dal.mysql.wm.transaction.MesWmTransactionMapper;
 import jakarta.annotation.Resource;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
@@ -61,6 +67,10 @@
     @Resource
     private MesProWorkOrderService workOrderService;
     @Resource
+    private MesProTaskService taskService;
+    @Resource
+    private MesProTaskIssueMapper taskIssueMapper;
+    @Resource
     private MesMdmItemService mdmItemService;
     @Resource
     private MesWmTransactionService wmTransactionService;
@@ -70,6 +80,8 @@
     private MesWmWarehouseLocationService locationService;
     @Resource
     private MesWmWarehouseAreaService areaService;
+    @Resource
+    private MesWmTransactionMapper transactionMapper;
 
     @Override
     @Transactional(rollbackFor = Exception.class)
@@ -77,7 +89,18 @@
         // 1. 鏍¢獙鍏宠仈鏁版嵁
         validateReturnIssueSaveData(createReqVO);
 
-        // 2. 鎻掑叆涓昏〃
+        // 2. 浠� task 鑷姩甯﹀叆 workOrderId 鍜� workstationId锛坱askId 闈炲繀濉級
+        if (createReqVO.getTaskId() != null) {
+            MesProTaskDO task = taskService.validateTaskExists(createReqVO.getTaskId());
+            if (createReqVO.getWorkOrderId() == null) {
+                createReqVO.setWorkOrderId(task.getWorkOrderId());
+            }
+            if (createReqVO.getWorkstationId() == null) {
+                createReqVO.setWorkstationId(task.getWorkstationId());
+            }
+        }
+
+        // 3. 鎻掑叆涓昏〃
         MesWmReturnIssueDO issue = BeanUtils.toBean(createReqVO, MesWmReturnIssueDO.class);
         issue.setStatus(MesWmReturnIssueStatusEnum.PREPARE.getStatus());
         issueMapper.insert(issue);
@@ -92,11 +115,22 @@
         // 1.2 鏍¢獙鍏宠仈鏁版嵁
         validateReturnIssueSaveData(updateReqVO);
 
-        // 2. 鏇存柊涓昏〃
+        // 2. 浠� task 鑷姩甯﹀叆 workOrderId 鍜� workstationId锛坱askId 闈炲繀濉級
+        if (updateReqVO.getTaskId() != null) {
+            MesProTaskDO task = taskService.validateTaskExists(updateReqVO.getTaskId());
+            if (updateReqVO.getWorkOrderId() == null) {
+                updateReqVO.setWorkOrderId(task.getWorkOrderId());
+            }
+            if (updateReqVO.getWorkstationId() == null) {
+                updateReqVO.setWorkstationId(task.getWorkstationId());
+            }
+        }
+
+        // 3. 鏇存柊涓昏〃
         MesWmReturnIssueDO updateObj = BeanUtils.toBean(updateReqVO, MesWmReturnIssueDO.class);
         issueMapper.updateById(updateObj);
 
-        // 3. 閫�鏂欑被鍨嬪彉鏇存椂锛屽埛鏂版墍鏈夎鐨勮川閲忕姸鎬�
+        // 4. 閫�鏂欑被鍨嬪彉鏇存椂锛屽埛鏂版墍鏈夎鐨勮川閲忕姸鎬�
         if (ObjUtil.notEqual(oldIssue.getType(), updateReqVO.getType())) {
             issueLineService.updateReturnIssueQualityStatusByIssueId(updateReqVO.getId(), updateReqVO.getType());
         }
@@ -192,7 +226,10 @@
         // 2. 閬嶅巻鎵�鏈夋槑缁嗭紝鍒涘缓搴撳瓨浜嬪姟锛堝鍔犲簱瀛� + 璁板綍娴佹按锛�
         createTransactionList(issue);
 
-        // 3. 鏇存柊閫�鏂欏崟鐘舵��
+        // 3. 鎵e噺浠诲姟鎶曟枡璁板綍鐨勫彲鐢ㄩ噺
+        updateTaskIssueOnReturn(issue);
+
+        // 4. 鏇存柊閫�鏂欏崟鐘舵��
         issueMapper.updateById(new MesWmReturnIssueDO()
                 .setId(id).setStatus(MesWmReturnIssueStatusEnum.FINISHED.getStatus()));
     }
@@ -203,7 +240,7 @@
         MesWmWarehouseLocationDO virtualLocation = locationService.getWarehouseLocationByCode(MesWmWarehouseLocationDO.WIP_VIRTUAL_LOCATION);
         MesWmWarehouseAreaDO virtualArea = areaService.getWarehouseAreaByCode(MesWmWarehouseAreaDO.WIP_VIRTUAL_AREA);
 
-        // 2. 閬嶅巻鏄庣粏锛屾瘡鏉℃槑缁嗕骇鐢� OUT锛堣櫄鎷熺嚎杈瑰簱鎵e噺锛�+ IN锛堝疄闄呬粨搴撳鍔狅級
+        // 2. 閬嶅巻鏄庣粏锛屾瘡鏉℃槑缁嗕骇鐢� OUT锛堣櫄鎷熺嚎杈瑰簱鎵e噺锛�+ IN锛堜笂鏋剁洰鏍囦粨搴撳鍔狅級
         List<MesWmReturnIssueDetailDO> details = issueDetailService.getReturnIssueDetailListByIssueId(issue.getId());
         for (MesWmReturnIssueDetailDO detail : details) {
             // 2.1 鍏堜粠铏氭嫙绾胯竟搴撳嚭搴擄紙搴撳瓨鍑忓皯锛�
@@ -212,18 +249,64 @@
                     .setQuantity(detail.getQuantity().negate()) // 搴撳瓨鍑忓皯
                     .setBatchId(detail.getBatchId()).setBatchCode(detail.getBatchCode())
                     .setWarehouseId(virtualWarehouse.getId()).setLocationId(virtualLocation.getId()).setAreaId(virtualArea.getId())
+                    .setMaterialStockId(detail.getMaterialStockId()) // 鎸囧畾铏氭嫙绾胯竟搴撳簱瀛樿褰�
                     .setCheckFlag(false) // 绾胯竟搴撳厑璁歌礋搴撳瓨
                     .setBizType(MesBizTypeConstants.WM_RETURN_ISSUE).setBizId(issue.getId())
                     .setBizCode(issue.getCode()).setBizLineId(detail.getLineId()));
-            // 2.2 鍐嶅叆瀹為檯浠撳簱锛堝簱瀛樺鍔狅級
+
+            // 2.2 纭畾涓婃灦鐩爣浠撳簱锛歞etail 鐨� warehouseId/locationId/areaId 鏄笂鏋剁洰鏍囷紙瀹為檯浠撳簱锛�
+            //     濡傛灉鍓嶇浠嶄紶浜嗚櫄鎷熺嚎杈瑰簱锛堝吋瀹规棫鏁版嵁锛夛紝鍒欓�氳繃浜嬪姟娴佹按鍙嶆煡鍘熷浠撳簱
+            Long targetWarehouseId = detail.getWarehouseId();
+            Long targetLocationId = detail.getLocationId();
+            Long targetAreaId = detail.getAreaId();
+            if (virtualWarehouse.getId().equals(targetWarehouseId) && detail.getMaterialStockId() != null) {
+                MesWmTransactionDO originalOutTx = transactionMapper.selectOriginalOutTransaction(detail.getMaterialStockId());
+                if (originalOutTx != null) {
+                    targetWarehouseId = originalOutTx.getWarehouseId();
+                    targetLocationId = originalOutTx.getLocationId();
+                    targetAreaId = originalOutTx.getAreaId();
+                }
+            }
+
+            // 2.3 鍏ヤ笂鏋剁洰鏍囦粨搴擄紙搴撳瓨澧炲姞锛�
             wmTransactionService.createTransaction(new MesWmTransactionSaveReqDTO()
                     .setType(MesWmTransactionTypeEnum.IN.getType()).setItemId(detail.getItemId())
                     .setQuantity(detail.getQuantity()) // 搴撳瓨澧炲姞锛堥��鏂欏洖浠擄級
                     .setBatchId(detail.getBatchId()).setBatchCode(detail.getBatchCode())
-                    .setWarehouseId(detail.getWarehouseId()).setLocationId(detail.getLocationId()).setAreaId(detail.getAreaId())
+                    .setWarehouseId(targetWarehouseId).setLocationId(targetLocationId).setAreaId(targetAreaId)
                     .setBizType(MesBizTypeConstants.WM_RETURN_ISSUE).setBizId(issue.getId())
                     .setBizCode(issue.getCode()).setBizLineId(detail.getLineId())
                     .setRelatedTransactionId(outTransactionId)); // 鍏宠仈鍑哄簱浜嬪姟
+        }
+    }
+
+    /**
+     * 閫�鏂欏畬鎴愭椂锛屾墸鍑忎换鍔℃姇鏂欒褰曠殑鍙敤閲�
+     * 鎸夎閬嶅巻锛屾壘鍒板搴旂殑 task_issue 璁板綍锛屾墸鍑� availableQuantity銆乮ssuedQuantity
+     */
+    private void updateTaskIssueOnReturn(MesWmReturnIssueDO issue) {
+        List<MesWmReturnIssueLineDO> lines = issueLineService.getReturnIssueLineListByIssueId(issue.getId());
+        if (CollUtil.isEmpty(lines)) {
+            return;
+        }
+        for (MesWmReturnIssueLineDO line : lines) {
+            Long taskId = line.getTaskId() != null ? line.getTaskId() : issue.getTaskId();
+            if (taskId == null) {
+                continue;
+            }
+            // 鏌ユ壘璇� task + item + 鏉ユ簮涓洪鏂欏嚭搴撳崟 鐨勬姇鏂欒褰�
+            List<MesProTaskIssueDO> taskIssues = taskIssueMapper.selectListByTaskId(taskId);
+            MesProTaskIssueDO matchedIssue = CollUtil.findOne(taskIssues,
+                    ti -> ti.getItemId().equals(line.getItemId())
+                            && "PRODUCT_ISSUE".equals(ti.getSourceDocType()));
+            if (matchedIssue != null) {
+                // 鎵e噺鎶曟枡閲忓拰鍙敤閲�
+                BigDecimal newIssuedQty = matchedIssue.getIssuedQuantity().subtract(line.getQuantity());
+                BigDecimal newAvailableQty = matchedIssue.getAvailableQuantity().subtract(line.getQuantity());
+                matchedIssue.setIssuedQuantity(newIssuedQty.compareTo(BigDecimal.ZERO) < 0 ? BigDecimal.ZERO : newIssuedQty);
+                matchedIssue.setAvailableQuantity(newAvailableQty.compareTo(BigDecimal.ZERO) < 0 ? BigDecimal.ZERO : newAvailableQty);
+                taskIssueMapper.updateById(matchedIssue);
+            }
         }
     }
 
@@ -289,7 +372,18 @@
      */
     private void validateReturnIssueSaveData(MesWmReturnIssueSaveReqVO reqVO) {
         validateCodeUnique(reqVO.getId(), reqVO.getCode());
-        workOrderService.validateWorkOrderConfirmed(reqVO.getWorkOrderId());
+        // 鏍¢獙 task 瀛樺湪锛坱askId 闈炲繀濉紝浼犱簡鎵嶆牎楠岋級
+        if (reqVO.getTaskId() != null) {
+            MesProTaskDO task = taskService.validateTaskExists(reqVO.getTaskId());
+            // 鏍¢獙 task 鍏宠仈鐨勫伐鍗曞凡纭
+            Long workOrderId = reqVO.getWorkOrderId() != null ? reqVO.getWorkOrderId() : task.getWorkOrderId();
+            if (workOrderId != null) {
+                workOrderService.validateWorkOrderConfirmed(workOrderId);
+            }
+        } else if (reqVO.getWorkOrderId() != null) {
+            // 娌′紶 taskId 浣嗕紶浜� workOrderId锛屾牎楠屽伐鍗曞凡纭
+            workOrderService.validateWorkOrderConfirmed(reqVO.getWorkOrderId());
+        }
         if (reqVO.getWorkstationId() != null) {
             workstationService.validateWorkstationExistsAndEnable(reqVO.getWorkstationId());
         }

--
Gitblit v1.9.3