in tools/parquet_reader.cc [24:79]
int main(int argc, char** argv) {
if (argc > 5 || argc < 2) {
std::cerr << "Usage: parquet_reader [--only-metadata] [--no-memory-map] [--json]"
"[--print-key-value-metadata] [--columns=...] <file>"
<< std::endl;
return -1;
}
std::string filename;
bool print_values = true;
bool print_key_value_metadata = false;
bool memory_map = true;
bool format_json = false;
// Read command-line options
const std::string COLUMNS_PREFIX = "--columns=";
std::list<int> columns;
char *param, *value;
for (int i = 1; i < argc; i++) {
if ((param = std::strstr(argv[i], "--only-metadata"))) {
print_values = false;
} else if ((param = std::strstr(argv[i], "--print-key-value-metadata"))) {
print_key_value_metadata = true;
} else if ((param = std::strstr(argv[i], "--no-memory-map"))) {
memory_map = false;
} else if ((param = std::strstr(argv[i], "--json"))) {
format_json = true;
} else if ((param = std::strstr(argv[i], COLUMNS_PREFIX.c_str()))) {
value = std::strtok(param + COLUMNS_PREFIX.length(), ",");
while (value) {
columns.push_back(std::atoi(value));
value = std::strtok(nullptr, ",");
}
} else {
filename = argv[i];
}
}
try {
std::unique_ptr<parquet::ParquetFileReader> reader =
parquet::ParquetFileReader::OpenFile(filename, memory_map);
parquet::ParquetFilePrinter printer(reader.get());
if (format_json) {
printer.JSONPrint(std::cout, columns, filename.c_str());
} else {
printer.DebugPrint(std::cout, columns, print_values,
print_key_value_metadata, filename.c_str());
}
} catch (const std::exception& e) {
std::cerr << "Parquet error: " << e.what() << std::endl;
return -1;
}
return 0;
}