Use header_names instead of column_N if available

mmolimar · Jul 11, 2020 · aa169ef · aa169ef
1 parent 7ec9293
commit aa169ef
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/main/java/com/github/mmolimar/kafka/connect/fs/file/reader/UnivocityFileReader.java b/src/main/java/com/github/mmolimar/kafka/connect/fs/file/reader/UnivocityFileReader.java
@@ -84,7 +84,12 @@ public UnivocityFileReader(FileSystem fs, Path filePath, Map<String, Object> con
         super(fs, filePath, new UnivocityToStruct(), config);
 
         this.iterator = iterateRecords();
-        this.schema = buildSchema(this.iterator, settings.isHeaderExtractionEnabled(), config);
+        Boolean hasHeader = settings.isHeaderExtractionEnabled();
+        if (!hasHeader) {
+            String[] headers = settings.getHeaders();
+            hasHeader = headers != null && headers.length > 0;
+        }
+        this.schema = buildSchema(this.iterator, hasHeader, config);
     }
 
     private Schema buildSchema(ResultIterator<Record, ParsingContext> it, boolean hasHeader, Map<String, Object> config) {