feat: Support planning ROUND(numeric, int)

cube-js · Jan 26, 2024 · 2035cf4 · 2035cf4
1 parent 333e8de
commit 2035cf4
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/datafusion/core/src/physical_plan/functions.rs b/datafusion/core/src/physical_plan/functions.rs
@@ -672,6 +672,15 @@ fn signature(fun: &BuiltinScalarFunction) -> Signature {
             ],
             fun.volatility(),
         ),
+        BuiltinScalarFunction::Round => Signature::one_of(
+            vec![
+                TypeSignature::Exact(vec![DataType::Float64]),
+                TypeSignature::Exact(vec![DataType::Float32]),
+                // NOTE: stub, won't execute
+                TypeSignature::Exact(vec![DataType::Decimal(38, 10), DataType::Int32]),
+            ],
+            fun.volatility(),
+        ),
         // math expressions expect 1 argument of type f64 or f32
         // priority is given to f64 because e.g. `sqrt(1i32)` is in IR (real numbers) and thus we
         // return the best approximation for it (in f64).