Function string_to_array_fallback

datafusion/functions-nested/src/string.rs:419–444

Fallback path for `string_to_array` when delimiter and/or null_value are array columns rather than scalars.

(args: &[ArrayRef])

/// Fallback path for `string_to_array` when delimiter and/or null_value
/// are array columns rather than scalars.
fn string_to_array_fallback(args: &[ArrayRef]) -> Result<ArrayRef> {
    let null_value_array = args.get(2);

    match args[0].data_type() {
        Utf8 => {
            let arr = args[0].as_string::<i32>();
            let builder =
                StringBuilder::with_capacity(arr.len(), arr.get_buffer_memory_size());
            string_to_array_column_args(&arr, &args[1], null_value_array, builder)
        }
        Utf8View => {
            let arr = args[0].as_string_view();
            let builder = StringViewBuilder::with_capacity(arr.len());
            string_to_array_column_args(&arr, &args[1], null_value_array, builder)
        }
        LargeUtf8 => {
            let arr = args[0].as_string::<i64>();
            let builder = LargeStringBuilder::with_capacity(
                arr.len(),
                arr.get_buffer_memory_size(),
            );
            string_to_array_column_args(&arr, &args[1], null_value_array, builder)
        }
        other => exec_err!("unsupported type for string_to_array function as {other:?}"),
    }
}

fn string_to_array_column_args<'a, StringArrType, StringBuilderType>(
    string_array: &StringArrType,