Function create_args_n

datafusion/functions-nested/benches/array_remove.rs:427–451

(
    haystack: ArrayRef,
    needle: ScalarValue,
    n: ScalarValue,
)

#[inline]
fn create_args_n(
    haystack: ArrayRef,
    needle: ScalarValue,
    n: ScalarValue,
) -> ScalarFunctionArgs {
    let number_rows = haystack.len();
    let haystack_type = haystack.data_type().clone();
    let needle_type = needle.data_type().clone();
    let n_type = n.data_type().clone();
    ScalarFunctionArgs {
        args: vec![
            ColumnarValue::Array(haystack),
            ColumnarValue::Scalar(needle),
            ColumnarValue::Scalar(n),
        ],
        arg_fields: vec![
            Field::new("haystack", haystack_type.clone(), true).into(),
            Field::new("needle", needle_type, true).into(),
            Field::new("n", n_type, true).into(),
        ],
        number_rows,
        return_field: Field::new("result", haystack_type, true).into(),
        config_options: Arc::new(ConfigOptions::default()),
    }
}

fn create_list_array<Builder, Item>(
    num_rows: usize,