Fix non-nullable type w/ ? warnings

partiql · Apr 18, 2024 · a696132 · a696132
1 parent fa0f3dc
commit a696132
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/pig/src/main/kotlin/org/partiql/pig/generator/kotlin/KTypeDomainConverter.kt b/pig/src/main/kotlin/org/partiql/pig/generator/kotlin/KTypeDomainConverter.kt
@@ -241,7 +241,16 @@ private class KTypeDomainConverter(
                 val elementIsKotlinPrimitive = isKotlinPrimitive(element)
                 val elementKotlinName = element.identifier.snakeToCamelCase()
                 when (element.typeReference.arity) {
-                    is Arity.Required, Arity.Optional -> {
+                    is Arity.Required -> {
+                        KConstructorArgument(
+                            kotlinName = elementKotlinName,
+                            value = elementKotlinName + when {
+                                elementIsKotlinPrimitive && useKotlinPrimitives -> ".asPrimitive()"
+                                else -> ""
+                            }
+                        )
+                    }
+                    is Arity.Optional -> {
                         KConstructorArgument(
                             kotlinName = elementKotlinName,
                             value = elementKotlinName + when {

diff --git a/pig/src/main/resources/org/partiql/pig/templates/kotlin-domain.ftl b/pig/src/main/resources/org/partiql/pig/templates/kotlin-domain.ftl
@@ -58,7 +58,11 @@ class ${t.kotlinName}(
             [#if p.variadic]
             if(${p.kotlinName}.any()) { ionSexpOf(ionSymbol("${p.tag}"), *${p.kotlinName}.map { it.toIonElement() }.toTypedArray()) } else { null }[#sep],[/#sep]
             [#else]
+            [#if p.nullable]
             ${p.kotlinName}?.let { ionSexpOf(ionSymbol("${p.tag}"), it.toIonElement()) }[#sep],[/#sep]
+            [#else]
+            ionSexpOf(ionSymbol("${p.tag}"), ${p.kotlinName}.toIonElement())[#sep],[/#sep]
+            [/#if]
             [/#if]
         [/#list]
         )