Spring/spring-batch
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add method to set a custom BufferedReaderFactory in FlatFileItemReaderBuilder

Browse Source 
Resolves BATCH-2770
This commit is contained in:
Mahmoud Ben Hassine
2019-05-09 16:48:53 +02:00
parent e6a84e1ac2
commit 4e7c64a443
2 changed files with 26 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
spring-batch-infrastructure/src/main/java/org/springframework/batch/item/file/builder/FlatFileItemReaderBuilder.java
View File

@@ -1,5 +1,5 @@
/*
* Copyright 2016-2018 the original author or authors.
* Copyright 2016-2019 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -29,6 +29,8 @@ import java.util.Set;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.batch.item.file.BufferedReaderFactory;
import org.springframework.batch.item.file.DefaultBufferedReaderFactory;
import org.springframework.batch.item.file.FlatFileItemReader;
import org.springframework.batch.item.file.LineCallbackHandler;
import org.springframework.batch.item.file.LineMapper;
@@ -69,6 +71,9 @@ public class FlatFileItemReaderBuilder<T> {
private RecordSeparatorPolicy recordSeparatorPolicy =
new SimpleRecordSeparatorPolicy();
private BufferedReaderFactory bufferedReaderFactory =
new DefaultBufferedReaderFactory();
private Resource resource;
private List<String> comments = new ArrayList<>();
@@ -201,6 +206,19 @@ public class FlatFileItemReaderBuilder<T> {
return this;
}
/**
* Configure a custom {@link BufferedReaderFactory} for the reader.
*
* @param factory custom factory
* @return The current instance of the builder.
* @see FlatFileItemReader#setBufferedReaderFactory(BufferedReaderFactory)
*/
public FlatFileItemReaderBuilder<T> bufferedReaderFactory(BufferedReaderFactory factory) {
this.bufferedReaderFactory = factory;
return this;
}
/**
* The {@link Resource} to be used as input.
*
@@ -429,6 +447,7 @@ public class FlatFileItemReaderBuilder<T> {
}
Assert.notNull(this.recordSeparatorPolicy, "A RecordSeparatorPolicy is required.");
Assert.notNull(this.bufferedReaderFactory, "A BufferedReaderFactory is required.");
int validatorValue = this.tokenizerValidator.intValue();
FlatFileItemReader<T> reader = new FlatFileItemReader<>();
@@ -500,6 +519,7 @@ public class FlatFileItemReaderBuilder<T> {
reader.setSkippedLinesCallback(this.skippedLinesCallback);
reader.setRecordSeparatorPolicy(this.recordSeparatorPolicy);
reader.setBufferedReaderFactory(this.bufferedReaderFactory);
reader.setMaxItemCount(this.maxItemCount);
reader.setCurrentItemCount(this.currentItemCount);
reader.setSaveState(this.saveState);