org.springframework.cloud.task.configuration.TaskProperties Java Examples

public TaskLauncherTasklet(
		TaskOperations taskOperations, TaskExplorer taskExplorer,
		ComposedTaskProperties composedTaskProperties, String taskName,
		TaskProperties taskProperties) {
	Assert.hasText(taskName, "taskName must not be empty nor null.");
	Assert.notNull(taskOperations, "taskOperations must not be null.");
	Assert.notNull(taskExplorer, "taskExplorer must not be null.");
	Assert.notNull(composedTaskProperties,
			"composedTaskProperties must not be null");

	this.taskName = taskName;
	this.taskOperations = taskOperations;
	this.taskExplorer = taskExplorer;
	this.composedTaskProperties = composedTaskProperties;
	this.taskProperties = taskProperties;
}
 

/**
 * @param taskRepository {@link TaskRepository} to record executions.
 * @param taskNameResolver {@link TaskNameResolver} used to determine task name for
 * task execution.
 * @param applicationArguments {@link ApplicationArguments} to be used for task
 * execution.
 * @param taskExplorer {@link TaskExplorer} to be used for task execution.
 * @param taskProperties {@link TaskProperties} to be used for the task execution.
 * @param taskListenerExecutorObjectFactory {@link TaskListenerExecutorObjectFactory}
 * to initialize TaskListenerExecutor for a task
 */
public TaskLifecycleListener(TaskRepository taskRepository,
		TaskNameResolver taskNameResolver, ApplicationArguments applicationArguments,
		TaskExplorer taskExplorer, TaskProperties taskProperties,
		TaskListenerExecutorObjectFactory taskListenerExecutorObjectFactory) {
	Assert.notNull(taskRepository, "A taskRepository is required");
	Assert.notNull(taskNameResolver, "A taskNameResolver is required");
	Assert.notNull(taskExplorer, "A taskExplorer is required");
	Assert.notNull(taskProperties, "TaskProperties is required");
	Assert.notNull(taskListenerExecutorObjectFactory,
			"A TaskListenerExecutorObjectFactory is required");

	this.taskRepository = taskRepository;
	this.taskNameResolver = taskNameResolver;
	this.applicationArguments = applicationArguments;
	this.taskExplorer = taskExplorer;
	this.taskProperties = taskProperties;
	this.taskListenerExecutorObjectFactory = taskListenerExecutorObjectFactory;
	this.taskMetrics = new TaskMetrics();
}
 

@Test
@DirtiesContext
public void testTaskLauncherTaskletWithTaskExecutionIdWithPreviousParentID() throws Exception{

	TaskLauncherTasklet taskLauncherTasklet = prepTaskLauncherTests();
	TaskProperties taskProperties = new TaskProperties();
	taskProperties.setExecutionid(88L);
	mockReturnValForTaskExecution(2L);
	ChunkContext chunkContext = chunkContext();
	createCompleteTaskExecution(0);
	chunkContext.getStepContext()
			.getStepExecution().getExecutionContext().put("task-arguments", new ArrayList<String>());
	((List)chunkContext.getStepContext()
			.getStepExecution().getExecutionContext()
			.get("task-arguments")).add("--spring.cloud.task.parent-execution-id=84");
	taskLauncherTasklet = getTaskExecutionTasklet(taskProperties);
	taskLauncherTasklet.setArguments(null);
	execute(taskLauncherTasklet, null, chunkContext);
	assertEquals(2L, chunkContext.getStepContext()
			.getStepExecution().getExecutionContext()
			.get("task-execution-id"));
	assertEquals("--spring.cloud.task.parent-execution-id=88", ((List)chunkContext.getStepContext()
			.getStepExecution().getExecutionContext()
			.get("task-arguments")).get(0));
}
 

