Function open_writer

benchmarks/src/bin/gen_wide_data.rs:233–250

(
    path: &Path,
    schema: SchemaRef,
    rows_per_group: usize,
)

}

fn open_writer(
    path: &Path,
    schema: SchemaRef,
    rows_per_group: usize,
) -> Result<ArrowWriter<File>> {
    if let Some(parent) = path.parent() {
        std::fs::create_dir_all(parent)
            .map_err(|e| exec_datafusion_err!("creating {}: {e}", parent.display()))?;
    }
    let file = File::create(path)
        .map_err(|e| exec_datafusion_err!("creating {}: {e}", path.display()))?;
    let writer_props = WriterProperties::builder()
        .set_max_row_group_row_count(Some(rows_per_group.max(1)))
        .set_compression(Compression::ZSTD(ZstdLevel::try_new(1).unwrap()))
        .build();
    ArrowWriter::try_new(file, schema, Some(writer_props))
        .map_err(|e| exec_datafusion_err!("creating ArrowWriter: {e}"))
}

fn main() -> Result<()> {
    let cli = Cli::parse();