feat(flink): implement translation of ZEROIFNULL

ibis-project · Sep 21, 2023 · 31560eb · 31560eb
1 parent 19514bc
commit 31560eb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ibis/backends/flink/registry.py b/ibis/backends/flink/registry.py
@@ -67,6 +67,11 @@ def _literal(translator: ExprTranslator, op: ops.Literal) -> str:
     return translate_literal(op)
 
 
+def _zeroifnull(translator: ExprTranslator, op: ops.Literal) -> str:
+    casted = translate_literal(ops.Literal("0", dtype=op.dtype))
+    return f"COALESCE({translator.translate(op.arg)}, {casted})"
+
+
 def _nullifzero(translator: ExprTranslator, op: ops.Literal) -> str:
     casted = translate_literal(ops.Literal("0", dtype=op.dtype))
     return f"NULLIF({translator.translate(op.arg)}, {casted})"
@@ -206,6 +211,7 @@ def _floor_divide(translator: ExprTranslator, op: ops.Node) -> str:
     {
         # Unary operations
         ops.IfNull: fixed_arity("ifnull", 2),
+        ops.ZeroIfNull: _zeroifnull,
         ops.NullIfZero: _nullifzero,
         ops.RandomScalar: lambda *_: "rand()",
         ops.Degrees: unary("degrees"),