From f61043f70c126c4b464a4ebc7dd635fcf7b679df Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期五, 24 七月 2026 17:25:33 +0800
Subject: [PATCH] feat(erp): 为ERP模块添加附件功能支持

---
 yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/pro/task/MesProTaskController.java |   40 +++++++++++++++++++++++++++++++++-------
 1 files changed, 33 insertions(+), 7 deletions(-)

diff --git a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/pro/task/MesProTaskController.java b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/pro/task/MesProTaskController.java
index 6f394e9..19dd295 100644
--- a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/pro/task/MesProTaskController.java
+++ b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/pro/task/MesProTaskController.java
@@ -10,11 +10,14 @@
 import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
 import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.task.vo.GanttDataRespVO;
+import cn.iocoder.yudao.module.mes.controller.admin.pro.task.vo.MesProTaskAutoGeneratePreviewRespVO;
+import cn.iocoder.yudao.module.mes.controller.admin.pro.task.vo.MesProTaskAutoGenerateReqVO;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.task.vo.MesProTaskPageReqVO;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.task.vo.MesProTaskRespVO;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.task.vo.MesProTaskSaveReqVO;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.workorder.vo.MesProWorkOrderPageReqVO;
-import cn.iocoder.yudao.module.mes.dal.dataobject.md.client.MesMdClientDO;
+import cn.iocoder.yudao.module.crm.api.customer.CrmCustomerApi;
+import cn.iocoder.yudao.module.crm.api.customer.dto.CrmCustomerRespDTO;
 import cn.iocoder.yudao.module.mes.dal.dataobject.md.item.MesMdItemDO;
 import cn.iocoder.yudao.module.mes.dal.dataobject.md.unitmeasure.MesMdUnitMeasureDO;
 import cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkstationDO;
@@ -22,7 +25,7 @@
 import cn.iocoder.yudao.module.mes.dal.dataobject.pro.task.MesProTaskDO;
 import cn.iocoder.yudao.module.mes.dal.dataobject.pro.workorder.MesProWorkOrderDO;
 import cn.iocoder.yudao.module.mes.enums.MesBizTypeConstants;
-import cn.iocoder.yudao.module.mes.service.md.client.MesMdClientService;
+import cn.iocoder.yudao.module.crm.api.customer.CrmCustomerApi;
 import cn.iocoder.yudao.module.mes.service.md.item.MesMdItemService;
 import cn.iocoder.yudao.module.mes.service.md.unitmeasure.MesMdUnitMeasureService;
 import cn.iocoder.yudao.module.mes.service.md.workstation.MesMdWorkstationService;
@@ -72,7 +75,7 @@
     @Resource
     private MesMdItemService itemService;
     @Resource
-    private MesMdClientService clientService;
+    private CrmCustomerApi customerApi;
     @Resource
     private MesMdUnitMeasureService unitMeasureService;
     @Resource
@@ -102,6 +105,23 @@
     public CommonResult<Boolean> deleteTask(@RequestParam("id") Long id) {
         taskService.deleteTask(id);
         return success(true);
+    }
+
+    @PostMapping("/auto-generate/preview")
+    @Operation(summary = "涓�閿帓浜ч瑙�")
+    @PreAuthorize("@ss.hasPermission('mes:pro-task:create')")
+    public CommonResult<MesProTaskAutoGeneratePreviewRespVO> previewAutoGenerate(
+            @Valid @RequestBody MesProTaskAutoGenerateReqVO reqVO) {
+        return success(taskService.previewAutoGenerate(reqVO));
+    }
+
+    @PostMapping("/auto-generate")
+    @Operation(summary = "涓�閿帓浜ф墽琛�")
+    @PreAuthorize("@ss.hasPermission('mes:pro-task:create')")
+    public CommonResult<java.util.Map<String, Object>> autoGenerate(
+            @Valid @RequestBody MesProTaskAutoGenerateReqVO reqVO) {
+        List<Long> taskIds = taskService.autoGenerateTasks(reqVO);
+        return success(java.util.Map.of("taskIds", taskIds));
     }
 
     @GetMapping("/get")
@@ -232,17 +252,21 @@
                 convertSet(list, MesProTaskDO::getItemId));
         Map<Long, MesMdUnitMeasureDO> unitMeasureMap = unitMeasureService.getUnitMeasureMap(
                 convertSet(itemMap.values(), MesMdItemDO::getUnitMeasureId));
-        Map<Long, MesMdClientDO> clientMap = clientService.getClientMap(
+        Map<Long, CrmCustomerRespDTO> customerMap = customerApi.getCustomerMap(
                 convertSet(list, MesProTaskDO::getClientId));
-        // 宸ュ簭鐨� checkFlag锛氭壒閲忔煡璇㈠悗鏋勫缓 routeId -> processId -> checkFlag 鐨勫弻灞� Map
+        // 宸ュ簭鐨� checkFlag + backflushFlag锛氭壒閲忔煡璇㈠悗鏋勫缓 routeId -> processId -> flag 鐨勫弻灞� Map
         Set<Long> routeIds = convertSet(list, MesProTaskDO::getRouteId);
         Map<Long, Map<Long, Boolean>> routeProcessCheckFlagMap = new HashMap<>();
+        Map<Long, Map<Long, Boolean>> routeProcessBackflushFlagMap = new HashMap<>();
         if (CollUtil.isNotEmpty(routeIds)) {
             List<MesProRouteProcessDO> allRouteProcesses = routeProcessService.getRouteProcessListByRouteIds(routeIds);
             for (MesProRouteProcessDO rp : allRouteProcesses) {
                 routeProcessCheckFlagMap
                         .computeIfAbsent(rp.getRouteId(), k -> new HashMap<>())
                         .put(rp.getProcessId(), Boolean.TRUE.equals(rp.getCheckFlag()));
+                routeProcessBackflushFlagMap
+                        .computeIfAbsent(rp.getRouteId(), k -> new HashMap<>())
+                        .put(rp.getProcessId(), Boolean.TRUE.equals(rp.getBackflushFlag()));
             }
         }
         // 鎶ュ伐鍦ㄩ�旈噺锛氭寜浠诲姟鑱氬悎 feedbackQuantity锛堣崏绋�/瀹℃壒涓�/寰呮楠� = 鍦ㄩ�旓紱宸插畬鎴� = 宸插鎵癸級
@@ -279,10 +303,12 @@
                 findAndThen(unitMeasureMap, item.getUnitMeasureId(), unit ->
                         vo.setUnitMeasureName(unit.getName()));
             });
-            findAndThen(clientMap, task.getClientId(), c ->
-                    vo.setClientName(c.getName()));
+            findAndThen(customerMap, task.getClientId(), customer ->
+                    vo.setClientName(customer.getName()));
             findAndThen(routeProcessCheckFlagMap, task.getRouteId(), processCheckMap ->
                     findAndThen(processCheckMap, task.getProcessId(), vo::setCheckFlag));
+            findAndThen(routeProcessBackflushFlagMap, task.getRouteId(), processBackflushMap ->
+                    findAndThen(processBackflushMap, task.getProcessId(), vo::setBackflushFlag));
             return vo;
         });
     }

--
Gitblit v1.9.3