GH-80 Support NULL_VALUE const val to represent raw NULL values (Resolve

#80)
dzikoysk · Sep 1, 2024 · 439244b · 439244b
1 parent 255b7d8
commit 439244b
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 11 deletions.
diff --git a/...bol-processor/src/main/kotlin/com/dzikoysk/sqiffy/processor/generators/EntityGenerator.kt b/...bol-processor/src/main/kotlin/com/dzikoysk/sqiffy/processor/generators/EntityGenerator.kt
@@ -1,6 +1,7 @@
 package com.dzikoysk.sqiffy.processor.generators
 
 import com.dzikoysk.sqiffy.definition.DataType
+import com.dzikoysk.sqiffy.definition.NULL_VALUE
 import com.dzikoysk.sqiffy.definition.ParsedDefinition
 import com.dzikoysk.sqiffy.definition.PropertyData
 import com.dzikoysk.sqiffy.dsl.Row
@@ -145,8 +146,12 @@ class EntityGenerator(private val context: KspContext) {
                     .build()
             )
 
-    private fun String.toKotlinCode(dataType: DataType, typeName: TypeName): String =
-        when (dataType) {
+    private fun String.toKotlinCode(dataType: DataType, typeName: TypeName): String {
+        if (this == NULL_VALUE) {
+            return "null"
+        }
+
+        return when (dataType) {
             /* Special types */
             DataType.UUID_TYPE -> "UUID.fromString(\"$this\")"
             DataType.ENUM -> "$typeName.$this"
@@ -165,5 +170,5 @@ class EntityGenerator(private val context: KspContext) {
             DataType.TIMESTAMP -> "Instant.parse(\"$this\")"
             else -> this
         }
-
+    }
 }
diff --git a/...main/kotlin/com/dzikoysk/sqiffy/changelog/generator/dialects/GenericSqlSchemeGenerator.kt b/...main/kotlin/com/dzikoysk/sqiffy/changelog/generator/dialects/GenericSqlSchemeGenerator.kt
@@ -17,6 +17,7 @@ import com.dzikoysk.sqiffy.definition.DataType.SERIAL
 import com.dzikoysk.sqiffy.definition.DataType.TEXT
 import com.dzikoysk.sqiffy.definition.DataType.UUID_TYPE
 import com.dzikoysk.sqiffy.definition.DataType.VARCHAR
+import com.dzikoysk.sqiffy.definition.NULL_VALUE
 import com.dzikoysk.sqiffy.definition.PropertyData
 
 abstract class GenericSqlSchemeGenerator : SqlSchemeGenerator {
@@ -98,15 +99,19 @@ abstract class GenericSqlSchemeGenerator : SqlSchemeGenerator {
             .let { dataType -> "$dataType ${property.default?.let { default -> "DEFAULT ${default.toSqlDefault(property)} " } ?: ""}" }
             .let { if (!property.nullable) "$it NOT NULL" else it }
 
-    private fun String.toSqlDefault(property: PropertyData): String =
-        when (property.rawDefault) {
-            true -> property.default!!
-            else ->
-                createSqlDefault(this, property)
-                    ?: createRegularDefault(this, property)
-                    ?: throw UnsupportedOperationException("Cannot create default value based on $this")
+    private fun String.toSqlDefault(property: PropertyData): String {
+        if (property.rawDefault) {
+            return property.default!!
         }
 
+        if (property.default == NULL_VALUE) {
+            return "NULL"
+        }
+
+        return createSqlDefault(this, property)
+            ?: createRegularDefault(this, property)
+            ?: throw UnsupportedOperationException("Cannot create default value based on $this")
+    }
 
     abstract fun createSqlDefault(rawDefault: String, property: PropertyData, dataType: DataType = property.type!!): String?
 

diff --git a/sqiffy/src/main/kotlin/com/dzikoysk/sqiffy/definition/DefinitionAnnotations.kt b/sqiffy/src/main/kotlin/com/dzikoysk/sqiffy/definition/DefinitionAnnotations.kt
@@ -10,8 +10,9 @@ import java.util.UUID
 import kotlin.annotation.AnnotationRetention.RUNTIME
 import kotlin.reflect.KClass
 
-const val NULL_STRING = "~NULL-STRING~"
 object NULL_CLASS
+const val NULL_STRING = "~NULL-STRING~"
+const val NULL_VALUE = "~NULL-VALUE~"
 
 enum class Kind {
     DIRECT,