Function general_position_dispatch

datafusion/functions-nested/src/position.rs:302–325

(args: &[ArrayRef])

}

fn general_position_dispatch<O: OffsetSizeTrait>(args: &[ArrayRef]) -> Result<ArrayRef> {
    let haystack = as_generic_list_array::<O>(&args[0])?;
    let needle = &args[1];

    crate::utils::check_datatypes("array_position", &[haystack.values(), needle])?;

    let arr_from = if args.len() == 3 {
        as_int64_array(&args[2])?
            .values()
            .iter()
            .map(|&x| x - 1)
            .collect::<Vec<_>>()
    } else {
        vec![0; haystack.len()]
    };

    for (row, &from) in haystack.iter().zip(arr_from.iter()) {
        if !row.is_none_or(|row| from >= 0 && (from as usize) <= row.len()) {
            return exec_err!("start_from out of bounds: {}", from + 1);
        }
    }

    generic_position::<O>(haystack, needle, &arr_from)
}

fn generic_position<O: OffsetSizeTrait>(
    haystack: &GenericListArray<O>,