首页 > 解决方案 > spring batch csv:向csv添加多个标题

问题描述

如何在 csv 中编写多个标头,以便第二个标头值应来自数据库

//预期输出

personId,firstName,lastName,email,age
fullname,total // this is the second header which  should come the database
kaa,karthi,sa,123@,34

//下面是spring批量写入csv的代码片段

@Bean(destroyMethod="")
    public JdbcCursorItemReader<Person> reader(){
        JdbcCursorItemReader<Person> cursorItemReader = new JdbcCursorItemReader<>();
        cursorItemReader.setDataSource(dataSource);
        cursorItemReader.setSql("SELECT person_id,first_name,last_name,email,age FROM springbatchdb.person");
        cursorItemReader.setRowMapper(new PersonRowMapper());
        return cursorItemReader;
    }
    @Bean(destroyMethod="")
    public PersonItenProcessor processor(){
        return new PersonItenProcessor();
    }

    @Bean(destroyMethod="")
    public FlatFileItemWriter<Person> writer(){
        FlatFileItemWriter<Person> writer = new FlatFileItemWriter<Person>();
        String exportFileHeader = "personId,firstName,lastName,email,age";
        StringHeaderWriter headerWriter = new StringHeaderWriter(exportFileHeader);
        writer.setHeaderCallback(headerWriter);

        writer.setResource(new ClassPathResource("person.csv"));
        DelimitedLineAggregator<Person> lineAggregator = new DelimitedLineAggregator<Person>();
        lineAggregator.setDelimiter(",");

        BeanWrapperFieldExtractor<Person>  fieldExtractor = new BeanWrapperFieldExtractor<Person>();
        fieldExtractor.setNames(new String[]{"personId","firstName","lastName","email","age"});
        lineAggregator.setFieldExtractor(fieldExtractor);

        writer.setLineAggregator(lineAggregator);
        return writer;
}

标签: javaspringspring-bootcsvspring-batch

解决方案


您可以在 a 中执行查询FlatFileHeaderCallback并将结果附加到标题的第一行。


推荐阅读