| 964 | void SetValues(); |
| 965 | |
| 966 | void WriteParquet() { |
| 967 | // Add writer properties |
| 968 | parquet::WriterProperties::Builder builder; |
| 969 | builder.compression(parquet::Compression::SNAPPY); |
| 970 | builder.created_by("parquet-cpp version 1.3.0"); |
| 971 | std::shared_ptr<parquet::WriterProperties> props = builder.build(); |
| 972 | |
| 973 | // Create a ParquetFileWriter instance |
| 974 | auto file_writer = parquet::ParquetFileWriter::Open(parquet_sink_, schema_, props); |
| 975 | |
| 976 | // Append a RowGroup with a specific number of rows. |
| 977 | auto rg_writer = file_writer->AppendRowGroup(); |
| 978 | |
| 979 | this->SetValues(); |
| 980 | |
| 981 | // Insert Values |
| 982 | for (int i = 0; i < static_cast<int>(fields_.size()); i++) { |
| 983 | auto column_writer = |
| 984 | static_cast<parquet::TypedColumnWriter<TestType>*>(rg_writer->NextColumn()); |
| 985 | column_writer->WriteBatch(NUM_VALUES, nullptr, nullptr, values_.data()); |
| 986 | } |
| 987 | } |
| 988 | |
| 989 | void VerifyParquetStats() { |
| 990 | ASSERT_OK_AND_ASSIGN(auto pbuffer, parquet_sink_->Finish()); |
no test coverage detected