Function validate_map_keys

datafusion/functions-nested/src/map.rs:190–228

Validates that map keys are non-null and unique.

(array: &dyn Array)

/// Validates that map keys are non-null and unique.
fn validate_map_keys(array: &dyn Array) -> Result<()> {
    match array.data_type() {
        DataType::Int8 => validate_unique_primitive_keys::<Int8Type>(array),
        DataType::Int16 => validate_unique_primitive_keys::<Int16Type>(array),
        DataType::Int32 => validate_unique_primitive_keys::<Int32Type>(array),
        DataType::Int64 => validate_unique_primitive_keys::<Int64Type>(array),
        DataType::UInt8 => validate_unique_primitive_keys::<UInt8Type>(array),
        DataType::UInt16 => validate_unique_primitive_keys::<UInt16Type>(array),
        DataType::UInt32 => validate_unique_primitive_keys::<UInt32Type>(array),
        DataType::UInt64 => validate_unique_primitive_keys::<UInt64Type>(array),
        DataType::Date32 => validate_unique_primitive_keys::<Date32Type>(array),
        DataType::Date64 => validate_unique_primitive_keys::<Date64Type>(array),
        DataType::Utf8 => {
            let arr = array.as_string::<i32>();
            validate_unique_str_keys(arr.null_count(), arr.len(), arr.iter().flatten())
        }
        DataType::LargeUtf8 => {
            let arr = array.as_string::<i64>();
            validate_unique_str_keys(arr.null_count(), arr.len(), arr.iter().flatten())
        }
        DataType::Utf8View => {
            let arr = array.as_string_view();
            validate_unique_str_keys(arr.null_count(), arr.len(), arr.iter().flatten())
        }
        DataType::Binary => {
            let arr = array.as_binary::<i32>();
            validate_unique_binary_keys(arr.null_count(), arr.len(), arr.iter().flatten())
        }
        DataType::LargeBinary => {
            let arr = array.as_binary::<i64>();
            validate_unique_binary_keys(arr.null_count(), arr.len(), arr.iter().flatten())
        }
        DataType::BinaryView => {
            let arr = array.as_binary_view();
            validate_unique_binary_keys(arr.null_count(), arr.len(), arr.iter().flatten())
        }
        _ => validate_unique_keys_generic(array),
    }
}

fn get_first_array_ref(columnar_value: &ColumnarValue) -> Result<ArrayRef> {
    match columnar_value {