Added useStrictPrimitiveDecoders

sksamuel · Apr 28, 2024 · 25c7974 · 25c7974
1 parent 5321e2f
commit 25c7974
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 5 deletions.
diff --git a/centurion-avro/src/main/kotlin/com/sksamuel/centurion/avro/decoders/Decoder.kt b/centurion-avro/src/main/kotlin/com/sksamuel/centurion/avro/decoders/Decoder.kt
@@ -27,14 +27,17 @@ fun interface Decoder<T> {
    }
 
    companion object {
+
+      var useStrictPrimitiveDecoders = true
+
       fun decoderFor(type: KType): Decoder<*> {
          val decoder: Decoder<*> = when (val classifier = type.classifier) {
             String::class -> StringDecoder
-            Boolean::class -> BooleanDecoder
-            Float::class -> FloatDecoder
-            Double::class -> DoubleDecoder
-            Int::class -> IntDecoder
-            Long::class -> LongDecoder
+            Boolean::class -> if (useStrictPrimitiveDecoders) StrictBooleanDecoder else BooleanDecoder
+            Float::class -> if (useStrictPrimitiveDecoders) StrictFloatDecoder else FloatDecoder
+            Double::class -> if (useStrictPrimitiveDecoders) StrictDoubleDecoder else DoubleDecoder
+            Int::class -> if (useStrictPrimitiveDecoders) StrictIntDecoder else IntDecoder
+            Long::class -> if (useStrictPrimitiveDecoders) StrictLongDecoder else LongDecoder
             List::class -> ListDecoder(decoderFor(type.arguments.first().type!!))
             Set::class -> SetDecoder(decoderFor(type.arguments.first().type!!))
             Map::class -> MapDecoder(decoderFor(type.arguments[1].type!!))

diff --git a/centurion-avro/src/main/kotlin/com/sksamuel/centurion/avro/decoders/primitives.kt b/centurion-avro/src/main/kotlin/com/sksamuel/centurion/avro/decoders/primitives.kt
@@ -41,6 +41,10 @@ object IntDecoder : Decoder<Int> {
    }
 }
 
+object StrictLongDecoder : Decoder<Long> {
+   override fun decode(schema: Schema): (Any?) -> Long = { it as Long }
+}
+
 object LongDecoder : Decoder<Long> {
    override fun decode(schema: Schema): (Any?) -> Long {
       require(schema.type == Schema.Type.LONG)
@@ -56,6 +60,10 @@ object LongDecoder : Decoder<Long> {
    }
 }
 
+object StrictIntDecoder : Decoder<Int> {
+   override fun decode(schema: Schema): (Any?) -> Int = { it as Int }
+}
+
 object DoubleDecoder : Decoder<Double> {
    override fun decode(schema: Schema): (Any?) -> Double {
       require(schema.type == Schema.Type.DOUBLE)
@@ -69,6 +77,10 @@ object DoubleDecoder : Decoder<Double> {
    }
 }
 
+object StrictDoubleDecoder : Decoder<Double> {
+   override fun decode(schema: Schema): (Any?) -> Double = { it as Double }
+}
+
 object FloatDecoder : Decoder<Float> {
    override fun decode(schema: Schema): (Any?) -> Float {
       require(schema.type == Schema.Type.FLOAT)
@@ -81,6 +93,10 @@ object FloatDecoder : Decoder<Float> {
    }
 }
 
+object StrictFloatDecoder : Decoder<Float> {
+   override fun decode(schema: Schema): (Any?) -> Float = { it as Float }
+}
+
 object BooleanDecoder : Decoder<Boolean> {
    override fun decode(schema: Schema): (Any?) -> Boolean {
       require(schema.type == Schema.Type.BOOLEAN)
@@ -92,3 +108,7 @@ object BooleanDecoder : Decoder<Boolean> {
       }
    }
 }
+
+object StrictBooleanDecoder : Decoder<Boolean> {
+   override fun decode(schema: Schema): (Any?) -> Boolean = { it as Boolean }
+}