Dataset Java API: Add CSV support

zhztheplayer · Feb 8, 2022 · 75bea9a · 75bea9a
1 parent 5cd0394
commit 75bea9a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/cpp/src/jni/dataset/jni_wrapper.cc b/cpp/src/jni/dataset/jni_wrapper.cc
@@ -82,6 +82,8 @@ arrow::Result<std::shared_ptr<arrow::dataset::FileFormat>> GetFileFormat(
   switch (file_format_id) {
     case 0:
       return std::make_shared<arrow::dataset::ParquetFileFormat>();
+    case 1:
+      return std::make_shared<arrow::dataset::CsvFileFormat>();
     default:
       std::string error_message =
           "illegal file format id: " + std::to_string(file_format_id);

diff --git a/java/dataset/src/main/java/org/apache/arrow/dataset/file/FileFormat.java b/java/dataset/src/main/java/org/apache/arrow/dataset/file/FileFormat.java
@@ -22,6 +22,7 @@
  */
 public enum FileFormat {
   PARQUET(0),
+  CSV(1),
   NONE(-1);
 
   private int id;