org.springframework.batch.core.StepExecutionListener Java Examples

@Before
public void setupMock() {
	when(this.taskletStep.getTasklet()).thenReturn(
			new ChunkOrientedTasklet(this.chunkProvider, this.chunkProcessor));
	when(this.taskletStep.getName()).thenReturn("FOOOBAR");

	registerAlias(ItemProcessListener.class,
			BatchEventAutoConfiguration.ITEM_PROCESS_EVENTS_LISTENER);
	registerAlias(StepExecutionListener.class,
			BatchEventAutoConfiguration.STEP_EXECUTION_EVENTS_LISTENER);
	registerAlias(ChunkListener.class,
			BatchEventAutoConfiguration.CHUNK_EVENTS_LISTENER);
	registerAlias(ItemReadListener.class,
			BatchEventAutoConfiguration.ITEM_READ_EVENTS_LISTENER);
	registerAlias(ItemWriteListener.class,
			BatchEventAutoConfiguration.ITEM_WRITE_EVENTS_LISTENER);
	registerAlias(SkipListener.class,
			BatchEventAutoConfiguration.SKIP_EVENTS_LISTENER);
}
 

private Step createTaskletStepWithListener(final String taskName,
		StepExecutionListener stepExecutionListener) {
	return this.steps.get(taskName)
			.tasklet(new Tasklet() {
				@Override
				public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws Exception {
					return RepeatStatus.FINISHED;
				}
			})
			.transactionAttribute(getTransactionAttribute())
			.listener(stepExecutionListener)
			.build();
}
 

private StepExecutionListener releaserListener(NamedArgumentsSupplier argsSupplier,
		ReleaserTask releaserTask) {
	return new StepExecutionListenerSupport() {
		@Override
		public ExitStatus afterStep(StepExecution stepExecution) {
			Arguments args = argsSupplier.get();
			FlowRunner.Decision decision = afterTask(args.options, args.properties,
					releaserTask);
			if (decision == FlowRunner.Decision.ABORT) {
				return ExitStatus.FAILED;
			}
			ExecutionResult result = (ExecutionResult) stepExecution
					.getExecutionContext().get("result");
			if (result == null || result.isSuccess()) {
				return stepExecution.getExitStatus();
			}
			else if (result.isUnstable()) {
				return ExitStatus.COMPLETED
						.addExitDescription(BuildUnstableException.DESCRIPTION);
			}
			else if (result.isFailure()) {
				return ExitStatus.FAILED;
			}
			return ExitStatus.COMPLETED;
		}
	};
}
 

private Step createTaskletStepWithListener(final String taskName,
		StepExecutionListener stepExecutionListener) {
	return this.steps.get(taskName)
			.tasklet(new Tasklet() {
				@Override
				public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws Exception {
					return RepeatStatus.FINISHED;
				}
			})
			.transactionAttribute(getTransactionAttribute())
			.listener(stepExecutionListener)
			.build();
}
 

@Bean
@ConditionalOnProperty(prefix = "spring.cloud.task.batch.events.step-execution",
		name = "enabled", havingValue = "true", matchIfMissing = true)
// @checkstyle:on
public StepExecutionListener stepExecutionEventsListener() {
	return new EventEmittingStepExecutionListener(
			this.listenerChannels.stepExecutionEvents(),
			this.taskEventProperties.getStepExecutionOrder());
}
 

private void registerStepExecutionEventListener(Object bean) {
	if (this.applicationContext.containsBean(
			BatchEventAutoConfiguration.STEP_EXECUTION_EVENTS_LISTENER)) {
		StepExecutionListener stepExecutionListener = (StepExecutionListener) this.applicationContext
				.getBean(BatchEventAutoConfiguration.STEP_EXECUTION_EVENTS_LISTENER);
		AbstractStep step = (AbstractStep) bean;
		step.registerStepExecutionListener(stepExecutionListener);
	}
}
 

@Bean
public StepExecutionListener composedTaskStepExecutionListener(TaskExplorer taskExplorer){
	return new ComposedTaskStepExecutionListener(taskExplorer);
}
 

@Bean
public StepExecutionListener composedTaskStepExecutionListener(TaskExplorer taskExplorer){
	return new org.springframework.cloud.dataflow.composedtaskrunner.ComposedTaskStepExecutionListener(taskExplorer);
}
 

@Override
public Set<StepExecutionListener> stepExecutionListeners() {
	Set<StepExecutionListener> listeners = new HashSet<>();
	listeners.add(metricsListener());
	return listeners;
}
 

@Override
public Set<StepExecutionListener> stepExecutionListeners() {
	return new HashSet<StepExecutionListener>();
}
 

/**
 * Returns a set of StepExecutionListeners that will be added to each Job. May not return null.
 * 
 * @return Returns a set of StepExecutionListeners that will be added to each Job. May not return null.
 */
public Set<StepExecutionListener> stepExecutionListeners();