Function compute_null_mask

datafusion/spark/src/function/null_utils.rs:34–72

(args: &[ColumnarValue])

}

pub(crate) fn compute_null_mask(args: &[ColumnarValue]) -> NullMaskResolution {
    let mut array_len = None;
    let mut has_null_scalar = false;

    for arg in args {
        match arg {
            ColumnarValue::Array(array) => {
                array_len.get_or_insert_with(|| array.len());
            }
            ColumnarValue::Scalar(scalar) => {
                has_null_scalar |= scalar.is_null();
            }
        }
    }

    let Some(array_len) = array_len else {
        // All arguments are scalars
        return if has_null_scalar {
            NullMaskResolution::ReturnNull
        } else {
            NullMaskResolution::NoMask
        };
    };

    if has_null_scalar {
        return NullMaskResolution::Apply(NullBuffer::new_null(array_len));
    }

    let combined_nulls =
        NullBuffer::union_many(args.iter().filter_map(|arg| match arg {
            ColumnarValue::Array(array) => Some(array.nulls()),
            ColumnarValue::Scalar(_) => None,
        }));

    match combined_nulls {
        Some(nulls) => NullMaskResolution::Apply(nulls),
        None => NullMaskResolution::NoMask,
    }
}

/// Apply NULL mask to the result using NullBuffer::union
pub(crate) fn apply_null_mask(