Function make_map_batch

datafusion/functions-nested/src/map.rs:66–107

(args: &[ColumnarValue])

}

fn make_map_batch(args: &[ColumnarValue]) -> Result<ColumnarValue> {
    let [keys_arg, values_arg] = take_function_args("make_map", args)?;

    let can_evaluate_to_const = can_evaluate_to_const(args);

    let keys = get_first_array_ref(keys_arg)?;
    let key_array = keys.as_ref();

    match keys_arg {
        ColumnarValue::Array(_) => match key_array.data_type() {
            DataType::List(_) => keys
                .as_list::<i32>()
                .iter()
                .flatten()
                .try_for_each(|row| validate_map_keys(row.as_ref()))?,
            DataType::LargeList(_) => keys
                .as_list::<i64>()
                .iter()
                .flatten()
                .try_for_each(|row| validate_map_keys(row.as_ref()))?,
            DataType::FixedSizeList(_, _) => {
                keys.as_fixed_size_list()
                    .iter()
                    .flatten()
                    .try_for_each(|row| validate_map_keys(row.as_ref()))?
            }
            data_type => {
                return exec_err!(
                    "Expected list, large_list or fixed_size_list, got {:?}",
                    data_type
                );
            }
        },
        ColumnarValue::Scalar(_) => {
            validate_map_keys(key_array)?;
        }
    }

    let values = get_first_array_ref(values_arg)?;

    make_map_batch_internal(&keys, &values, can_evaluate_to_const, &keys_arg.data_type())
}

fn validate_unique_primitive_keys<T: ArrowPrimitiveType>(array: &dyn Array) -> Result<()>
where