Spring Batch Reader With Job Parameters

1 min read

스프링 배치에서 잡 파라미터 사용하는 방법.

  • 빈을 @StepScope로 등록한다.
  • @Value("#{jobParameters[parameterName]}")으로 값을 주입받는다.
  • :parameterName 으로 쿼리를 지정한다.
  • 맵으로 값을 전달한다.

jobParameters

@Bean
@StepScope
public JpaPagingItemReader<User> reader(@Value("#{jobParameters[name]}") String name) {
  Map<String, Object> params = new HashMap<>();
  params.put("name", name);

  JpaPagingItemReader<User> reader = new JpaPagingItemReader<>();
  reader.setQueryString("select u from User u where u.name=:name");
  reader.setParameterValues(params);
  ...
}

stepExecution

    @StepScope
    public JpaPagingItemReader<Employee> taxCalculatorItemReader(@Value("#{stepExecution}") StepExecution stepExecution) {
        JpaPagingItemReader<Employee> employeeItemReader = new JpaPagingItemReader<>();
        employeeItemReader.setEntityManagerFactory(persistenceConfig.entityManagerFactory());
        employeeItemReader.setQueryString(TaxCalculation.GET_UNPROCESSED_EMPLOYEES_BY_YEAR_AND_MONTH_QUERY_SLAVE);
        Map<String, Object> parameters = new HashMap<>();
        parameters.put("year", stepExecution.getJobParameters().getLong("year").intValue());
        parameters.put("month", stepExecution.getJobParameters().getLong("month").intValue());
        parameters.put("jobExecutionId", stepExecution.getJobExecutionId());
        parameters.put("minId", stepExecution.getExecutionContext().getLong("minValue"));
        parameters.put("maxId", stepExecution.getExecutionContext().getLong("maxValue"));
        employeeItemReader.setParameterValues(parameters);
        return employeeItemReader;
    }

© 2023 Raegon Kim