Optimized binaryArray. (#597)

jorgecarleitao · Nov 12, 2021 · edf5cbc · edf5cbc
1 parent 3caa204
commit edf5cbc
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/array/binary/mutable.rs b/src/array/binary/mutable.rs
@@ -136,11 +136,10 @@ impl<O: Offset> MutableBinaryArray<O> {
     }
 
     fn init_validity(&mut self) {
-        self.validity = Some(MutableBitmap::from_trusted_len_iter(
-            std::iter::repeat(true)
-                .take(self.len() - 1)
-                .chain(std::iter::once(false)),
-        ))
+        let mut validity = MutableBitmap::with_capacity(self.offsets.capacity());
+        validity.extend_constant(self.len(), true);
+        validity.set(self.len() - 1, false);
+        self.validity = Some(validity)
     }
 
     /// Converts itself into an [`Array`].