|
30 | 30 | #ifdef ARROW_CSV |
31 | 31 | #include "arrow/dataset/file_csv.h" |
32 | 32 | #endif |
| 33 | +#ifdef ARROW_ORC |
| 34 | +#include "arrow/dataset/file_orc.h" |
| 35 | +#endif |
| 36 | +#ifdef ARROW_JSON |
| 37 | +#include "arrow/dataset/file_json.h" |
| 38 | +#endif |
33 | 39 | #include "arrow/filesystem/api.h" |
34 | 40 | #include "arrow/filesystem/path_util.h" |
35 | 41 | #include "arrow/engine/substrait/util.h" |
@@ -141,20 +147,25 @@ arrow::Result<std::shared_ptr<arrow::dataset::FileFormat>> GetFileFormat( |
141 | 147 | jint file_format_id) { |
142 | 148 | switch (file_format_id) { |
143 | 149 | case 0: |
144 | | - return std::make_shared<arrow::dataset::ParquetFileFormat>(); |
| 150 | + return std::static_pointer_cast<arrow::dataset::FileFormat>( |
| 151 | + std::make_shared<arrow::dataset::ParquetFileFormat>()); |
145 | 152 | case 1: |
146 | | - return std::make_shared<arrow::dataset::IpcFileFormat>(); |
| 153 | + return std::static_pointer_cast<arrow::dataset::FileFormat>( |
| 154 | + std::make_shared<arrow::dataset::IpcFileFormat>()); |
147 | 155 | #ifdef ARROW_ORC |
148 | 156 | case 2: |
149 | | - return std::make_shared<arrow::dataset::OrcFileFormat>(); |
| 157 | + return std::static_pointer_cast<arrow::dataset::FileFormat>( |
| 158 | + std::make_shared<arrow::dataset::OrcFileFormat>()); |
150 | 159 | #endif |
151 | 160 | #ifdef ARROW_CSV |
152 | 161 | case 3: |
153 | | - return std::make_shared<arrow::dataset::CsvFileFormat>(); |
| 162 | + return std::static_pointer_cast<arrow::dataset::FileFormat>( |
| 163 | + std::make_shared<arrow::dataset::CsvFileFormat>()); |
154 | 164 | #endif |
155 | 165 | #ifdef ARROW_JSON |
156 | 166 | case 4: |
157 | | - return std::make_shared<arrow::dataset::JsonFileFormat>(); |
| 167 | + return std::static_pointer_cast<arrow::dataset::FileFormat>( |
| 168 | + std::make_shared<arrow::dataset::JsonFileFormat>()); |
158 | 169 | #endif |
159 | 170 | default: |
160 | 171 | std::string error_message = |
|
0 commit comments