org.springframework.batch.core.job.flow.support.state.StepState java code examples

Add the Codota plugin to your IDE and get smart completions

private void myMethod () {
 LocalDateTime l =new LocalDateTime()
LocalDateTime.now()
DateTimeFormatter formatter;String text;formatter.parseLocalDateTime(text)
Smart code suggestions by Tabnine
}

private State createState(Object input) {
  State result;
  if (input instanceof Step) {
    if (!states.containsKey(input)) {
      Step step = (Step) input;
      states.put(input, new StepState(prefix + step.getName(), step));
    }
    result = states.get(input);
  }
  else if (input instanceof JobExecutionDecider) {
    if (!states.containsKey(input)) {
      states.put(input, new DecisionState((JobExecutionDecider) input, prefix + "decision"
          + (decisionCounter++)));
    }
    result = states.get(input);
  }
  else if (input instanceof Flow) {
    if (!states.containsKey(input)) {
      states.put(input, new FlowState((Flow) input, prefix + ((Flow) input).getName()));
    }
    result = states.get(input);
  }
  else {
    throw new FlowBuilderException("No state can be created for: " + input);
  }
  dirty = true;
  return result;
}

/**
 * Provides an extension point to provide alternative {@link StepState}
 * implementations within a {@link SimpleFlow}
 *
 * @param state The state that will be used to create the StepState
 * @param oldName The name to be replaced
 * @param stateName The name for the new State
 * @return a state for the requested data
 */
protected State createNewStepState(State state, String oldName,
    String stateName) {
  return new StepState(stateName, ((StepState) state).getStep(oldName));
}

  @Override
  public FlowExecutionStatus handle(FlowExecutor executor) throws Exception {
    FlowExecutionStatus result = super.handle(executor);

    executor.getJobExecution().getExecutionContext().put("batch.lastSteps", Collections.singletonList(getStep().getName()));

    return result;
  }
}

@Bean
public SimpleFlow simpleFlow() {
  SimpleFlow simpleFlow = new SimpleFlow("simpleFlow");
  List<StateTransition> transitions = new ArrayList<>();
  transitions.add(StateTransition.createStateTransition(new StepState(dummyStep()), "end0"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end0")));
  simpleFlow.setStateTransitions(transitions);
  return simpleFlow;
}

/**
 * Provides an extension point to provide alternative {@link StepState}
 * implementations within a {@link SimpleFlow}
 *
 * @param state The state that will be used to create the StepState
 * @param oldName The name to be replaced
 * @param stateName The name for the new State
 * @return a state for the requested data
 */
protected State createNewStepState(State state, String oldName,
    String stateName) {
  return new StepState(stateName, ((StepState) state).getStep(oldName));
}

  @Override
  public FlowExecutionStatus handle(FlowExecutor executor) throws Exception {
    FlowExecutionStatus result = super.handle(executor);

    executor.getJobExecution().getExecutionContext().put("batch.lastSteps", Collections.singletonList(getStep().getName()));

    return result;
  }
}

@Test
public void testGetSteps() throws Exception {
  SimpleFlow flow = new JsrFlow("job");
  List<StateTransition> transitions = new ArrayList<>();
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "step2"));
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step2")), "end0"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end0")));
  flow.setStateTransitions(transitions);
  flow.afterPropertiesSet();
  job.setFlow(flow);
  job.afterPropertiesSet();
  assertEquals(2, job.getStepNames().size());
}

/**
 * Provides an extension point to provide alternative {@link StepState}
 * implementations within a {@link SimpleFlow}
 *
 * @param state The state that will be used to create the StepState
 * @param oldName The name to be replaced
 * @param stateName The name for the new State
 * @return
 */
protected State createNewStepState(State state, String oldName,
    String stateName) {
  return new StepState(stateName, ((StepState) state).getStep(oldName));
}

  @Override
  public FlowExecutionStatus handle(FlowExecutor executor) throws Exception {
    FlowExecutionStatus result = super.handle(executor);

    executor.getJobExecution().getExecutionContext().put("batch.lastSteps", Collections.singletonList(getStep().getName()));

    return result;
  }
}

@Test
public void testGetSteps() throws Exception {
  SimpleFlow flow = new SimpleFlow("job");
  List<StateTransition> transitions = new ArrayList<>();
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "step2"));
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step2")), "end0"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end0")));
  flow.setStateTransitions(transitions);
  flow.afterPropertiesSet();
  job.setFlow(flow);
  job.afterPropertiesSet();
  assertEquals(2, job.getStepNames().size());
}

/**
 * Provides an extension point to provide alternative {@link StepState}
 * implementations within a {@link SimpleFlow}
 *
 * @param state The state that will be used to create the StepState
 * @param oldName The name to be replaced
 * @param stateName The name for the new State
 * @return a state for the requested data
 */
protected State createNewStepState(State state, String oldName,
    String stateName) {
  return new StepState(stateName, ((StepState) state).getStep(oldName));
}

  @Override
  public FlowExecutionStatus handle(FlowExecutor executor) throws Exception {
    FlowExecutionStatus result = super.handle(executor);

    executor.getJobExecution().getExecutionContext().put("batch.lastSteps", Collections.singletonList(getStep().getName()));

    return result;
  }
}

