如何在春季批次中使用读取具有未知列名和未知列数的csv



我的步骤有以下FlatFileItemReader配置:

@Bean
@StepScope
public FlatFileItemReader<RawInput> reader(FieldSetMapper<RawInput> fieldSetMapper, @Value("#{jobParameters['files.location']}") Resource resource) {
var reader = new FlatFileItemReader<RawInput>();
reader.setName("my-reader");
reader.setResource(resource);
var mapper = new DefaultLineMapper<RawInput>();
mapper.setLineTokenizer(crmCsvLineTokenizer());
mapper.setFieldSetMapper(fieldSetMapper);
mapper.afterPropertiesSet();
reader.setLineMapper(mapper);
return reader;
}

RawInput包含1个字段,因此它允许我用单列读取csv。目前需求已经发生了变化,现在我必须能够读取任何行数的csv文件,因此我需要以某种方式传递数组,而不是RawInput。是否可以使用FlatFileItemReader,或者我应该更改实现?

有效:

var reader = new FlatFileItemReader<List<String>>();
reader.setName("reader");
reader.setResource(resource);
//line mapper
var lineMapper = new DefaultLineMapper<List<String>>();        
lineMapper.setLineTokenizer(new DelimitedLineTokenizer());
lineMapper.setFieldSetMapper(myFieldSetMapper); // see implementation below
lineMapper.afterPropertiesSet();
reader.setLineMapper(lineMapper);
return reader;

@Component
public class MyFieldSetMapper implements FieldSetMapper<List<String>> {
@NonNull
@Override
public List<String> mapFieldSet(@NonNull FieldSet fieldSet) {
return Arrays.stream(fieldSet.getValues())
.map(StringUtils::lowerCase) // optional
.map(StringUtils::trimToNull) // optional
.collect(Collectors.toList());
}
}

最新更新