Fixed check on computeIfAbsent

Author: GoodforGod

cache/cache-symbol-processor/src/main/kotlin/io/koraframework/cache/symbol/processor/CacheSymbolProcessor.kt

@@ -357,7 +357,7 @@ class CacheSymbolProcessor(
             )
 
             val keyName = "_key" + (i + 1)
-            keyBuilder.addStatement("val %L = %L.apply(key.%L!!)!!", keyName, mapperName, recordField.simpleName.asString())
+            keyBuilder.addStatement("val %L = %L.apply(key!!.%L!!)!!", keyName, mapperName, recordField.simpleName.asString())
             if (i == 0) {
                 compositeKeyBuilder.add("val _compositeKey = %T(", ByteArray::class)
                 for (j in recordFields.indices) {

cache/cache-symbol-processor/src/main/kotlin/io/koraframework/cache/symbol/processor/aop/CacheableAopKoraAspect.kt

@@ -49,21 +49,23 @@ class CacheableAopKoraAspect(private val resolver: Resolver) : AbstractAopCacheA
         val superMethod = getSuperMethod(method, superCall)
         val builder = CodeBlock.builder()
 
+        val isResultNullable = method.returnType!!.resolve().isMarkedNullable
+        val suffixCheck = if (isResultNullable) "" else "!!"
         if (operation.executions.size == 1) {
             val keyBlock = CodeBlock.of("val _key = %L\n", operation.executions[0].cacheKey!!.code)
             val isSingleNullableParam = operation.executions[0].type.isMarkedNullable
             val codeBlock = if (isSingleNullableParam) {
                 CodeBlock.of(
                     """
                         return if (_key != null) {
-                            %L.computeIfAbsent(_key) { %L }
+                            %L.computeIfAbsent(_key) { %L }%L
                         } else {
                             %L
                         }
-                    """.trimIndent(), operation.executions[0].field, superMethod, superMethod
+                    """.trimIndent(), operation.executions[0].field, superMethod, suffixCheck, superMethod
                 )
             } else {
-                CodeBlock.of("return %L.computeIfAbsent(_key) { %L }", operation.executions[0].field, superMethod)
+                CodeBlock.of("return %L.computeIfAbsent(_key) { %L }%L", operation.executions[0].field, superMethod, suffixCheck)
             }
 
             return CodeBlock.builder()