Fix CSV File Scan Operator (#3217)

Fix the issue #3055 that CSV Scan an empty file would result in error. --------- Co-authored-by: Jiadong Bai <[email protected]>
Texera · Jan 20, 2025 · 586496c · 586496c
1 parent 58a214f
commit 586496c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...rator/src/main/scala/edu/uci/ics/amber/operator/source/scan/csv/CSVScanSourceOpDesc.scala b/...rator/src/main/scala/edu/uci/ics/amber/operator/source/scan/csv/CSVScanSourceOpDesc.scala
@@ -90,8 +90,11 @@ class CSVScanSourceOpDesc extends ScanSourceOpDesc {
     val attributeTypeList: Array[AttributeType] = inferSchemaFromRows(
       data.iterator.asInstanceOf[Iterator[Array[Any]]]
     )
+
     val header: Array[String] =
-      if (hasHeader) parser.getContext.headers()
+      if (hasHeader)
+        Option(parser.getContext.headers())
+          .getOrElse((1 to attributeTypeList.length).map(i => "column-" + i).toArray)
       else (1 to attributeTypeList.length).map(i => "column-" + i).toArray
 
     header.indices.foldLeft(Schema()) { (schema, i) =>