@Test
public void testGetStepNotStepState() throws Exception {
  SimpleFlow flow = new SimpleFlow("job");
  List<StateTransition> transitions = new ArrayList<>();
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "step2"));
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step2")), "end0"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end0")));
  flow.setStateTransitions(transitions);
  flow.afterPropertiesSet();
  job.setFlow(flow);
  job.afterPropertiesSet();
  Step step = job.getStep("end0");
  assertNull(step);
}

/**
 * Convenience method to get a state that proxies the input but with a
 * different name, appropriate to this flow. If the state is a StepState
 * then the step name is also changed.
 * 
 * @param state
 * @return
 */
private State getProxyState(State state) {
  String oldName = state.getName();
  if (oldName.startsWith(prefix)) {
    return state;
  }
  String stateName = prefix + oldName;
  if (state instanceof StepState) {
    return new StepState(stateName, ((StepState) state).getStep());
  }
  return new DelegateState(stateName, state);
}

@Test
public void testGetStepNotExists() throws Exception {
  SimpleFlow flow = new JsrFlow("job");
  List<StateTransition> transitions = new ArrayList<>();
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "step2"));
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step2")), "end0"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end0")));
  flow.setStateTransitions(transitions);
  flow.afterPropertiesSet();
  job.setFlow(flow);
  job.afterPropertiesSet();
  Step step = job.getStep("foo");
  assertNull(step);
}

@Test
public void testGetStepNotExists() throws Exception {
  SimpleFlow flow = new SimpleFlow("job");
  List<StateTransition> transitions = new ArrayList<>();
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "step2"));
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step2")), "end0"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end0")));
  flow.setStateTransitions(transitions);
  flow.afterPropertiesSet();
  job.setFlow(flow);
  job.afterPropertiesSet();
  Step step = job.getStep("foo");
  assertNull(step);
}

@Test
public void testGetStepNotStepState() throws Exception {
  SimpleFlow flow = new JsrFlow("job");
  List<StateTransition> transitions = new ArrayList<>();
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "step2"));
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step2")), "end0"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end0")));
  flow.setStateTransitions(transitions);
  flow.afterPropertiesSet();
  job.setFlow(flow);
  job.afterPropertiesSet();
  Step step = job.getStep("end0");
  assertNull(step);
}

@Test
public void testGetStepNestedFlow() throws Exception {
  SimpleFlow nested = new JsrFlow("nested");
  List<StateTransition> transitions = new ArrayList<>();
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step2")), "end1"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end1")));
  nested.setStateTransitions(transitions);
  nested.afterPropertiesSet();
  SimpleFlow flow = new JsrFlow("job");
  transitions = new ArrayList<>();
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "nested"));
  transitions.add(StateTransition.createStateTransition(new FlowState(nested, "nested"), "end0"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end0")));
  flow.setStateTransitions(transitions);
  flow.afterPropertiesSet();
  job.setFlow(flow);
  job.afterPropertiesSet();
  List<String> names = new ArrayList<>(job.getStepNames());
  Collections.sort(names);
  assertEquals("[step1, step2]", names.toString());
}

@Test
public void testBasicFlow() throws Throwable {
  SimpleFlow flow = new SimpleFlow("job");
  List<StateTransition> transitions = new ArrayList<>();
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step")), "end0"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end0")));
  flow.setStateTransitions(transitions);
  job.setFlow(flow);
  job.execute(jobExecution);
  if (!jobExecution.getAllFailureExceptions().isEmpty()) {
    throw jobExecution.getAllFailureExceptions().get(0);
  }
  assertEquals(BatchStatus.COMPLETED, jobExecution.getStatus());
}

@Test
public void testBasicFlow() throws Throwable {
  SimpleFlow flow = new JsrFlow("job");
  List<StateTransition> transitions = new ArrayList<>();
  transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step")), "end0"));
  transitions.add(StateTransition.createEndStateTransition(new EndState(FlowExecutionStatus.COMPLETED, "end0")));
  flow.setStateTransitions(transitions);
  job.setFlow(flow);
  job.execute(jobExecution);
  if (!jobExecution.getAllFailureExceptions().isEmpty()) {
    throw jobExecution.getAllFailureExceptions().get(0);
  }
  assertEquals(BatchStatus.COMPLETED, jobExecution.getStatus());
}

Javadoc

State implementation that delegates to a FlowExecutor to execute the specified Step.

Most used methods

Popular in Java

Running tasks concurrently on multiple threads
getContentResolver (Context)
compareTo (BigDecimal)
putExtra (Intent)
Table (com.google.common.collect)
A collection that associates an ordered pair of keys, called a row key and a column key, with a sing
InputStreamReader (java.io)
A class for turning a byte stream into a character stream. Data read from the source input stream is
Time (java.sql)
Java representation of an SQL TIME value. Provides utilities to format and parse the time's represen
Collections (java.util)
This class consists exclusively of static methods that operate on or return collections. It contains
GregorianCalendar (java.util)
GregorianCalendar is a concrete subclass of Calendarand provides the standard calendar used by most
HttpServlet (javax.servlet.http)
Provides an abstract class to be subclassed to create an HTTP servlet suitable for a Web site. A sub
Top plugins for WebStorm

How to useStepState in org.springframework.batch.core.job.flow.support.state

Best Java code snippets using org.springframework.batch.core.job.flow.support.state.StepState (Showing top 20 results out of 315)

How to use
StepState
in
org.springframework.batch.core.job.flow.support.state