perf: Add benchmarks for BtreeMap::Iterator::count()

benchmarks/src/btreemap.rs

@@ -1,3 +1,5 @@
+use std::ops::Bound;
+
 use crate::Random;
 use canbench::{benchmark, macros::bench, BenchResult};
 use ic_stable_structures::{storable::Blob, BTreeMap, DefaultMemoryImpl, Storable};
@@ -219,6 +221,53 @@ pub fn btreemap_insert_10mib_values() -> BenchResult {
     })
 }
 
+#[bench]
+pub fn btreemap_iter_count() -> BenchResult {
+    let mut btree = BTreeMap::new(DefaultMemoryImpl::default());
+    let size: u8 = 200;
+    for i in 0..size {
+        btree.insert(i, i);
+    }
+
+    benchmark(|| {
+        for i in 0..size {
+            for j in i + 1..size {
+                btree
+                    .range((Bound::Included(i), Bound::Included(j)))
+                    .count();
+            }
+        }
+    })
+}
+
+#[bench]
+pub fn btreemap_iter_count_10mib_values() -> BenchResult {
+    let mut btree = BTreeMap::new(DefaultMemoryImpl::default());
+
+    // Insert 200 10MiB values.
+    let mut rng = Rng::from_seed(0);
+    let mut values = vec![];
+    for _ in 0..200 {
+        values.push(rng.iter(Rand::rand_u8).take(10 * 1024).collect::<Vec<_>>());
+    }
+
+    let mut i = 0u8;
+    for value in values.into_iter() {
+        btree.insert(i, value);
+        i += 1;
+    }
+
+    benchmark(|| {
+        for j in 0..i {
+            for k in j + 1..i {
+                btree
+                    .range((Bound::Included(j), Bound::Included(k)))
+                    .count();
+            }
+        }
+    })
+}
+
 /// Benchmarks removing keys from a BTreeMap.
 #[bench]
 pub fn btreemap_remove_blob_4_1024() -> BenchResult {