public TaskLauncherTasklet(
		TaskOperations taskOperations, TaskExplorer taskExplorer,
		ComposedTaskProperties composedTaskProperties, String taskName,
		TaskProperties taskProperties) {
	Assert.hasText(taskName, "taskName must not be empty nor null.");
	Assert.notNull(taskOperations, "taskOperations must not be null.");
	Assert.notNull(taskExplorer, "taskExplorer must not be null.");
	Assert.notNull(composedTaskProperties,
			"composedTaskProperties must not be null");

	this.taskName = taskName;
	this.taskOperations = taskOperations;
	this.taskExplorer = taskExplorer;
	this.composedTaskProperties = composedTaskProperties;
	this.taskProperties = taskProperties;
}
 

@Test
@DirtiesContext
public void testTaskLauncherTaskletWithTaskExecutionId() throws Exception{
	TaskLauncherTasklet taskLauncherTasklet = prepTaskLauncherTests();

	TaskProperties taskProperties = new TaskProperties();
	taskProperties.setExecutionid(88L);
	mockReturnValForTaskExecution(2L);
	ChunkContext chunkContext = chunkContext();
	createCompleteTaskExecution(0);
	taskLauncherTasklet = getTaskExecutionTasklet(taskProperties);
	taskLauncherTasklet.setArguments(null);
	execute(taskLauncherTasklet, null, chunkContext);
	assertEquals(2L, chunkContext.getStepContext()
			.getStepExecution().getExecutionContext()
			.get("task-execution-id"));
	assertEquals("--spring.cloud.task.parent-execution-id=88", ((List)chunkContext.getStepContext()
			.getStepExecution().getExecutionContext()
			.get("task-arguments")).get(0));
}
 

@Test
@DirtiesContext
public void testTaskLauncherTaskletWithTaskExecutionId() throws Exception{
	createCompleteTaskExecution(0);
	TaskLauncherTasklet taskLauncherTasklet =
			getTaskExecutionTasklet();
	ChunkContext chunkContext = chunkContext();
	mockReturnValForTaskExecution(1L);
	execute(taskLauncherTasklet, null, chunkContext);
	assertEquals(1L, chunkContext.getStepContext()
			.getStepExecution().getExecutionContext()
			.get("task-execution-id"));
	assertNull(chunkContext.getStepContext()
			.getStepExecution().getExecutionContext()
			.get("task-arguments"));

	TaskProperties taskProperties = new TaskProperties();
	taskProperties.setExecutionid(88l);
	mockReturnValForTaskExecution(2L);
	chunkContext = chunkContext();
	createCompleteTaskExecution(0);
	taskLauncherTasklet = getTaskExecutionTasklet(taskProperties);
	taskLauncherTasklet.setArguments(null);
	execute(taskLauncherTasklet, null, chunkContext);
	assertEquals(2L, chunkContext.getStepContext()
			.getStepExecution().getExecutionContext()
			.get("task-execution-id"));
	assertEquals("--spring.cloud.task.parent-execution-id=88", ((List)chunkContext.getStepContext()
			.getStepExecution().getExecutionContext()
			.get("task-arguments")).get(0));
}
 

@Override
public void afterPropertiesSet() throws Exception {
	boolean isInitializeEnabled = (this.taskProperties.isInitializeEnabled() != null)
			? this.taskProperties.isInitializeEnabled()
			: this.taskInitializationEnabled;
	if (this.dataSource != null && isInitializeEnabled && this.taskProperties
			.getTablePrefix().equals(TaskProperties.DEFAULT_TABLE_PREFIX)) {
		String platform = getDatabaseType(this.dataSource);
		if ("hsql".equals(platform)) {
			platform = "hsqldb";
		}
		if ("postgres".equals(platform)) {
			platform = "postgresql";
		}
		if ("oracle".equals(platform)) {
			platform = "oracle10g";
		}
		if ("mysql".equals(platform)) {
			platform = "mysql";
		}
		if ("sqlserver".equals(platform)) {
			platform = "sqlserver";
		}
		ResourceDatabasePopulator populator = new ResourceDatabasePopulator();
		String schemaLocation = schema;
		schemaLocation = schemaLocation.replace("@@platform@@", platform);
		populator.addScript(this.resourceLoader.getResource(schemaLocation));
		populator.setContinueOnError(true);
		logger.debug(
				String.format("Initializing task schema for %s database", platform));
		DatabasePopulatorUtils.execute(populator, this.dataSource);
	}
}
 

