arrays can be null and should be unboxed when deserializing items

jorgecarleitao · Jan 3, 2022 · 4d894d0 · 4d894d0
1 parent ef7937d
commit 4d894d0
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/io/avro/read/deserialize.rs b/src/io/avro/read/deserialize.rs
@@ -96,10 +96,13 @@ fn deserialize_value<'a>(
     let data_type = array.data_type();
     match data_type {
         DataType::List(inner) => {
-            let avro_inner = if let AvroSchema::Array(inner) = avro_field {
-                inner.as_ref()
-            } else {
-                unreachable!()
+            let avro_inner = match avro_field {
+                AvroSchema::Array(inner) => inner.as_ref(),
+                AvroSchema::Union(u) => match &u.as_slice() {
+                    &[AvroSchema::Array(inner), _] | &[_, AvroSchema::Array(inner)] => inner.as_ref(),
+                    _ => unreachable!()
+                }
+                _ => unreachable!()
             };
 
             let is_nullable = inner.is_nullable();