natas_hw
/
dolphinscheduler
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge 3d9049bc24 into d6714bb77a

This commit is contained in:
gitjxm 2024-06-07 10:11:05 +08:00 committed by GitHub
parent d6714bb77a 3d9049bc24
commit db17483f71
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/process/ProcessServiceImpl.java
View File

@ -492,10 +492,22 @@ public class ProcessServiceImpl implements ProcessService {
*/
@Override
public List<Long> findAllSubWorkflowDefinitionCode(long parentCode) {
return findAllSubWorkflowDefinitionCode(parentCode, new HashSet<>());
}
/**
* recursive query sub process definition id by parent id.
*
* @param parentCode parentCode
* @param visited visited
*/
private List<Long> findAllSubWorkflowDefinitionCode(long parentCode, Set<Long> visited) {
List<TaskDefinition> taskNodeList = taskDefinitionDao.getTaskDefinitionListByDefinition(parentCode);
if (CollectionUtils.isEmpty(taskNodeList)) {
return Collections.emptyList();
}
visited.add(parentCode);
List<Long> subWorkflowDefinitionCodes = new ArrayList<>();
for (TaskDefinition taskNode : taskNodeList) {
@ -504,10 +516,15 @@ public class ProcessServiceImpl implements ProcessService {
if (parameterJson.get(CMD_PARAM_SUB_PROCESS_DEFINE_CODE) != null) {
SubProcessParameters subProcessParam = JSONUtils.parseObject(parameter, SubProcessParameters.class);
long subWorkflowDefinitionCode = subProcessParam.getProcessDefinitionCode();
if (visited.contains(subWorkflowDefinitionCode)) {
throw new ServiceException("Detected a cycle in the workflow definitions with code: " + subWorkflowDefinitionCode);
}
subWorkflowDefinitionCodes.add(subWorkflowDefinitionCode);
subWorkflowDefinitionCodes.addAll(findAllSubWorkflowDefinitionCode(subWorkflowDefinitionCode));
subWorkflowDefinitionCodes.addAll(findAllSubWorkflowDefinitionCode(subWorkflowDefinitionCode, visited));
}
}
visited.remove(parentCode);
return subWorkflowDefinitionCodes;
}