mongo adapter: Allow reading of old INT64 Snowflake from databases (#238

) * mongo adapter: Allow reading of old INT64 Snowflake from databases * Update SnowflakeCodec.kt
Kord-Extensions · Sep 16, 2023 · 1b01954 · 1b01954
1 parent 5c370d7
commit 1b01954
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/...c/main/kotlin/com/kotlindiscord/kord/extensions/adapters/mongodb/codecs/SnowflakeCodec.kt b/...c/main/kotlin/com/kotlindiscord/kord/extensions/adapters/mongodb/codecs/SnowflakeCodec.kt
@@ -7,15 +7,19 @@
 package com.kotlindiscord.kord.extensions.adapters.mongodb.codecs
 
 import dev.kord.common.entity.Snowflake
+import org.bson.BsonInvalidOperationException
 import org.bson.BsonReader
 import org.bson.BsonWriter
 import org.bson.codecs.Codec
 import org.bson.codecs.DecoderContext
 import org.bson.codecs.EncoderContext
 
 public object SnowflakeCodec : Codec<Snowflake> {
-    override fun decode(reader: BsonReader, decoderContext: DecoderContext): Snowflake =
-        Snowflake(reader.readString())
+    override fun decode(reader: BsonReader, decoderContext: DecoderContext): Snowflake = try {
+	    Snowflake(reader.readString())
+    } catch (e: BsonInvalidOperationException) {
+	    Snowflake(reader.readInt64())
+    }
 
     override fun encode(writer: BsonWriter, value: Snowflake, encoderContext: EncoderContext) {
         writer.writeString(value.toString())