Added support for nested binary

jorgecarleitao · May 5, 2022 · 11dc5b8 · 11dc5b8
1 parent 9a38663
commit 11dc5b8
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/src/io/parquet/read/deserialize/mod.rs b/src/io/parquet/read/deserialize/mod.rs
@@ -181,6 +181,24 @@ where
                 chunk_size,
             )
         }
+        LargeUtf8 => {
+            types.pop();
+            binary::iter_to_arrays_nested::<i64, Utf8Array<i64>, _>(
+                columns.pop().unwrap(),
+                init.pop().unwrap(),
+                field.data_type().clone(),
+                chunk_size,
+            )
+        }
+        Binary => {
+            types.pop();
+            binary::iter_to_arrays_nested::<i32, BinaryArray<i32>, _>(
+                columns.pop().unwrap(),
+                init.pop().unwrap(),
+                field.data_type().clone(),
+                chunk_size,
+            )
+        }
         LargeBinary => {
             types.pop();
             binary::iter_to_arrays_nested::<i64, BinaryArray<i64>, _>(
@@ -190,7 +208,7 @@ where
                 chunk_size,
             )
         }
-        List(inner) => {
+        List(inner) | LargeList(inner) => {
             let iter = columns_to_iter_recursive(
                 vec![columns.pop().unwrap()],
                 types,