Kotlinx Serialization? about kmongo HOT 4 OPEN

open class MongoDBStorage<T : Any>(clazz: Class<T>, databaseName: String, collectionName: String) : Storage<T> {
    private val client = KMongo.createClient(
        MongoClientSettings.builder()
            .uuidRepresentation(UuidRepresentation.STANDARD)
            .build()
    )


    private val database = client.getDatabase(databaseName)
    private val collection = database.getCollection(collectionName, clazz)


    override fun add(id: Id<T>, entity: T) {
        // Perform the update operation using MongoDB driver
        collection.updateOneById(id, entity, UpdateOptions().upsert(true))
    }


    override fun get(id: Id<T>): T? {
        return collection.findOneById(id)
    }

    override fun getAll(): List<T> {
        return collection.find().toList()
    }

    override fun remove(id: Id<T>) {
        collection.deleteOneById(id)
    }
}

@Serializable
public data class UserEntity(
    @Contextual
    @SerialName("_id")
    val id: Id<UserEntity> = newId(),

    @SerialName("first_name")
    val firstName: String,

    @SerialName("last_name")
    val lastName: String
)

public interface GenericRepository<T> {
    public suspend fun add(id: Id<T>, entity: T)

    public suspend fun get(id: Id<T>): T?
}

internal class MongoGenericRepository<T: Any>(
    private val kMongoDatabase: CoroutineDatabase,
    private val collectionName: String,
    private val clazz: Class<T>
) : GenericRepository<T> {

    private val collection: CoroutineCollection<T> by lazy {
        kMongoDatabase.database.getCollection(collectionName, clazz).coroutine
    }

    override suspend fun get(id: Id<T>): T? {
        return collection.findOneById(id)
    }

    override suspend fun add(id: Id<T>, entity: T) {
        collection.updateOneById(id, entity, UpdateOptions().upsert(true))
    }
}

configuration = configuration.copy(
    encodeDefaults = false,
    classDiscriminator = SchemaRegistry.TYPE_DISCRIMINATOR
)