From ad36ece0ad8ff269194b4c82fdfdf53b325ca2a8 Mon Sep 17 00:00:00 2001 From: Steven Gu Date: Sat, 25 Dec 2021 09:13:23 +0000 Subject: [PATCH] Update to `MutableBitmap::extend_constant` to initialize a Bitmap of all true values. --- src/compute/boolean.rs | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/compute/boolean.rs b/src/compute/boolean.rs index 60c5f2a107b..4ef5b46b4b0 100644 --- 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()),