Added len to every array

src/array/binary/iterator.rs

@@ -1,5 +1,5 @@
 use crate::{
-    array::{Array, Offset},
+    array::Offset,
     bitmap::utils::{zip_validity, ZipValidity},
     trusted_len::TrustedLen,
 };

src/array/binary/mod.rs

@@ -161,6 +161,12 @@ impl<O: Offset> BinaryArray<O> {
 
 // accessors
 impl<O: Offset> BinaryArray<O> {
+    /// Returns the length of this array
+    #[inline]
+    pub fn len(&self) -> usize {
+        self.offsets.len() - 1
+    }
+
     /// Returns the element at index `i`
     /// # Panics
     /// iff `i >= self.len()`
@@ -211,7 +217,7 @@ impl<O: Offset> Array for BinaryArray<O> {
 
     #[inline]
     fn len(&self) -> usize {
-        self.offsets.len() - 1
+        self.len()
     }
 
     #[inline]

src/array/boolean/mod.rs

@@ -99,6 +99,12 @@ impl BooleanArray {
 
 // accessors
 impl BooleanArray {
+    /// Returns the length of this array
+    #[inline]
+    pub fn len(&self) -> usize {
+        self.values.len()
+    }
+
     /// Returns the value at index `i`
     /// # Panic
     /// This function panics iff `i >= self.len()`.
@@ -136,7 +142,7 @@ impl Array for BooleanArray {
 
     #[inline]
     fn len(&self) -> usize {
-        self.values.len()
+        self.len()
     }
 
     #[inline]

src/array/dictionary/iterator.rs

@@ -1,4 +1,3 @@
-use crate::array::Array;
 use crate::bitmap::utils::{zip_validity, ZipValidity};
 use crate::scalar::Scalar;
 use crate::trusted_len::TrustedLen;

src/array/dictionary/mod.rs

@@ -119,6 +119,15 @@ impl<K: DictionaryKey> DictionaryArray<K> {
         arr.values = Arc::from(arr.values.with_validity(validity));
         arr
     }
+}
+
+// accessors
+impl<K: DictionaryKey> DictionaryArray<K> {
+    /// Returns the length of this array
+    #[inline]
+    pub fn len(&self) -> usize {
+        self.keys.len()
+    }
 
     /// The optional validity. Equivalent to `self.keys().validity()`.
     #[inline]
@@ -169,7 +178,7 @@ impl<K: DictionaryKey> Array for DictionaryArray<K> {
 
     #[inline]
     fn len(&self) -> usize {
-        self.keys.len()
+        self.len()
     }
 
     #[inline]

src/array/equal/boolean.rs

@@ -1,4 +1,4 @@
-use crate::array::{Array, BooleanArray};
+use crate::array::BooleanArray;
 
 pub(super) fn equal(lhs: &BooleanArray, rhs: &BooleanArray) -> bool {
     lhs.len() == rhs.len() && lhs.iter().eq(rhs.iter())

src/array/fixed_size_binary/mod.rs

@@ -85,6 +85,27 @@ impl FixedSizeBinaryArray {
         }
     }
 
+    /// Sets the validity bitmap on this [`FixedSizeBinaryArray`].
+    /// # Panic
+    /// This function panics iff `validity.len() != self.len()`.
+    pub fn with_validity(&self, validity: Option<Bitmap>) -> Self {
+        if matches!(&validity, Some(bitmap) if bitmap.len() != self.len()) {
+            panic!("validity should be as least as large as the array")
+        }
+        let mut arr = self.clone();
+        arr.validity = validity;
+        arr
+    }
+}
+
+// accessors
+impl FixedSizeBinaryArray {
+    /// Returns the length of this array
+    #[inline]
+    pub fn len(&self) -> usize {
+        self.values.len() / self.size as usize
+    }
+
     /// The optional validity.
     #[inline]
     pub fn validity(&self) -> Option<&Bitmap> {
@@ -118,18 +139,6 @@ impl FixedSizeBinaryArray {
     pub fn size(&self) -> usize {
         self.size
     }
-
-    /// Sets the validity bitmap on this [`FixedSizeBinaryArray`].
-    /// # Panic
-    /// This function panics iff `validity.len() != self.len()`.
-    pub fn with_validity(&self, validity: Option<Bitmap>) -> Self {
-        if matches!(&validity, Some(bitmap) if bitmap.len() != self.len()) {
-            panic!("validity should be as least as large as the array")
-        }
-        let mut arr = self.clone();
-        arr.validity = validity;
-        arr
-    }
 }
 
 impl FixedSizeBinaryArray {
@@ -149,7 +158,7 @@ impl Array for FixedSizeBinaryArray {
 
     #[inline]
     fn len(&self) -> usize {
-        self.values.len() / self.size as usize
+        self.len()
     }
 
     #[inline]

src/array/fixed_size_list/mod.rs

@@ -99,6 +99,27 @@ impl FixedSizeListArray {
         }
     }
 
+    /// Sets the validity bitmap on this [`FixedSizeListArray`].
+    /// # Panic
+    /// This function panics iff `validity.len() != self.len()`.
+    pub fn with_validity(&self, validity: Option<Bitmap>) -> Self {
+        if matches!(&validity, Some(bitmap) if bitmap.len() != self.len()) {
+            panic!("validity should be as least as large as the array")
+        }
+        let mut arr = self.clone();
+        arr.validity = validity;
+        arr
+    }
+}
+
+// accessors
+impl FixedSizeListArray {
+    /// Returns the length of this array
+    #[inline]
+    pub fn len(&self) -> usize {
+        self.values.len() / self.size as usize
+    }
+
     /// The optional validity.
     #[inline]
     pub fn validity(&self) -> Option<&Bitmap> {
@@ -127,18 +148,6 @@ impl FixedSizeListArray {
         self.values
             .slice_unchecked(i * self.size as usize, self.size as usize)
     }
-
-    /// Sets the validity bitmap on this [`FixedSizeListArray`].
-    /// # Panic
-    /// This function panics iff `validity.len() != self.len()`.
-    pub fn with_validity(&self, validity: Option<Bitmap>) -> Self {
-        if matches!(&validity, Some(bitmap) if bitmap.len() != self.len()) {
-            panic!("validity should be as least as large as the array")
-        }
-        let mut arr = self.clone();
-        arr.validity = validity;
-        arr
-    }
 }
 
 impl FixedSizeListArray {
@@ -164,7 +173,7 @@ impl Array for FixedSizeListArray {
 
     #[inline]
     fn len(&self) -> usize {
-        self.values.len() / self.size as usize
+        self.len()
     }
 
     #[inline]

src/array/list/mod.rs

@@ -123,6 +123,12 @@ impl<O: Offset> ListArray<O> {
 
 // Accessors
 impl<O: Offset> ListArray<O> {
+    /// Returns the length of this array
+    #[inline]
+    pub fn len(&self) -> usize {
+        self.offsets.len() - 1
+    }
+
     /// Returns the element at index `i`
     #[inline]
     pub fn value(&self, i: usize) -> Box<dyn Array> {
@@ -211,7 +217,7 @@ impl<O: Offset> Array for ListArray<O> {
 
     #[inline]
     fn len(&self) -> usize {
-        self.offsets.len() - 1
+        self.len()
     }
 
     #[inline]

src/array/map/mod.rs

@@ -115,6 +115,12 @@ impl MapArray {
 
 // Accessors
 impl MapArray {
+    /// Returns the length of this array
+    #[inline]
+    pub fn len(&self) -> usize {
+        self.offsets.len() - 1
+    }
+
     /// returns the offsets
     #[inline]
     pub fn offsets(&self) -> &Buffer<i32> {
@@ -161,7 +167,7 @@ impl Array for MapArray {
 
     #[inline]
     fn len(&self) -> usize {
-        self.offsets.len() - 1
+        self.len()
     }
 
     #[inline]

src/array/null.rs

@@ -32,6 +32,11 @@ impl NullArray {
             length,
         }
     }
+
+    #[inline]
+    fn len(&self) -> usize {
+        self.length
+    }
 }
 
 impl Array for NullArray {
@@ -42,7 +47,7 @@ impl Array for NullArray {
 
     #[inline]
     fn len(&self) -> usize {
-        self.length
+        self.len()
     }
 
     #[inline]

src/array/primitive/mod.rs

@@ -120,6 +120,14 @@ impl<T: NativeType> PrimitiveArray<T> {
         arr.validity = validity;
         arr
     }
+}
+
+impl<T: NativeType> PrimitiveArray<T> {
+    /// Returns the length of this array
+    #[inline]
+    pub fn len(&self) -> usize {
+        self.values.len()
+    }
 
     /// The optional validity.
     #[inline]

src/array/struct_.rs

@@ -148,6 +148,11 @@ impl StructArray {
 
 // Accessors
 impl StructArray {
+    #[inline]
+    fn len(&self) -> usize {
+        self.values[0].len()
+    }
+
     /// The optional validity.
     #[inline]
     pub fn validity(&self) -> Option<&Bitmap> {
@@ -184,7 +189,7 @@ impl Array for StructArray {
 
     #[inline]
     fn len(&self) -> usize {
-        self.values[0].len()
+        self.len()
     }
 
     #[inline]

src/array/union/iterator.rs

@@ -1,4 +1,4 @@
-use super::{Array, UnionArray};
+use super::UnionArray;
 use crate::{scalar::Scalar, trusted_len::TrustedLen};
 
 #[derive(Debug, Clone)]