From 83d6b764019d3eef0f9c98cf11b98f98e1ef9274 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期三, 26 八月 2026 15:07:53 +0800
Subject: [PATCH] feat(mes): 增加工艺参数模板功能支持

---
 yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/task/MesProTaskServiceImpl.java |   70 ++++++++++++++++++++++++++++++++++
 1 files changed, 69 insertions(+), 1 deletions(-)

diff --git a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/task/MesProTaskServiceImpl.java b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/task/MesProTaskServiceImpl.java
index d6ebc52..5f82736 100644
--- a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/task/MesProTaskServiceImpl.java
+++ b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/task/MesProTaskServiceImpl.java
@@ -33,6 +33,7 @@
 import cn.iocoder.yudao.module.mes.service.pro.workorder.MesProWorkOrderService;
 import cn.iocoder.yudao.module.mes.dal.dataobject.pro.workorder.MesProWorkOrderProcessDO;
 import cn.iocoder.yudao.module.mes.service.pro.workorder.MesProWorkOrderProcessService;
+import cn.iocoder.yudao.module.mes.service.pro.workorder.MesProWorkOrderParamRecordService;
 import jakarta.annotation.Resource;
 import org.springframework.context.annotation.Lazy;
 import org.springframework.stereotype.Service;
@@ -58,6 +59,8 @@
 import cn.iocoder.yudao.framework.common.util.object.ObjectUtils;
 import cn.iocoder.yudao.module.mes.dal.dataobject.pro.task.MesProTaskPauseRecordDO;
 import cn.iocoder.yudao.module.mes.dal.mysql.pro.task.MesProTaskPauseRecordMapper;
+import cn.iocoder.yudao.module.mes.dal.dataobject.pro.task.MesProTaskAssigneeDO;
+import cn.iocoder.yudao.module.mes.dal.mysql.pro.task.MesProTaskAssigneeMapper;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.task.vo.MesProTaskPauseReqVO;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.task.vo.MesProTaskResumeReqVO;
 
@@ -78,6 +81,9 @@
 
     @Resource
     private MesProTaskPauseRecordMapper taskPauseRecordMapper;
+
+    @Resource
+    private MesProTaskAssigneeMapper taskAssigneeMapper;
 
     @Resource
     @Lazy
@@ -101,6 +107,9 @@
     @Resource
     @Lazy
     private MesProWorkOrderProcessService workOrderProcessService;
+    @Resource
+    @Lazy
+    private MesProWorkOrderParamRecordService workOrderParamRecordService;
     @Resource
     private MesMdAutoCodeRecordService autoCodeRecordService;
     @Resource
@@ -141,6 +150,13 @@
         }
         // 2.2 鎻掑叆
         taskMapper.insert(task);
+        if (task.getParamTemplateId() != null) {
+            workOrderParamRecordService.createTaskParamRecord(task.getId(), task.getParamTemplateId());
+        }
+
+        // 2.3 淇濆瓨鎸囧畾鎶ュ伐浜哄憳
+        updateTaskAssignees(task.getId(), createReqVO.getAssigneeUserIds());
+        
         return task.getId();
     }
 
@@ -173,6 +189,34 @@
         MesProTaskDO updateObj = BeanUtils.toBean(updateReqVO, MesProTaskDO.class)
                 .setName(buildTaskName(item, quantity));
         taskMapper.updateById(updateObj);
