@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.dao.utils;
import com.fasterxml.jackson.core.JsonProcessingException ;
import com.fasterxml.jackson.core.JsonProcessingException ;
import org.apache.dolphinscheduler.common.Constants ;
import org.apache.dolphinscheduler.common.Constants ;
import org.apache.dolphinscheduler.common.enums.ExecutionStatus ;
import org.apache.dolphinscheduler.common.enums.TaskDependType ;
import org.apache.dolphinscheduler.common.enums.TaskDependType ;
import org.apache.dolphinscheduler.common.graph.DAG ;
import org.apache.dolphinscheduler.common.graph.DAG ;
import org.apache.dolphinscheduler.common.model.TaskNode ;
import org.apache.dolphinscheduler.common.model.TaskNode ;
@ -27,6 +28,7 @@ import org.apache.dolphinscheduler.common.process.ProcessDag;
import org.apache.dolphinscheduler.common.utils.JSONUtils ;
import org.apache.dolphinscheduler.common.utils.JSONUtils ;
import org.apache.dolphinscheduler.dao.entity.ProcessData ;
import org.apache.dolphinscheduler.dao.entity.ProcessData ;
import org.apache.dolphinscheduler.dao.entity.TaskInstance ;
import org.apache.dolphinscheduler.dao.entity.TaskInstance ;
import org.junit.Assert ;
import org.junit.Assert ;
import org.junit.Test ;
import org.junit.Test ;
@ -34,6 +36,8 @@ import java.util.ArrayList;
import java.util.HashMap ;
import java.util.HashMap ;
import java.util.List ;
import java.util.List ;
import java.util.Map ;
import java.util.Map ;
import java.util.Set ;
/ * *
/ * *
* dag helper test
* dag helper test
@ -41,15 +45,17 @@ import java.util.Map;
public class DagHelperTest {
public class DagHelperTest {
/ * *
/ * *
* test task node can submit
* test task node can submit
*
* @throws JsonProcessingException if error throws JsonProcessingException
* @throws JsonProcessingException if error throws JsonProcessingException
* /
* /
@Test
@Test
public void testTaskNodeCanSubmit ( ) throws JsonProcessingException {
public void testTaskNodeCanSubmit ( ) throws JsonProcessingException {
//1->2->3->5
//1->2->3->5->7
//4->3
//4->3->6
DAG < String , TaskNode , TaskNodeRelation > dag = generateDag ( ) ;
DAG < String , TaskNode , TaskNodeRelation > dag = generateDag ( ) ;
TaskNode taskNode3 = dag . getNode ( "3" ) ;
TaskNode taskNode3 = dag . getNode ( "3" ) ;
Map < String , TaskInstance > completeTaskList = new HashMap < > ( ) ;
Map < String , TaskInstance > completeTaskList = new HashMap < > ( ) ;
Map < String , TaskNode > skipNodeList = new HashMap < > ( ) ;
completeTaskList . putIfAbsent ( "1" , new TaskInstance ( ) ) ;
completeTaskList . putIfAbsent ( "1" , new TaskInstance ( ) ) ;
Boolean canSubmit = false ;
Boolean canSubmit = false ;
@ -58,27 +64,199 @@ public class DagHelperTest {
node2 . setRunFlag ( Constants . FLOWNODE_RUN_FLAG_FORBIDDEN ) ;
node2 . setRunFlag ( Constants . FLOWNODE_RUN_FLAG_FORBIDDEN ) ;
TaskNode nodex = dag . getNode ( "4" ) ;
TaskNode nodex = dag . getNode ( "4" ) ;
nodex . setRunFlag ( Constants . FLOWNODE_RUN_FLAG_FORBIDDEN ) ;
nodex . setRunFlag ( Constants . FLOWNODE_RUN_FLAG_FORBIDDEN ) ;
canSubmit = DagHelper . taskNodeCanSubmit ( taskNode3 , dag , completeTaskList ) ;
canSubmit = DagHelper . allDependsForbiddenOrEnd ( taskNode3 , dag , skipNodeList , completeTaskList ) ;
Assert . assertEquals ( canSubmit , true ) ;
Assert . assertEquals ( canSubmit , true ) ;
// 2forbidden, 3 cannot be submit
// 2forbidden, 3 cannot be submit
completeTaskList . putIfAbsent ( "2" , new TaskInstance ( ) ) ;
completeTaskList . putIfAbsent ( "2" , new TaskInstance ( ) ) ;
TaskNode nodey = dag . getNode ( "4" ) ;
TaskNode nodey = dag . getNode ( "4" ) ;
nodey . setRunFlag ( "" ) ;
nodey . setRunFlag ( "" ) ;
canSubmit = DagHelper . taskNodeCanSubmit ( taskNode3 , dag , completeTaskList ) ;
canSubmit = DagHelper . allDependsForbiddenOrEnd ( taskNode3 , dag , skipNodeList , completeTaskList ) ;
Assert . assertEquals ( canSubmit , false ) ;
Assert . assertEquals ( canSubmit , false ) ;
// 2/3 forbidden submit 5
// 2/3 forbidden submit 5
TaskNode node3 = dag . getNode ( "3" ) ;
TaskNode node3 = dag . getNode ( "3" ) ;
node3 . setRunFlag ( Constants . FLOWNODE_RUN_FLAG_FORBIDDEN ) ;
node3 . setRunFlag ( Constants . FLOWNODE_RUN_FLAG_FORBIDDEN ) ;
TaskNode node8 = dag . getNode ( "8" ) ;
node8 . setRunFlag ( Constants . FLOWNODE_RUN_FLAG_FORBIDDEN ) ;
TaskNode node5 = dag . getNode ( "5" ) ;
TaskNode node5 = dag . getNode ( "5" ) ;
canSubmit = DagHelper . taskNodeCanSubmit ( node5 , dag , completeTaskList ) ;
canSubmit = DagHelper . allDependsForbiddenOrEnd ( node5 , dag , skipNodeList , completeTaskList ) ;
Assert . assertEquals ( canSubmit , true ) ;
Assert . assertEquals ( canSubmit , true ) ;
}
}
/ * *
/ * *
* 1 - > 2 - > 3 - > 5
* test parse post node list
* 4 - > 3
* /
@Test
public void testParsePostNodeList ( ) throws JsonProcessingException {
DAG < String , TaskNode , TaskNodeRelation > dag = generateDag ( ) ;
Map < String , TaskInstance > completeTaskList = new HashMap < > ( ) ;
Map < String , TaskNode > skipNodeList = new HashMap < > ( ) ;
Set < String > postNodes = null ;
//complete : null
// expect post: 1/4
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 2 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "1" ) ) ;
Assert . assertTrue ( postNodes . contains ( "4" ) ) ;
//complete : 1
// expect post: 2/4
completeTaskList . put ( "1" , new TaskInstance ( ) ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 2 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "2" ) ) ;
Assert . assertTrue ( postNodes . contains ( "4" ) ) ;
// complete : 1/2
// expect post: 4
completeTaskList . put ( "2" , new TaskInstance ( ) ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 2 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "4" ) ) ;
Assert . assertTrue ( postNodes . contains ( "8" ) ) ;
// complete : 1/2/4
// expect post: 3
completeTaskList . put ( "4" , new TaskInstance ( ) ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 2 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "3" ) ) ;
Assert . assertTrue ( postNodes . contains ( "8" ) ) ;
// complete : 1/2/4/3
// expect post: 8/6
completeTaskList . put ( "3" , new TaskInstance ( ) ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 2 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "8" ) ) ;
Assert . assertTrue ( postNodes . contains ( "6" ) ) ;
// complete : 1/2/4/3/8
// expect post: 6/5
completeTaskList . put ( "8" , new TaskInstance ( ) ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 2 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "5" ) ) ;
Assert . assertTrue ( postNodes . contains ( "6" ) ) ;
// complete : 1/2/4/3/5/6/8
// expect post: 7
completeTaskList . put ( "6" , new TaskInstance ( ) ) ;
completeTaskList . put ( "5" , new TaskInstance ( ) ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 1 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "7" ) ) ;
}
/ * *
* test forbidden post node
* @throws JsonProcessingException
* /
@Test
public void testForbiddenPostNode ( ) throws JsonProcessingException {
DAG < String , TaskNode , TaskNodeRelation > dag = generateDag ( ) ;
Map < String , TaskInstance > completeTaskList = new HashMap < > ( ) ;
Map < String , TaskNode > skipNodeList = new HashMap < > ( ) ;
Set < String > postNodes = null ;
// dag: 1-2-3-5-7 4-3-6 2-8-5-7
// forbid:2 complete:1 post:4/8
completeTaskList . put ( "1" , new TaskInstance ( ) ) ;
TaskNode node2 = dag . getNode ( "2" ) ;
node2 . setRunFlag ( Constants . FLOWNODE_RUN_FLAG_FORBIDDEN ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 2 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "4" ) ) ;
Assert . assertTrue ( postNodes . contains ( "8" ) ) ;
//forbid:2/4 complete:1 post:3/8
TaskNode node4 = dag . getNode ( "4" ) ;
node4 . setRunFlag ( Constants . FLOWNODE_RUN_FLAG_FORBIDDEN ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 2 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "3" ) ) ;
Assert . assertTrue ( postNodes . contains ( "8" ) ) ;
//forbid:2/4/5 complete:1/8 post:3
completeTaskList . put ( "8" , new TaskInstance ( ) ) ;
TaskNode node5 = dag . getNode ( "5" ) ;
node5 . setRunFlag ( Constants . FLOWNODE_RUN_FLAG_FORBIDDEN ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 1 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "3" ) ) ;
}
/ * *
* test condition post node
* @throws JsonProcessingException
* /
@Test
public void testConditionPostNode ( ) throws JsonProcessingException {
DAG < String , TaskNode , TaskNodeRelation > dag = generateDag ( ) ;
Map < String , TaskInstance > completeTaskList = new HashMap < > ( ) ;
Map < String , TaskNode > skipNodeList = new HashMap < > ( ) ;
Set < String > postNodes = null ;
// dag: 1-2-3-5-7 4-3-6 2-8-5-7
// 3-if
completeTaskList . put ( "1" , new TaskInstance ( ) ) ;
completeTaskList . put ( "2" , new TaskInstance ( ) ) ;
completeTaskList . put ( "4" , new TaskInstance ( ) ) ;
TaskNode node3 = dag . getNode ( "3" ) ;
node3 . setType ( "CONDITIONS" ) ;
node3 . setConditionResult ( "{\n" +
" \"successNode\": [5\n" +
" ],\n" +
" \"failedNode\": [6\n" +
" ]\n" +
" }" ) ;
completeTaskList . remove ( "3" ) ;
TaskInstance taskInstance = new TaskInstance ( ) ;
taskInstance . setState ( ExecutionStatus . SUCCESS ) ;
//complete 1/2/3/4 expect:8
completeTaskList . put ( "3" , taskInstance ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 1 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "8" ) ) ;
//2.complete 1/2/3/4/8 expect:5 skip:6
completeTaskList . put ( "8" , new TaskInstance ( ) ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertTrue ( postNodes . contains ( "5" ) ) ;
Assert . assertEquals ( 1 , skipNodeList . size ( ) ) ;
Assert . assertTrue ( skipNodeList . containsKey ( "6" ) ) ;
// 3.complete 1/2/3/4/5/8 expect post:7 skip:6
skipNodeList . clear ( ) ;
TaskInstance taskInstance1 = new TaskInstance ( ) ;
taskInstance . setState ( ExecutionStatus . SUCCESS ) ;
completeTaskList . put ( "5" , taskInstance1 ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 1 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "7" ) ) ;
Assert . assertEquals ( 1 , skipNodeList . size ( ) ) ;
Assert . assertTrue ( skipNodeList . containsKey ( "6" ) ) ;
// dag: 1-2-3-5-7 4-3-6
// 3-if , complete:1/2/3/4
// 1.failure:3 expect post:6 skip:5/7
skipNodeList . clear ( ) ;
completeTaskList . remove ( "3" ) ;
taskInstance = new TaskInstance ( ) ;
taskInstance . setState ( ExecutionStatus . FAILURE ) ;
completeTaskList . put ( "3" , taskInstance ) ;
postNodes = DagHelper . parsePostNodes ( null , skipNodeList , dag , completeTaskList ) ;
Assert . assertEquals ( 1 , postNodes . size ( ) ) ;
Assert . assertTrue ( postNodes . contains ( "6" ) ) ;
Assert . assertEquals ( 2 , skipNodeList . size ( ) ) ;
Assert . assertTrue ( skipNodeList . containsKey ( "5" ) ) ;
Assert . assertTrue ( skipNodeList . containsKey ( "7" ) ) ;
}
/ * *
* 1 - > 2 - > 3 - > 5 - > 7
* 4 - > 3 - > 6
* 2 - > 8 - > 5 - > 7
*
* @return dag
* @return dag
* @throws JsonProcessingException if error throws JsonProcessingException
* @throws JsonProcessingException if error throws JsonProcessingException
* /
* /
@ -87,11 +265,13 @@ public class DagHelperTest {
TaskNode node1 = new TaskNode ( ) ;
TaskNode node1 = new TaskNode ( ) ;
node1 . setId ( "1" ) ;
node1 . setId ( "1" ) ;
node1 . setName ( "1" ) ;
node1 . setName ( "1" ) ;
node1 . setType ( "SHELL" ) ;
taskNodeList . add ( node1 ) ;
taskNodeList . add ( node1 ) ;
TaskNode node2 = new TaskNode ( ) ;
TaskNode node2 = new TaskNode ( ) ;
node2 . setId ( "2" ) ;
node2 . setId ( "2" ) ;
node2 . setName ( "2" ) ;
node2 . setName ( "2" ) ;
node2 . setType ( "SHELL" ) ;
List < String > dep2 = new ArrayList < > ( ) ;
List < String > dep2 = new ArrayList < > ( ) ;
dep2 . add ( "1" ) ;
dep2 . add ( "1" ) ;
node2 . setDepList ( dep2 ) ;
node2 . setDepList ( dep2 ) ;
@ -101,11 +281,13 @@ public class DagHelperTest {
TaskNode node4 = new TaskNode ( ) ;
TaskNode node4 = new TaskNode ( ) ;
node4 . setId ( "4" ) ;
node4 . setId ( "4" ) ;
node4 . setName ( "4" ) ;
node4 . setName ( "4" ) ;
node4 . setType ( "SHELL" ) ;
taskNodeList . add ( node4 ) ;
taskNodeList . add ( node4 ) ;
TaskNode node3 = new TaskNode ( ) ;
TaskNode node3 = new TaskNode ( ) ;
node3 . setId ( "3" ) ;
node3 . setId ( "3" ) ;
node3 . setName ( "3" ) ;
node3 . setName ( "3" ) ;
node3 . setType ( "SHELL" ) ;
List < String > dep3 = new ArrayList < > ( ) ;
List < String > dep3 = new ArrayList < > ( ) ;
dep3 . add ( "2" ) ;
dep3 . add ( "2" ) ;
dep3 . add ( "4" ) ;
dep3 . add ( "4" ) ;
@ -115,20 +297,48 @@ public class DagHelperTest {
TaskNode node5 = new TaskNode ( ) ;
TaskNode node5 = new TaskNode ( ) ;
node5 . setId ( "5" ) ;
node5 . setId ( "5" ) ;
node5 . setName ( "5" ) ;
node5 . setName ( "5" ) ;
node5 . setType ( "SHELL" ) ;
List < String > dep5 = new ArrayList < > ( ) ;
List < String > dep5 = new ArrayList < > ( ) ;
dep5 . add ( "3" ) ;
dep5 . add ( "3" ) ;
dep5 . add ( "8" ) ;
node5 . setDepList ( dep5 ) ;
node5 . setDepList ( dep5 ) ;
taskNodeList . add ( node5 ) ;
taskNodeList . add ( node5 ) ;
TaskNode node6 = new TaskNode ( ) ;
node6 . setId ( "6" ) ;
node6 . setName ( "6" ) ;
node6 . setType ( "SHELL" ) ;
List < String > dep6 = new ArrayList < > ( ) ;
dep6 . add ( "3" ) ;
node6 . setDepList ( dep6 ) ;
taskNodeList . add ( node6 ) ;
TaskNode node7 = new TaskNode ( ) ;
node7 . setId ( "7" ) ;
node7 . setName ( "7" ) ;
node7 . setType ( "SHELL" ) ;
List < String > dep7 = new ArrayList < > ( ) ;
dep7 . add ( "5" ) ;
node7 . setDepList ( dep7 ) ;
taskNodeList . add ( node7 ) ;
TaskNode node8 = new TaskNode ( ) ;
node8 . setId ( "8" ) ;
node8 . setName ( "8" ) ;
node8 . setType ( "SHELL" ) ;
List < String > dep8 = new ArrayList < > ( ) ;
dep8 . add ( "2" ) ;
node8 . setDepList ( dep8 ) ;
taskNodeList . add ( node8 ) ;
List < String > startNodes = new ArrayList < > ( ) ;
List < String > startNodes = new ArrayList < > ( ) ;
List < String > recoveryNodes = new ArrayList < > ( ) ;
List < String > recoveryNodes = new ArrayList < > ( ) ;
List < TaskNode > destTaskNodeList = DagHelper . generateFlowNodeListByStartNode ( taskNodeList ,
List < TaskNode > destTaskNodeList = DagHelper . generateFlowNodeListByStartNode ( taskNodeList ,
startNodes , recoveryNodes , TaskDependType . TASK_POST ) ;
startNodes , recoveryNodes , TaskDependType . TASK_POST ) ;
List < TaskNodeRelation > taskNodeRelations = DagHelper . generateRelationListByFlowNodes ( destTaskNodeList ) ;
List < TaskNodeRelation > taskNodeRelations = DagHelper . generateRelationListByFlowNodes ( destTaskNodeList ) ;
ProcessDag processDag = new ProcessDag ( ) ;
ProcessDag processDag = new ProcessDag ( ) ;
processDag . setEdges ( taskNodeRelations ) ;
processDag . setEdges ( taskNodeRelations ) ;
processDag . setNodes ( destTaskNodeList ) ;
processDag . setNodes ( destTaskNodeList ) ;
return DagHelper . buildDagGraph ( processDag ) ;
return DagHelper . buildDagGraph ( processDag ) ;
}
}