Bulk insert Variants

Author: friendlymatthew

parquet-variant-compute/src/variant_array.rs

@@ -451,7 +451,7 @@ impl<'m, 'v> FromIterator<Option<Variant<'m, 'v>>> for VariantArray {
 impl<'m, 'v> FromIterator<Variant<'m, 'v>> for VariantArray {
     fn from_iter<T: IntoIterator<Item = Variant<'m, 'v>>>(iter: T) -> Self {
         let mut b = VariantArrayBuilder::new(0);
-        b.extend(iter.into_iter().map(Some));
+        b.extend(iter);
         b.build()
     }
 }
@@ -1451,7 +1451,7 @@ mod test {
             Variant::ShortString(ShortString::try_new("norm").unwrap()),
         ];
 
-        let variant_array = VariantArray::from_iter(v.into_iter());
+        let variant_array = VariantArray::from_iter(v);
 
         assert_eq!(variant_array.len(), 3);
 

parquet-variant-compute/src/variant_array_builder.rs

@@ -173,6 +173,12 @@ impl VariantArrayBuilder {
     }
 }
 
+impl<'m, 'v> Extend<Variant<'m, 'v>> for VariantArrayBuilder {
+    fn extend<T: IntoIterator<Item = Variant<'m, 'v>>>(&mut self, iter: T) {
+        iter.into_iter().for_each(|v| self.append_variant(v));
+    }
+}
+
 impl<'m, 'v> Extend<Option<Variant<'m, 'v>>> for VariantArrayBuilder {
     fn extend<T: IntoIterator<Item = Option<Variant<'m, 'v>>>>(&mut self, iter: T) {
         for v in iter {