diff --git a/src/supplemental/nanolib/parquet/parquet.cc b/src/supplemental/nanolib/parquet/parquet.cc index dc4d9870e..4f63cffed 100644 --- a/src/supplemental/nanolib/parquet/parquet.cc +++ b/src/supplemental/nanolib/parquet/parquet.cc @@ -1,4 +1,5 @@ #include +#include #include #include @@ -450,6 +451,12 @@ parquet_write_core(conf_parquet *conf, char *filename, shared_ptr file_writer = parquet::ParquetFileWriter::Open(out_file, schema, props); + static int num = 0; + shared_ptr key_value_metadata = make_shared(); + key_value_metadata->Append("number", std::to_string(num++)); + + file_writer->AddKeyValueMetadata(key_value_metadata); + // Append a RowGroup with a specific number of rows. parquet::RowGroupWriter *rg_writer = file_writer->AppendRowGroup();