From 9df3088c23a8ef3d4ce6798ecf8698ec9d272d78 Mon Sep 17 00:00:00 2001
From: wenchieh <alwjlola@gmail.com>
Date: Thu, 10 Mar 2022 22:12:36 +0800
Subject: [PATCH] Fix Long and Float Type error

---
 app/src/main/java/com/lwjlol/ormkv/demo/UserModel.kt      | 2 +-
 .../lwjlol/ormkv/annotation/compiler/KspOrmkvProvider.kt  | 8 +++++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/app/src/main/java/com/lwjlol/ormkv/demo/UserModel.kt b/app/src/main/java/com/lwjlol/ormkv/demo/UserModel.kt
index 0899ce9..6dbc547 100644
--- a/app/src/main/java/com/lwjlol/ormkv/demo/UserModel.kt
+++ b/app/src/main/java/com/lwjlol/ormkv/demo/UserModel.kt
@@ -23,7 +23,7 @@ data class UserModel(
     val id: Long,
     @ColumnInfo(defValue = "false")
     val isMan: Boolean,
-    @ColumnInfo(defValue = "0F", enableReset = false)
+    @ColumnInfo(defValue = "1231", enableReset = false)
     val temperature: Float
 
 ) {
diff --git a/compiler/src/main/java/com/lwjlol/ormkv/annotation/compiler/KspOrmkvProvider.kt b/compiler/src/main/java/com/lwjlol/ormkv/annotation/compiler/KspOrmkvProvider.kt
index f8a6bea..83f3be5 100644
--- a/compiler/src/main/java/com/lwjlol/ormkv/annotation/compiler/KspOrmkvProvider.kt
+++ b/compiler/src/main/java/com/lwjlol/ormkv/annotation/compiler/KspOrmkvProvider.kt
@@ -137,6 +137,9 @@ class KspOrmkvProcessor(
                 }
                 "kotlin.Float" -> {
                     defValue = defValue.ifEmpty { "0F" }
+                    if (!defValue.endsWith('F')) {
+                        defValue = "${defValue}F"
+                    }
                     getName = "get(\"$keyName\", $defValue) as Float"
                 }
                 "kotlin.Int" -> {
@@ -145,6 +148,9 @@ class KspOrmkvProcessor(
                 }
                 "kotlin.Long" -> {
                     defValue = defValue.ifEmpty { "0L" }
+                    if (!defValue.endsWith('L')) {
+                        defValue = "${defValue}L"
+                    }
                     getName = "get(\"$keyName\", $defValue) as Long"
                 }
                 "kotlin.Boolean" -> {
@@ -334,6 +340,6 @@ class KspOrmkvProvider : SymbolProcessorProvider {
 }
 
 private const val END_FIX = "Registry"
-private const val LOG = true
+private const val LOG = false
 private const val HANDLER = "kvHandler"
 private const val TAG = "KspOrmkv"
\ No newline at end of file