Simplify customization of bson-kotlinx (#1293)

Allow configuration to be set on the KotlinSerializerCodecProvider JAVA-5276
mongodb · Jan 24, 2024 · baebf90 · baebf90
1 parent 0d8cc80
commit baebf90
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 2 deletions.
diff --git a/bson-kotlinx/src/main/kotlin/org/bson/codecs/kotlinx/KotlinSerializerCodecProvider.kt b/bson-kotlinx/src/main/kotlin/org/bson/codecs/kotlinx/KotlinSerializerCodecProvider.kt
@@ -15,6 +15,8 @@
  */
 package org.bson.codecs.kotlinx
 
+import kotlinx.serialization.ExperimentalSerializationApi
+import kotlinx.serialization.modules.SerializersModule
 import org.bson.codecs.Codec
 import org.bson.codecs.configuration.CodecProvider
 import org.bson.codecs.configuration.CodecRegistry
@@ -24,8 +26,12 @@ import org.bson.codecs.configuration.CodecRegistry
  *
  * The underlying class must be annotated with the `@Serializable`.
  */
-public class KotlinSerializerCodecProvider : CodecProvider {
+@OptIn(ExperimentalSerializationApi::class)
+public class KotlinSerializerCodecProvider(
+    private val serializersModule: SerializersModule = defaultSerializersModule,
+    private val bsonConfiguration: BsonConfiguration = BsonConfiguration()
+) : CodecProvider {
 
     override fun <T : Any> get(clazz: Class<T>, registry: CodecRegistry): Codec<T>? =
-        KotlinSerializerCodec.create(clazz.kotlin)
+        KotlinSerializerCodec.create(clazz.kotlin, serializersModule, bsonConfiguration)
 }
diff --git a/bson-kotlinx/src/test/kotlin/org/bson/codecs/kotlinx/KotlinSerializerCodecProviderTest.kt b/bson-kotlinx/src/test/kotlin/org/bson/codecs/kotlinx/KotlinSerializerCodecProviderTest.kt
@@ -20,6 +20,20 @@ import kotlin.test.assertEquals
 import kotlin.test.assertNotNull
 import kotlin.test.assertNull
 import kotlin.test.assertTrue
+import kotlinx.serialization.ExperimentalSerializationApi
+import kotlinx.serialization.modules.SerializersModule
+import kotlinx.serialization.modules.plus
+import kotlinx.serialization.modules.polymorphic
+import kotlinx.serialization.modules.subclass
+import org.bson.BsonDocument
+import org.bson.BsonDocumentReader
+import org.bson.BsonDocumentWriter
+import org.bson.codecs.DecoderContext
+import org.bson.codecs.EncoderContext
+import org.bson.codecs.kotlinx.samples.DataClassContainsOpen
+import org.bson.codecs.kotlinx.samples.DataClassOpen
+import org.bson.codecs.kotlinx.samples.DataClassOpenA
+import org.bson.codecs.kotlinx.samples.DataClassOpenB
 import org.bson.codecs.kotlinx.samples.DataClassParameterized
 import org.bson.codecs.kotlinx.samples.DataClassWithSimpleValues
 import org.bson.conversions.Bson
@@ -60,4 +74,37 @@ class KotlinSerializerCodecProviderTest {
         assertTrue { codec is KotlinSerializerCodec }
         assertEquals(DataClassWithSimpleValues::class.java, codec.encoderClass)
     }
+
+    @OptIn(ExperimentalSerializationApi::class)
+    @Test
+    fun shouldAllowOverridingOfSerializersModuleAndBsonConfigurationInConstructor() {
+
+        val serializersModule =
+            SerializersModule {
+                this.polymorphic(DataClassOpen::class) {
+                    this.subclass(DataClassOpenA::class)
+                    this.subclass(DataClassOpenB::class)
+                }
+            } + defaultSerializersModule
+
+        val bsonConfiguration = BsonConfiguration(classDiscriminator = "__type")
+        val dataClassContainsOpenB = DataClassContainsOpen(DataClassOpenB(1))
+
+        val codec =
+            KotlinSerializerCodecProvider(serializersModule, bsonConfiguration)
+                .get(DataClassContainsOpen::class.java, Bson.DEFAULT_CODEC_REGISTRY)!!
+
+        assertTrue { codec is KotlinSerializerCodec }
+        val encodedDocument = BsonDocument()
+        val writer = BsonDocumentWriter(encodedDocument)
+        codec.encode(writer, dataClassContainsOpenB, EncoderContext.builder().build())
+        writer.flush()
+
+        assertEquals(
+            BsonDocument.parse("""{"open": {"__type": "org.bson.codecs.kotlinx.samples.DataClassOpenB", "b": 1}}"""),
+            encodedDocument)
+
+        assertEquals(
+            dataClassContainsOpenB, codec.decode(BsonDocumentReader(encodedDocument), DecoderContext.builder().build()))
+    }
 }