Function spark_mod

datafusion/spark/src/function/math/modulus.rs:58–66

Spark-compatible `mod` 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_mod(
    args: &[ColumnarValue],
    enable_ansi_mode: bool,
) -> Result<ColumnarValue> {
    assert_eq_or_internal_err!(args.len(), 2, "mod expects exactly two arguments");
    let args = ColumnarValue::values_to_arrays(args)?;
    let result = try_rem(&args[0], &args[1], enable_ansi_mode)?;
    Ok(ColumnarValue::Array(result))
}

/// Spark-compatible `pmod` function
/// In ANSI mode, division by zero throws an error.