How to use

setStateTransitionComparator

method

in

org.springframework.batch.core.job.flow.support.SimpleFlow

@Override
public SimpleFlow getObject() throws Exception {
  SimpleFlow flow = flowType.getConstructor(String.class).newInstance(name);
  flow.setStateTransitionComparator(stateTransitionComparator);
  List<StateTransition> updatedTransitions = new ArrayList<>();
  for (StateTransition stateTransition : stateTransitions) {
    State state = getProxyState(stateTransition.getState());
    updatedTransitions.add(StateTransition.switchOriginAndDestination(stateTransition, state,
        getNext(stateTransition.getNext())));
  }
  flow.setStateTransitions(updatedTransitions);
  flow.afterPropertiesSet();
  return flow;
}

@Test
public void testBranching() throws Exception {
  SimpleFlow flow = new SimpleFlow("job");
  List<StateTransition> transitions = new ArrayList<>();
  StepState step1 = new StepState(new StubStep("step1"));
  transitions.add(StateTransition.createStateTransition(step1, "step2"));
  transitions.add(StateTransition.createStateTransition(step1, "COMPLETED", "step3"));
  StepState step2 = new StepState(new StubStep("step2"));
  transitions.add(StateTransition.createStateTransition(step2, ExitStatus.COMPLETED.getExitCode(), "end0"));
  transitions.add(StateTransition.createStateTransition(step2, ExitStatus.FAILED.getExitCode(), "end1"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end0")));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.FAILED, "end1")));
  StepState step3 = new StepState(new StubStep("step3"));
  transitions.add(StateTransition.createStateTransition(step3, ExitStatus.FAILED.getExitCode(), "end2"));
  transitions.add(StateTransition.createStateTransition(step3, ExitStatus.COMPLETED.getExitCode(), "end3"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.FAILED, "end2")));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end3")));
  flow.setStateTransitions(transitions);
  flow.setStateTransitionComparator(new DefaultStateTransitionComparator());
  job.setFlow(flow);
  job.afterPropertiesSet();
  job.doExecute(jobExecution);
  StepExecution stepExecution = getStepExecution(jobExecution, "step3");
  assertEquals(ExitStatus.COMPLETED, stepExecution.getExitStatus());
  assertEquals(2, jobExecution.getStepExecutions().size());
}

transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end3")));
flow.setStateTransitions(transitions);
flow.setStateTransitionComparator(new DefaultStateTransitionComparator());

@Test
public void testBranching() throws Exception {
  flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1"), "step2"),
      StateTransition.createStateTransition(new StubState("step1"), ExitStatus.COMPLETED.getExitCode(), "step3"),
      StateTransition.createEndStateTransition(new StubState("step2")), StateTransition
      .createEndStateTransition(new StubState("step3"))));
  flow.setStateTransitionComparator(new DefaultStateTransitionComparator());
  flow.afterPropertiesSet();
  FlowExecution execution = flow.start(executor);
  assertEquals(FlowExecutionStatus.COMPLETED, execution.getStatus());
  assertEquals("step3", execution.getName());
}

@Override
public SimpleFlow getObject() throws Exception {
  SimpleFlow flow = flowType.getConstructor(String.class).newInstance(name);
  flow.setStateTransitionComparator(stateTransitionComparator);
  List<StateTransition> updatedTransitions = new ArrayList<StateTransition>();
  for (StateTransition stateTransition : stateTransitions) {
    State state = getProxyState(stateTransition.getState());
    updatedTransitions.add(StateTransition.switchOriginAndDestination(stateTransition, state,
        getNext(stateTransition.getNext())));
  }
  flow.setStateTransitions(updatedTransitions);
  flow.afterPropertiesSet();
  return flow;
}

@Override
public SimpleFlow getObject() throws Exception {
  SimpleFlow flow = flowType.getConstructor(String.class).newInstance(name);
  flow.setStateTransitionComparator(stateTransitionComparator);
  List<StateTransition> updatedTransitions = new ArrayList<StateTransition>();
  for (StateTransition stateTransition : stateTransitions) {
    State state = getProxyState(stateTransition.getState());
    updatedTransitions.add(StateTransition.switchOriginAndDestination(stateTransition, state,
        getNext(stateTransition.getNext())));
  }
  flow.setStateTransitions(updatedTransitions);
  flow.afterPropertiesSet();
  return flow;
}

@Override
public SimpleFlow getObject() throws Exception {
  SimpleFlow flow = flowType.getConstructor(String.class).newInstance(name);
  flow.setStateTransitionComparator(stateTransitionComparator);
  List<StateTransition> updatedTransitions = new ArrayList<StateTransition>();
  for (StateTransition stateTransition : stateTransitions) {
    State state = getProxyState(stateTransition.getState());
    updatedTransitions.add(StateTransition.switchOriginAndDestination(stateTransition, state,
        getNext(stateTransition.getNext())));
  }
  flow.setStateTransitions(updatedTransitions);
  flow.afterPropertiesSet();
  return flow;
}