Function spark_pmod

datafusion/spark/src/function/math/modulus.rs:71–86

Spark-compatible `pmod` function In ANSI mode, division by zero throws an error. In legacy mode, division by zero returns NULL (Spark behavior).

(
    args: &[ColumnarValue],
    enable_ansi_mode: bool,
)

/// In ANSI mode, division by zero throws an error.
/// In legacy mode, division by zero returns NULL (Spark behavior).
pub fn spark_pmod(
    args: &[ColumnarValue],
    enable_ansi_mode: bool,
) -> Result<ColumnarValue> {
    assert_eq_or_internal_err!(args.len(), 2, "pmod expects exactly two arguments");
    let args = ColumnarValue::values_to_arrays(args)?;
    let left = &args[0];
    let right = &args[1];
    let zero = ScalarValue::new_zero(left.data_type())?.to_array_of_size(left.len())?;
    let result = try_rem(left, right, enable_ansi_mode)?;
    let neg = lt(&result, &zero)?;
    let plus = zip(&neg, right, &zero)?;
    let result = add(&plus, &result)?;
    let result = try_rem(&result, right, enable_ansi_mode)?;
    Ok(ColumnarValue::Array(result))
}

/// SparkMod implements the Spark-compatible modulo function
#[derive(Debug, PartialEq, Eq, Hash)]