将行写入csv文件时，如何在Spring Batch中的FlatFileItemWriter中包含计数器

Question

所以，我使用FlatFileItemWriter从数据库中成功读取的数据中编写csv文件。

我正在努力编写一个整数（即行计数器）对应于我正在写入文件的行。似乎是一件容易的事情，但很简单，我被难住了。

一切正常（从数据库读取的数据生成文件）。但我似乎不知道如何实现getCount（）方法，从而获得相应行的计数。我认为这与利用ChunkContext有关，但我似乎无法理解。

因此，在我的作业配置中，我在bean中有以下内容。

@Bean
public FlatFileItemWriter<Customer> customerItemWriter() throws Exception {

    FlatFileItemWriter<Customer> itemWriter = new FlatFileItemWriter<>();

    itemWriter.setLineAggregator(new CustomerLineAggregator());
    itemWriter.setResource(new FileSystemResource("/some/directory/file.csv"));
    itemWriter.afterPropertiesSet();

    return itemWriter;
}

我还有下面的LineAggregator实现。

public class CustomerLineAggregator implements LineAggregator<Customer> {
    
    private ChunkContext chunkContext;
    
    @BeforeChunk
    private void beforeChunk(ChunkContext chunkContext) {
        this.setChunkContext(chunkContext);
    }
        
    @Override
    html" target="_blank">public String aggregate(Customer item) {
        
        return getCount() + "," + convertTime(item.getTime(), 3) + " ET," + item.getCustomerId() + "," + item.getLink() + "," + item.getName();
    }
    
    
    private String convertTime(String timeString, int offset) {
        
        LocalTime timeObject = LocalTime.parse(timeString);
        
        LocalTime timeOffsetObject = timeObject.plusHours(offset);
        
        return timeOffsetObject.toString();
        
    }
    
    private String getCount() {
        // how do I make use of the chunk context to infer the integer corresponding to the item assuming my chunk size is 1.
        return someCountString;
    }
    

    public ChunkContext getChunkContext() {
        return chunkContext;
    }

    public void setChunkContext(ChunkContext chunkContext) {
        this.chunkContext = chunkContext;
    }
    

}

预期的输出文件应如下所示：

1, 10:09 ET, 742, https://www.firm.com, John Doe
2, 12:30 ET, 235, http://www.firm.com, Jane Doe
3, 9:21 ET, 398, http://www.thomas.io, Thomas Chan
4, 14:38 ET, 104, http://www.googl.com, Cindy Chen

除了第一列，我可以成功地生成上述内容，第一列只是每个记录的行数。我尝试过的任何方法都无法在作业运行之间重置自己。

Answer 1

您可以为此使用ItemCountAware接口。此接口由您的域对象（在您的情况下似乎是Client）实现，并且将在读取时由任何扩展AbstractItemCounting ItemStreamItemReader的阅读器调用。

因此，如果你的读者是他们中的一员，你可以获得你物品的物品数量，并根据需要在线性聚集器中使用它。

编辑：当读取器未扩展AbstractItemCountingItemStreamItemReader

您始终可以在ItemReadListener#postRead中分配项目编号并在聚合器中使用它，例如：

class ItemNumberListener implements ItemReadListener<Customer> {
    int number;
    @Override
    public void afterRead(Customer customer) {
        customer.setNumber(number++);
    }
    // other methods from ItemReadListener
}

设置好后，在步骤中注册此侦听器，并在线性聚合器中使用项目编号。此方法适用于所有读卡器（基于光标或基于分页）。

注意：如果不能在Customer对象（可能是一个不能修改的类）中添加字段getter/setter，可以将其封装在自定义对象中，如numberawarecuster，并在批处理应用程序中使用该类型。