Function main

cpp/examples/parquet/low_level_api/encryption_reader_writer.cc:47–449

const arrow::util::SecureString kColumnEncryptionKey2("1234567890123451");

int main(int argc, char** argv) {
  /**********************************************************************************
                             PARQUET ENCRYPTION WRITER EXAMPLE
  **********************************************************************************/

  try {
    // Create a local file output stream instance.
    using FileClass = ::arrow::io::FileOutputStream;
    std::shared_ptr<FileClass> out_file;
    PARQUET_ASSIGN_OR_THROW(out_file, FileClass::Open(PARQUET_FILENAME));

    // Setup the parquet schema
    std::shared_ptr<GroupNode> schema = SetupSchema();

    // Add encryption properties
    // Encryption configuration: Encrypt two columns and the footer.
    std::map<std::string, std::shared_ptr<parquet::ColumnEncryptionProperties>>
        encryption_cols;

    parquet::SchemaDescriptor schema_desc;
    schema_desc.Init(schema);
    auto column_path1 = schema_desc.Column(5)->path()->ToDotString();
    auto column_path2 = schema_desc.Column(4)->path()->ToDotString();

    encryption_cols[column_path1] =
        parquet::ColumnEncryptionProperties::WithColumnKey(kColumnEncryptionKey1, "kc1");
    encryption_cols[column_path2] =
        parquet::ColumnEncryptionProperties::WithColumnKey(kColumnEncryptionKey2, "kc2");

    parquet::FileEncryptionProperties::Builder file_encryption_builder(
        kFooterEncryptionKey);

    parquet::WriterProperties::Builder builder;
    // Add the current encryption configuration to WriterProperties.
    builder.encryption(file_encryption_builder.footer_key_metadata("kf")
                           ->encrypted_columns(std::move(encryption_cols))
                           ->build());

    // Add other writer properties
    builder.compression(parquet::Compression::SNAPPY);

    std::shared_ptr<parquet::WriterProperties> props = builder.build();

    // Create a ParquetFileWriter instance
    std::shared_ptr<parquet::ParquetFileWriter> file_writer =
        parquet::ParquetFileWriter::Open(out_file, schema, props);

    // Append a RowGroup with a specific number of rows.
    parquet::RowGroupWriter* rg_writer = file_writer->AppendRowGroup();

    // Write the Bool column
    parquet::BoolWriter* bool_writer =
        static_cast<parquet::BoolWriter*>(rg_writer->NextColumn());
    for (int i = 0; i < NUM_ROWS_PER_ROW_GROUP; i++) {
      bool value = ((i % 2) == 0) ? true : false;
      bool_writer->WriteBatch(1, nullptr, nullptr, &value);
    }