Merge pull request #295 from Dr-Emann/push-wlprxywvqwns

Lazily compute Iterator len
RoaringBitmap · Oct 30, 2024 · 07d6bc3 · 07d6bc3
2 parents c4e3b34 + 820b270
commit 07d6bc3
Show file tree

Hide file tree

Showing 5 changed files with 370 additions and 46 deletions.
diff --git a/roaring/src/bitmap/container.rs b/roaring/src/bitmap/container.rs
@@ -300,6 +300,21 @@ impl Iterator for Iter<'_> {
     fn next(&mut self) -> Option<u32> {
         self.inner.next().map(|i| util::join(self.key, i))
     }
+
+    fn size_hint(&self) -> (usize, Option<usize>) {
+        self.inner.size_hint()
+    }
+
+    fn count(self) -> usize
+    where
+        Self: Sized,
+    {
+        self.inner.count()
+    }
+
+    fn nth(&mut self, n: usize) -> Option<Self::Item> {
+        self.inner.nth(n).map(|i| util::join(self.key, i))
+    }
 }
 
 impl DoubleEndedIterator for Iter<'_> {
@@ -308,6 +323,8 @@ impl DoubleEndedIterator for Iter<'_> {
     }
 }
 
+impl ExactSizeIterator for Iter<'_> {}
+
 impl fmt::Debug for Container {
     fn fmt(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
         format!("Container<{:?} @ {:?}>", self.len(), self.key).fmt(formatter)