make AvroBinaryInputStream backward compatible

sksamuel · Jan 23, 2024 · 4936b35 · 4936b35
1 parent 77e44c3
commit 4936b35
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/avro4s-core/src/main/scala/com/sksamuel/avro4s/AvroBinaryInputStream.scala b/avro4s-core/src/main/scala/com/sksamuel/avro4s/AvroBinaryInputStream.scala
@@ -23,12 +23,15 @@ import scala.util.Try
   * @see https://avro.apache.org/docs/current/spec.html#binary_encoding
   */
 class AvroBinaryInputStream[T](in: InputStream,
-                               writerSchema: Schema)
+                               writerSchema: Schema,
+                               readerSchema: Schema)
                               (using decoder: Decoder[T]) extends AvroInputStream[T] {
 
-  private val datumReader = GenericData.get.createDatumReader(writerSchema).asInstanceOf[DatumReader[Any]]
+  def this(in: InputStream, writerSchema: Schema)(using decoder: Decoder[T]) = this(in, writerSchema, writerSchema)
+
+  private val datumReader = new GenericDatumReader[Any](writerSchema, readerSchema, GenericData.get)
   private val avroDecoder = DecoderFactory.get().binaryDecoder(in, null)
-  private val decodeT = decoder.decode(writerSchema)
+  private val decodeT = decoder.decode(readerSchema)
 
   private val _iter = new Iterator[Any] {
     override def hasNext: Boolean = !avroDecoder.isEnd