+        
+        // 3. 鏇存柊鎸囧畾鎶ュ伐浜哄憳锛堟瘮瀵规洿鏂帮級
+        updateTaskAssignees(task.getId(), updateReqVO.getAssigneeUserIds());
+    }
+
+    private void updateTaskAssignees(Long taskId, List<Long> assigneeUserIds) {
+        if (assigneeUserIds == null) {
+            assigneeUserIds = Collections.emptyList();
+        }
+        // 1. 鏌ヨ鏁版嵁搴撲腑鍘熸湁鐨�
+        List<MesProTaskAssigneeDO> oldAssignees = taskAssigneeMapper.selectListByTaskId(taskId);
+        Set<Long> oldUserIds = convertSet(oldAssignees, MesProTaskAssigneeDO::getUserId);
+        Set<Long> newUserIds = new java.util.HashSet<>(assigneeUserIds);
+
+        // 2. 璁$畻闇�瑕佹柊澧炵殑鍜岄渶瑕佸垹闄ょ殑
+        Set<Long> toAddUserIds = newUserIds.stream().filter(id -> !oldUserIds.contains(id)).collect(java.util.stream.Collectors.toSet());
+        Set<Long> toRemoveUserIds = oldUserIds.stream().filter(id -> !newUserIds.contains(id)).collect(java.util.stream.Collectors.toSet());
+
+        // 3. 鏂板
+        if (CollUtil.isNotEmpty(toAddUserIds)) {
+            List<MesProTaskAssigneeDO> addList = convertList(toAddUserIds, userId -> new MesProTaskAssigneeDO().setTaskId(taskId).setUserId(userId));
+            taskAssigneeMapper.insertBatch(addList);
+        }
+
+        // 4. 鍒犻櫎
+        if (CollUtil.isNotEmpty(toRemoveUserIds)) {
+            taskAssigneeMapper.deleteByTaskIdAndUserIds(taskId, toRemoveUserIds);
+        }
     }
 
     /**
@@ -220,6 +264,7 @@
                 result.add(new MesProRouteProcessDO()
                         .setRouteId(routeId)
                         .setProcessId(wp.getProcessId())
+                        .setParamTemplateId(wp.getParamTemplateId())
                         .setSort(wp.getSort())
                         .setKeyFlag(wp.getKeyFlag())
                         .setCheckFlag(wp.getCheckFlag())
@@ -245,8 +290,10 @@
         if (feedbackMapper.selectCountByTaskId(id) > 0) {
             throw exception(PRO_TASK_HAS_FEEDBACK);
         }
-        // 3. 鍒犻櫎
+        // 3. 鍒犻櫎浠诲姟涓昏〃
         taskMapper.deleteById(id);
+        // 4. 鍒犻櫎鍏宠仈鐨勪汉鍛�
+        taskAssigneeMapper.deleteByTaskId(id);
     }
 
     @Override
@@ -270,6 +317,14 @@
             return Collections.emptyList();
         }
         return taskMapper.selectListByWorkOrderIds(workOrderIds);
+    }
+
+    @Override
+    public List<MesProTaskAssigneeDO> getTaskAssigneeListByTaskIds(Collection<Long> taskIds) {
+        if (CollUtil.isEmpty(taskIds)) {
+            return Collections.emptyList();
+        }
+        return taskAssigneeMapper.selectListByTaskIds(taskIds);
     }
 
     @Override
@@ -337,6 +392,15 @@
         validateTaskExists(id);
         // 鏇存柊鏁伴噺
         taskMapper.updateProducedQuantity(id, incrProducedQuantity, incrQualifyQuantity, incrUnqualifyQuantity);
+        MesProTaskDO task = taskMapper.selectById(id);
+        if (task != null) {
+            LocalDateTime now = LocalDateTime.now();
+            MesProTaskDO timeUpdate = new MesProTaskDO().setId(id).setLastFeedbackTime(now);
+            if (task.getFirstFeedbackTime() == null) {
+                timeUpdate.setFirstFeedbackTime(now);
+            }
+            taskMapper.updateById(timeUpdate);
+        }
     }
 
     @Override
@@ -430,9 +494,13 @@
                     .setName(buildTaskName(item, workOrder.getQuantity()))
                     .setCode(autoCodeRecordService.generateAutoCode(MesMdAutoCodeRuleCodeEnum.PRO_TASK_CODE.getCode()))
                     .setStatus(MesProTaskStatusEnum.PREPARE.getStatus())
+                    .setParamTemplateId(rp.getParamTemplateId())
                     .setClientId(workOrder.getClientId())
                     .setColorCode(rp.getColorCode());
             taskMapper.insert(task);
+            if (task.getParamTemplateId() != null) {
+                workOrderParamRecordService.createTaskParamRecord(task.getId(), task.getParamTemplateId());
+            }
             taskIds.add(task.getId());
         }
         // 7. 鏇存柊宸ュ崟宸叉帓浜ф暟閲�

--
Gitblit v1.9.3