Function map_keys_inner

datafusion/functions-nested/src/map_keys.rs:108–123

(args: &[ArrayRef])

}

fn map_keys_inner(args: &[ArrayRef]) -> Result<ArrayRef> {
    let [map_arg] = take_function_args("map_keys", args)?;

    let map_array = match map_arg.data_type() {
        DataType::Map(_, _) => as_map_array(&map_arg)?,
        _ => return exec_err!("Argument for map_keys should be a map"),
    };

    Ok(Arc::new(ListArray::new(
        // internal array nullability is true to be in sync with DuckDB
        Arc::new(Field::new_list_field(map_array.key_type().clone(), true)),
        map_array.offsets().clone(),
        Arc::clone(map_array.keys()),
        map_array.nulls().cloned(),
    )))
}