/**
 * @param dataSource  used by the dao to execute queries and updates the tables.
 * @param taskProperties the {@link TaskProperties} to use for this dao.
 */
public JdbcDataflowTaskExecutionDao(DataSource dataSource, TaskProperties taskProperties) {
	Assert.notNull(dataSource, "The dataSource must not be null.");
	Assert.notNull(taskProperties, "taskProperties must not be null");
	this.jdbcTemplate = new NamedParameterJdbcTemplate(dataSource);
	this.taskProperties = taskProperties;
}
 

@Bean
DataflowTaskExecutionDao dataflowTaskExecutionDao(DataSource dataSource, TaskProperties taskProperties) {
	return new JdbcDataflowTaskExecutionDao(dataSource, taskProperties);
}
 

private TaskLauncherTasklet getTaskExecutionTasklet() {
	return getTaskExecutionTasklet(new TaskProperties());
}
 

public TaskRepositoryInitializer(TaskProperties taskProperties) {
	this.taskProperties = taskProperties;
}
 

private TaskLauncherTasklet getTaskExecutionTasklet(TaskProperties taskProperties) {
	return new TaskLauncherTasklet(this.taskOperations,
			this.taskExplorer, this.composedTaskProperties,
			TASK_NAME, taskProperties);
}
 

@Bean
public DataflowTaskExecutionDao dataflowTaskExecutionDao(DataSource dataSource) {
	return new JdbcDataflowTaskExecutionDao(dataSource, new TaskProperties());
}
 

@Bean
public DataflowTaskExecutionDao dataflowTaskExecutionDao(DataSource dataSource) {
	return new JdbcDataflowTaskExecutionDao(dataSource, new TaskProperties());
}
 

@Bean
public DataflowTaskExecutionDao dataflowTaskExecutionDao(DataSource dataSource, TaskProperties taskProperties) {
	return new JdbcDataflowTaskExecutionDao(dataSource, taskProperties);
}
 

@Bean
public DataflowTaskExecutionDao dataflowTaskExecutionDao(DataSource dataSource) {
	return new JdbcDataflowTaskExecutionDao(dataSource, new TaskProperties());
}
 

@Bean
public DataflowTaskExecutionDao dataflowTaskExecutionDao(DataSource dataSource, TaskProperties taskProperties) {
	return new JdbcDataflowTaskExecutionDao(dataSource, taskProperties);
}
 

@Bean
public ComposedTaskRunnerStepFactory stepFactory(TaskProperties taskProperties) {
	return new ComposedTaskRunnerStepFactory(new ComposedTaskProperties(), "FOOBAR");
}
 

@Bean
public TaskProperties taskProperties() {
	return new TaskProperties();
}
 

@Bean
public TaskProperties taskProperties() {
	return new TaskProperties();
}
 

@Bean
public ComposedTaskRunnerStepFactory stepFactory(TaskProperties taskProperties) {
	return new ComposedTaskRunnerStepFactory(new ComposedTaskProperties(), "FOOBAR");
}
 

@Bean
public TaskProperties taskProperties() {
	return new TaskProperties();
}
 

@Bean
TaskRepositoryInitializer taskRepositoryInitializer() {
	return new TaskRepositoryInitializer(new TaskProperties());
}
 

private TaskLauncherTasklet getTaskExecutionTasklet(TaskProperties taskProperties) {
	return new TaskLauncherTasklet(this.taskOperations,
			this.taskExplorer, this.composedTaskProperties,
			TASK_NAME, taskProperties);
}
 

private TaskLauncherTasklet getTaskExecutionTasklet() {
	return getTaskExecutionTasklet(new TaskProperties());
}