Function map_array

datafusion/common/benches/with_hashes.rs:391–430

(num_rows: usize)

}

fn map_array(num_rows: usize) -> ArrayRef {
    let mut rng = make_rng();
    let entries_per_row = 5;
    let total_entries = num_rows * entries_per_row;

    let keys: Int32Array = (0..total_entries)
        .map(|_| Some(rng.random::<i32>()))
        .collect();
    let values: Int64Array = (0..total_entries)
        .map(|_| Some(rng.random::<i64>()))
        .collect();
    let offsets: Vec<i32> = (0..=num_rows)
        .map(|i| (i * entries_per_row) as i32)
        .collect();

    let entries = StructArray::try_new(
        Fields::from(vec![
            Field::new("keys", DataType::Int32, false),
            Field::new("values", DataType::Int64, true),
        ]),
        vec![Arc::new(keys), Arc::new(values)],
        None,
    )
    .unwrap();

    Arc::new(MapArray::new(
        Arc::new(Field::new(
            "entries",
            DataType::Struct(Fields::from(vec![
                Field::new("keys", DataType::Int32, false),
                Field::new("values", DataType::Int64, true),
            ])),
            false,
        )),
        OffsetBuffer::new(ScalarBuffer::from(offsets)),
        entries,
        None,
        false,
    ))
}

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