Fixes a test referencing ExprRow

partiql · Nov 27, 2024 · 4035919 · 4035919
1 parent b288d2d
commit 4035919
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/partiql-planner/src/main/kotlin/org/partiql/planner/internal/transforms/RexConverter.kt b/partiql-planner/src/main/kotlin/org/partiql/planner/internal/transforms/RexConverter.kt
@@ -45,6 +45,7 @@ import org.partiql.ast.expr.ExprOverlay
 import org.partiql.ast.expr.ExprPath
 import org.partiql.ast.expr.ExprPosition
 import org.partiql.ast.expr.ExprQuerySet
+import org.partiql.ast.expr.ExprRow
 import org.partiql.ast.expr.ExprSessionAttribute
 import org.partiql.ast.expr.ExprStruct
 import org.partiql.ast.expr.ExprSubstring
@@ -126,6 +127,12 @@ internal object RexConverter {
         override fun defaultReturn(node: AstNode, context: Env): Rex =
             throw IllegalArgumentException("unsupported rex $node")
 
+        override fun visitExprRow(node: ExprRow, ctx: Env): Rex {
+            val values = node.values.map { visitExprCoerce(it, ctx) }
+            val op = rexOpCollection(values)
+            return rex(LIST, op) // TODO: We only do this for legacy reasons. This should return a rexOpRow!
+        }
+
         override fun visitExprLit(node: ExprLit, context: Env): Rex {
             val type = when (val value = node.value) {
                 // Specifically handle the lack of an infinite precision/scale