Function criterion_benchmark

arrow-array/benches/union_array.rs:49–77

(c: &mut Criterion)

}

fn criterion_benchmark(c: &mut Criterion) {
    for with_nulls in 1..12 {
        for without_nulls in [0, 1, 10] {
            c.bench_function(
                &format!("union logical nulls 4096 {with_nulls} children with nulls, {without_nulls} without nulls"),
                |b| {
                    let type_ids = 0..with_nulls+without_nulls;

                    let fields = UnionFields::try_new(
                        type_ids.clone(),
                        type_ids.clone().map(|i| Field::new(format!("f{i}"), DataType::Int32, true)),
                    ).unwrap();

                    let array = UnionArray::try_new(
                        fields,
                        type_ids.cycle().take(4096).collect(),
                        None,
                        std::iter::repeat_n(array_with_nulls(), with_nulls as usize)
                            .chain(std::iter::repeat_n(array_without_nulls(), without_nulls as usize))
                            .collect(),
                    )
                    .unwrap();

                    b.iter(|| hint::black_box(array.logical_nulls()))
                },
            );
        }
    }
}

criterion_group!(benches, criterion_benchmark);
criterion_main!(benches);