Support ANY type in Header functions And Planning of Control Flow Fun…

…ctions. (#1256) * coalesce; nullif; ANY type in function signature.
partiql · Nov 2, 2023 · 5cdc8bb · 5cdc8bb
1 parent a365a45
commit 5cdc8bb
Show file tree

Hide file tree

Showing 6 changed files with 268 additions and 69 deletions.
diff --git a/partiql-lang/build.gradle.kts b/partiql-lang/build.gradle.kts
@@ -85,5 +85,6 @@ tasks.processTestResources {
 
 tasks.shadowJar {
     archiveBaseName.set("shadow")
+    exclude("**/*.kotlin_metadata")
     archiveClassifier.set("")
 }
diff --git a/partiql-planner/src/main/kotlin/org/partiql/planner/Header.kt b/partiql-planner/src/main/kotlin/org/partiql/planner/Header.kt
@@ -43,7 +43,7 @@ public abstract class Header {
      * For functions, output CREATE FUNCTION statements.
      */
     override fun toString(): String = buildString {
-        functions.groupBy { it.name }.forEach {
+        (functions + operators + aggregations).groupBy { it.name }.forEach {
             appendLine("-- [${it.key}] ---------")
             appendLine()
             it.value.forEach { fn -> appendLine(fn) }
@@ -58,23 +58,25 @@ public abstract class Header {
     companion object {
 
         @JvmStatic
-        internal fun unary(name: String, returns: PartiQLValueType, value: PartiQLValueType) =
+        internal fun unary(name: String, returns: PartiQLValueType, value: PartiQLValueType, isMissable: Boolean = false) =
             FunctionSignature.Scalar(
                 name = name,
                 returns = returns,
                 parameters = listOf(FunctionParameter("value", value)),
                 isNullCall = true,
                 isNullable = false,
+                isMissable = isMissable
             )
 
         @JvmStatic
-        internal fun binary(name: String, returns: PartiQLValueType, lhs: PartiQLValueType, rhs: PartiQLValueType) =
+        internal fun binary(name: String, returns: PartiQLValueType, lhs: PartiQLValueType, rhs: PartiQLValueType, isMissable: Boolean = false) =
             FunctionSignature.Scalar(
                 name = name,
                 returns = returns,
                 parameters = listOf(FunctionParameter("lhs", lhs), FunctionParameter("rhs", rhs)),
                 isNullCall = true,
                 isNullable = false,
+                isMissable = isMissable
             )
     }
 }