diff options
-rw-r--r-- | src/main/java/it/alessandroiezzi/csv/CSVParser.java | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/main/java/it/alessandroiezzi/csv/CSVParser.java b/src/main/java/it/alessandroiezzi/csv/CSVParser.java index 413d5d4..cf78c8c 100644 --- a/src/main/java/it/alessandroiezzi/csv/CSVParser.java +++ b/src/main/java/it/alessandroiezzi/csv/CSVParser.java @@ -11,10 +11,12 @@ import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; +import java.util.function.BiConsumer; public class CSVParser<T> { private final InputStream csvFile; private String separator = ";"; + private final Map<String, BiConsumer<T, String>> mappers = new HashMap<>(); private final Class<T> cls; private void checkDefaultConstructor() { @@ -31,6 +33,11 @@ public class CSVParser<T> { checkDefaultConstructor(); } + public CSVParser<T> map(String csvHeader, BiConsumer<T, String> func) { + mappers.put(csvHeader, func); + return this; + } + public CSVParser<T> withSeparator(char separator) { this.separator = "" + separator; return this; |