add null_count for MutablePrimitiveArray

jorgecarleitao · Nov 6, 2023 · ac5f5d7 · ac5f5d7
1 parent 747005e
commit ac5f5d7
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/array/primitive/mutable.rs b/src/array/primitive/mutable.rs
@@ -93,6 +93,20 @@ impl<T: NativeType> MutablePrimitiveArray<T> {
     pub fn apply_values<F: Fn(&mut [T])>(&mut self, f: F) {
         f(&mut self.values);
     }
+
+    /// The number of null slots on this [`Array`].
+    /// # Implementation
+    /// This is `O(1)` since the number of null elements is pre-computed.
+    #[inline]
+    pub fn null_count(&self) -> usize {
+        if self.data_type() == &DataType::Null {
+            return self.len();
+        };
+        self.validity()
+            .as_ref()
+            .map(|x| x.unset_bits())
+            .unwrap_or(0)
+    }
 }
 
 impl<T: NativeType> Default for MutablePrimitiveArray<T> {