Update to MutableBitmap::extend_constant to initialize a Bitmap of …

…all true values.
jorgecarleitao · Dec 25, 2021 · ad36ece · ad36ece
1 parent 4dfbe3f
commit ad36ece
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/compute/boolean.rs b/src/compute/boolean.rs
@@ -177,8 +177,9 @@ pub fn and_scalar(array: &BooleanArray, scalar: &BooleanScalar) -> BooleanArray
 pub fn or_scalar(array: &BooleanArray, scalar: &BooleanScalar) -> BooleanArray {
     match scalar.value() {
         Some(true) => {
-            let values = Bitmap::from_trusted_len_iter(std::iter::repeat(true).take(array.len()));
-            BooleanArray::from_data(DataType::Boolean, values, array.validity().cloned())
+            let mut values = MutableBitmap::new();
+            values.extend_constant(array.len(), true);
+            BooleanArray::from_data(DataType::Boolean, values.into(), array.validity().cloned())
         }
         Some(false) => array.clone(),
         None => BooleanArray::new_null(DataType::Boolean, array.len()),