Method invoke_with_args

datafusion/functions-nested/src/array_any_match.rs:183–257

(&self, args: HigherOrderFunctionArgs)

    }

    fn invoke_with_args(&self, args: HigherOrderFunctionArgs) -> Result<ColumnarValue> {
        let [ValueOrLambda::Value(list), ValueOrLambda::Lambda(lambda)] =
            take_function_args(self.name(), &args.args)?
        else {
            return exec_err!("{} expects a value followed by a lambda", self.name());
        };

        let list_array = list.to_array(args.number_rows)?;

        // fast path: fully null input — also required for FixedSizeList which can't be
        // handled by clear_null_values when fully null
        if list_array.null_count() == list_array.len() {
            return Ok(ColumnarValue::Array(new_null_array(
                args.return_type(),
                list_array.len(),
            )));
        }

        let list_values = list_values(&list_array)?;

        let values_param = || Ok(Arc::clone(&list_values));

        let predicate_results = lambda
            .evaluate(&[&values_param], |arrays| {
                let indices = list_values_row_number(&list_array)?;
                Ok(take_arrays(arrays, &indices, None)?)
            })?
            .into_array(list_values.len())?;

        let predicate_bool = predicate_results
            .as_any()
            .downcast_ref::<BooleanArray>()
            .ok_or_else(|| {
                exec_datafusion_err!(
                    "{} predicate must return boolean array",
                    self.name()
                )
            })?;

        let mut values = BooleanBuilder::with_capacity(list_array.len());

        // Maps predicate results (flat over all elements) back to one Boolean per row.
        // Uses adjusted offsets so sliced lists index correctly into the predicate array.
        macro_rules! process_list {
            ($list_typed:expr) => {{
                let offsets = adjust_offsets_for_slice($list_typed);
                for i in 0..$list_typed.len() {
                    let start = offsets[i].as_usize();
                    let end = offsets[i + 1].as_usize();
                    // any_match_for_range returns None when nulls poison the result;
                    // null rows produce an empty range and return Some(false), but their
                    // null bit is preserved by attaching the original null bitmap below.
                    values.append_option(any_match_for_range(predicate_bool, start, end));
                }
            }};
        }

        match list_array.data_type() {