Jeff Cook
Jeff Cook

Reputation: 8774

Pass JobParameters and ExecutionContext to @Bean Tasklet?

I am converting Spring Batch XML based application into the Spring Boot annotations. In my XML file, I am not sure how to pass those jobParameters and execution context to tasklet ?

<beans xmlns="http://www.springframework.org/schema/beans"
    xmlns:batch="http://www.springframework.org/schema/batch" xmlns:jdbc="http://www.springframework.org/schema/jdbc"
    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:context="http://www.springframework.org/schema/context"
    xmlns:task="http://www.springframework.org/schema/task" xmlns:util="http://www.springframework.org/schema/util"
    xsi:schemaLocation="http://www.springframework.org/schema/jdbc http://www.springframework.org/schema/jdbc/spring-jdbc.xsd
        http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd
        http://www.springframework.org/schema/batch http://www.springframework.org/schema/batch/spring-batch.xsd
        http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context.xsd
        http://www.springframework.org/schema/task  http://www.springframework.org/schema/task/spring-task.xsd
        http://www.springframework.org/schema/util http://www.springframework.org/schema/util/spring-util.xsd">

    <batch:job id="job1">
        <batch:step id="contextStep" >
            <batch:tasklet ref="contextTasklet" />
        </batch:step>
    </batch:job>

    <bean id="contextTasklet" class="com.XXX.batch.tasklet.ContextTasklet" scope="step">
        <property name="runMode" value="#{jobParameters['runMode']}" />
        <property name="executionContext" value="#{stepExecution.jobExecution.executionContext}" />
    </bean>
</beans>

ContextTasklet.java

@Data
@Configuration
@StepScope
public class ContextTasklet implements Tasklet{

    private String runMode;
    private ExecutionContext executionContext;

    @Override
    public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws Exception {
        chunkContext.getStepContext().getStepExecution().getJobExecution().getExecutionContext().put("mode",
                runMode);

        return RepeatStatus.FINISHED;
    }
}

Upvotes: 1

Views: 3345

Answers (1)

Mahmoud Ben Hassine
Mahmoud Ben Hassine

Reputation: 31600

You can use the @Value annotation to inject job parameters and execution context values:

@Data
@Configuration
@StepScope
public class ContextTasklet implements Tasklet{

   @Value("#{jobParameters['runMode']}")
   private String runMode;

   @Value("#{stepExecution.jobExecution.executionContext}")
   private ExecutionContext executionContext;

   @Override
   public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws Exception {
       chunkContext.getStepContext().getStepExecution().getJobExecution().getExecutionContext().put("mode",
            runMode);

       return RepeatStatus.FINISHED;
    }
}

Upvotes: 1

Related Questions