Improved performance of is_no_null. (#600)

jorgecarleitao · Nov 12, 2021 · e661065 · e661065
1 parent eb6ef19
commit e661065
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/compute/boolean.rs b/src/compute/boolean.rs
@@ -124,7 +124,11 @@ pub fn is_null(input: &dyn Array) -> BooleanArray {
 /// ```
 pub fn is_not_null(input: &dyn Array) -> BooleanArray {
     let values = match input.validity() {
-        None => Bitmap::from_trusted_len_iter(std::iter::repeat(true).take(input.len())),
+        None => {
+            let mut mutable = MutableBitmap::new();
+            mutable.extend_constant(input.len(), true);
+            mutable.into()
+        }
         Some(buffer) => buffer.clone(),
     };
     BooleanArray::from_data(DataType::Boolean, values, None)