Function sparse_union_array

datafusion/common/benches/with_hashes.rs:432–460

(num_rows: usize)

}

fn sparse_union_array(num_rows: usize) -> ArrayRef {
    let mut rng = make_rng();
    let num_types = 5;

    let type_ids: Vec<i8> = (0..num_rows)
        .map(|_| rng.random_range(0..num_types) as i8)
        .collect();
    let (fields, children): (Vec<_>, Vec<_>) = (0..num_types)
        .map(|i| {
            (
                (
                    i as i8,
                    Arc::new(Field::new(format!("f{i}"), DataType::Int64, true)),
                ),
                primitive_array::<Int64Type>(num_rows),
            )
        })
        .unzip();

    Arc::new(
        UnionArray::try_new(
            UnionFields::from_iter(fields),
            ScalarBuffer::from(type_ids),
            None,
            children,
        )
        .unwrap(),
    )
}

fn dense_union_array(num_rows: usize) -> ArrayRef {
    let mut rng = make_rng();