review: 代码重构

lizhixian
1 parent c9690e14
Showing 1 changed file with 4 additions and 9 deletions
yudao-module-bpm/yudao-module-bpm-biz/src/main/java/cn/iocoder/yudao/module/bpm/service/definition/BpmModelServiceImpl.java
@@ -248,16 +248,11 @@ public class BpmModelServiceImpl implements BpmModelService {
                 throw exception(MODEL_DEPLOY_FAIL_BPMN_USER_TASK_NAME_NOT_EXISTS, userTask.getId());
             }
         });
-        // TODO @小北：是不是可以 UserTask firUserTask = CollUtil.get(userTasks, BpmModelTypeEnum.BPMN.getType().equals(type) ? 0 : 1)；然后，最好判空。。。极端情况下，没 usertask ，哈哈哈哈。
-        // 3. 校验第一个用户任务节点的规则类型是否为“审批人自选”
-        Map<Integer, UserTask> userTaskMap = new HashMap<>();
-        // BPMN 设计器，校验第一个用户任务节点
-        userTaskMap.put(BpmModelTypeEnum.BPMN.getType(), userTasks.get(0));
-        // SIMPLE 设计器，第一个节点固定为发起人所以校验第二个用户任务节点
-        userTaskMap.put(BpmModelTypeEnum.SIMPLE.getType(), userTasks.get(1));
-        Integer candidateStrategy = parseCandidateStrategy(userTaskMap.get(type));
+        // 3. 校验第一个用户任务节点的规则类型是否为“审批人自选”，BPMN 设计器，校验第一个用户任务节点，SIMPLE 设计器，第一个节点固定为发起人所以校验第二个用户任务节点
+        UserTask firUserTask = CollUtil.get(userTasks, BpmModelTypeEnum.BPMN.getType().equals(type) ? 0 : 1);
+        Integer candidateStrategy = parseCandidateStrategy(firUserTask);
         if (Objects.equals(candidateStrategy, BpmTaskCandidateStrategyEnum.APPROVE_USER_SELECT.getStrategy())) {
-            throw exception(MODEL_DEPLOY_FAIL_FIRST_USER_TASK_CANDIDATE_STRATEGY_ERROR, userTaskMap.get(type).getName());
+            throw exception(MODEL_DEPLOY_FAIL_FIRST_USER_TASK_CANDIDATE_STRATEGY_ERROR, firUserTask.getName());
         }
     }