在映射器中使用jobParameters

时间:2020-08-17 11:24:03

标签: spring spring-batch jobs

我正在使用Spring Batch创建作业,该作业是从骆驼处理器手动启动的。

我需要将一些参数传递给我的工作,其中一个参数需要在我的工作的映射器中使用,但我找不到实现此目标的方法。

这是我的工作:

    @Bean
    public LineMapper<MyEntity> lineMapper() {
        DefaultLineMapper<MyEntity> lineMapper = new DefaultLineMapper<MyEntity>();
        DelimitedLineTokenizer lineTokenizer = new DelimitedLineTokenizer() {
            {
                setNames(new String[] { "val1", "val2" });
            }
        };
        lineTokenizer.setDelimiter(";");

        lineMapper.setLineTokenizer(lineTokenizer);

        lineMapper.setFieldSetMapper(new MyEntityFieldsetMapper());

        return lineMapper;
    }

    public class MyEntityFieldsetMapper implements FieldSetMapper<MyEntity> {

        @Override
        public MyEntity mapFieldSet(FieldSet fieldSet) throws BindException {
            MyEntity MyEntity = new MyEntity();
            
            MyEntity.setValOne(fieldSet.readString("val1"));
            MyEntity.setValTwo(fieldSet.readString("val2"));
            
            // Here I need to set the variable in job parameters
            MyEntity.setValThree();

            return MyEntity;
        }
    }

    @Bean
    @StepScope
    public FlatFileItemReader<MyEntity> MyEntityFlatFileItemReader(
            @Value("#{jobParameters[inputCsvPath]}")
                    String inputCsvPath, @Value("#{jobParameters[myCustomVal]}")
                    String myCustomVal) {
        //Create reader instance
        FlatFileItemReader<MyEntity> reader = new FlatFileItemReader<MyEntity>();

        //Set input file location
        reader.setResource(new FileSystemResource(inputCsvPath));

        //Set number of lines to skips. Use it if file has header rows.
        reader.setLinesToSkip(0);

        //Configure how each line will be parsed and mapped to different values
        reader.setLineMapper(lineMapper());
        return reader;
    }

    @Bean
    public Job readCSVFilesJob(JobBuilderFactory jobBuilderFactory, StepBuilderFactory stepBuilderFactory,
                               ItemReader<MyEntity> MyEntityItemReader, ItemWriter<MyEntity> MyEntityWriter) {

        Step step = stepBuilderFactory.get("step").<MyEntity, MyEntity>chunk(10000).reader(MyEntityItemReader)
                .writer(MyEntityWriter).build();

        return jobBuilderFactory.get("readCSVFilesJob").incrementer(new RunIdIncrementer()).start(step).build();
    }

我可以在jobParameters中获得ItemReader,但是不能将其传递给我的lineMapperFieldSetMapper

有没有办法做到这一点?

1 个答案:

答案 0 :(得分:1)

使LineMapper bean步骤的作用域确定范围,并像为阅读器所做的那样在其中注入作业参数,例如:

@Bean
@StepScope
public LineMapper<MyEntity> lineMapper(@Value("#{jobParameters[inputCsvPath]}") String inputCsvPath) {
   // use job parameter here
}

如果还想在其中插入作业参数,则可能还需要声明MyEntityFieldsetMapper作为步骤作用域bean。