Fix getCodec of matching type argument bug (#1339)

bson-kotlin optional type fix Port bson-kotlin test to bson-kotlinx (#1452) JAVA-5365
mongodb · Aug 21, 2024 · 4fb1305 · 4fb1305
1 parent 4577f95
commit 4fb1305
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 11 deletions.
diff --git a/bson-kotlin/src/main/kotlin/org/bson/codecs/kotlin/DataClassCodec.kt b/bson-kotlin/src/main/kotlin/org/bson/codecs/kotlin/DataClassCodec.kt
@@ -18,10 +18,10 @@ package org.bson.codecs.kotlin
 import java.lang.reflect.ParameterizedType
 import java.lang.reflect.Type
 import kotlin.reflect.KClass
+import kotlin.reflect.KClassifier
 import kotlin.reflect.KFunction
 import kotlin.reflect.KParameter
 import kotlin.reflect.KProperty1
-import kotlin.reflect.KType
 import kotlin.reflect.KTypeParameter
 import kotlin.reflect.KTypeProjection
 import kotlin.reflect.full.createType
@@ -142,7 +142,9 @@ internal data class DataClassCodec<T : Any>(
                 val primaryConstructor =
                     kClass.primaryConstructor ?: throw CodecConfigurationException("No primary constructor for $kClass")
                 val typeMap =
-                    types.mapIndexed { i, k -> primaryConstructor.typeParameters[i].createType() to k }.toMap()
+                    types
+                        .mapIndexed { i, k -> primaryConstructor.typeParameters[i].createType().classifier!! to k }
+                        .toMap()
 
                 val propertyModels =
                     primaryConstructor.parameters.map { kParameter ->
@@ -193,18 +195,20 @@ internal data class DataClassCodec<T : Any>(
         @Suppress("UNCHECKED_CAST")
         private fun getCodec(
             kParameter: KParameter,
-            typeMap: Map<KType, Type>,
+            typeMap: Map<KClassifier, Type>,
             codecRegistry: CodecRegistry
         ): Codec<Any> {
             return when (kParameter.type.classifier) {
                 is KClass<*> -> {
                     codecRegistry.getCodec(
                         kParameter,
                         (kParameter.type.classifier as KClass<Any>).javaObjectType,
-                        kParameter.type.arguments.mapNotNull { typeMap[it.type] ?: computeJavaType(it) }.toList())
+                        kParameter.type.arguments
+                            .mapNotNull { typeMap[it.type?.classifier] ?: computeJavaType(it) }
+                            .toList())
                 }
                 is KTypeParameter -> {
-                    when (val pType = typeMap[kParameter.type] ?: kParameter.type.javaType) {
+                    when (val pType = typeMap[kParameter.type.classifier] ?: kParameter.type.javaType) {
                         is Class<*> ->
                             codecRegistry.getCodec(kParameter, (pType as Class<Any>).kotlin.java, emptyList())
                         is ParameterizedType ->

diff --git a/bson-kotlin/src/test/kotlin/org/bson/codecs/kotlin/DataClassCodecTest.kt b/bson-kotlin/src/test/kotlin/org/bson/codecs/kotlin/DataClassCodecTest.kt
@@ -320,15 +320,17 @@ class DataClassCodecTest {
             |"nestedParameterized": {
             |  "parameterizedDataClass":
             |  {"number": 4.2, "string": "myString", "parameterizedList": [{"name": "embedded1"}]},
-            |  "other": "myOtherString"
+            |  "other": "myOtherString", "optionalOther": "myOptionalOtherString"
             | }
             |}"""
                 .trimMargin()
         val dataClass =
             DataClassWithNestedParameterizedDataClass(
                 "myId",
                 DataClassWithNestedParameterized(
-                    DataClassParameterized(4.2, "myString", listOf(DataClassEmbedded("embedded1"))), "myOtherString"))
+                    DataClassParameterized(4.2, "myString", listOf(DataClassEmbedded("embedded1"))),
+                    "myOtherString",
+                    "myOptionalOtherString"))
 
         assertRoundTrips(expected, dataClass)
     }

diff --git a/bson-kotlin/src/test/kotlin/org/bson/codecs/kotlin/samples/DataClasses.kt b/bson-kotlin/src/test/kotlin/org/bson/codecs/kotlin/samples/DataClasses.kt
@@ -178,7 +178,8 @@ data class DataClassWithNestedParameterizedDataClass(
 
 data class DataClassWithNestedParameterized<A, B, C : Number>(
     val parameterizedDataClass: DataClassParameterized<C, A>,
-    val other: B
+    val other: B,
+    val optionalOther: B?
 )
 
 data class DataClassWithPair(val pair: Pair<String, Int>)

diff --git a/bson-kotlinx/src/test/kotlin/org/bson/codecs/kotlinx/KotlinSerializerCodecTest.kt b/bson-kotlinx/src/test/kotlin/org/bson/codecs/kotlinx/KotlinSerializerCodecTest.kt
@@ -396,15 +396,17 @@ class KotlinSerializerCodecTest {
             |"nestedParameterized": {
             |  "parameterizedDataClass":
             |  {"number": 4.2, "string": "myString", "parameterizedList": [{"name": "embedded1"}]},
-            |  "other": "myOtherString"
+            |  "other": "myOtherString", "optionalOther": "myOptionalOtherString"
             | }
             |}"""
                 .trimMargin()
         val dataClass =
             DataClassWithNestedParameterizedDataClass(
                 "myId",
                 DataClassWithNestedParameterized(
-                    DataClassParameterized(4.2, "myString", listOf(DataClassEmbedded("embedded1"))), "myOtherString"))
+                    DataClassParameterized(4.2, "myString", listOf(DataClassEmbedded("embedded1"))),
+                    "myOtherString",
+                    "myOptionalOtherString"))
 
         assertRoundTrips(expected, dataClass)
     }

diff --git a/bson-kotlinx/src/test/kotlin/org/bson/codecs/kotlinx/samples/DataClasses.kt b/bson-kotlinx/src/test/kotlin/org/bson/codecs/kotlinx/samples/DataClasses.kt
@@ -125,7 +125,8 @@ data class DataClassWithNestedParameterizedDataClass(
 @Serializable
 data class DataClassWithNestedParameterized<A, B, C : Number>(
     val parameterizedDataClass: DataClassParameterized<C, A>,
-    val other: B
+    val other: B,
+    val optionalOther: B?
 )
 
 @Serializable data class DataClassWithPair(val pair: Pair<String, Int>)