Merge branch 'main' into perf/replace-partition-point

Author: marvin-j97

Cargo.toml

@@ -48,6 +48,9 @@ nanoid = "0.4.0"
 rand = "0.9.0"
 test-log = "0.2.16"
 
+# half 2.5.0 has MSRV 1.81
+half = "=2.4.0"
+
 [package.metadata.cargo-all-features]
 denylist = []
 

src/abstract.rs

@@ -101,6 +101,11 @@ pub trait AbstractTree {
     /// Returns the amount of disk segments currently in the tree.
     fn segment_count(&self) -> usize;
 
+    /// Returns the amount of segments in levels[idx].
+    ///
+    /// Returns `None` if the level does not exist (if idx >= 7).
+    fn level_segment_count(&self, idx: usize) -> Option<usize>;
+
     /// Returns the amount of blob files currently in the tree.
     fn blob_file_count(&self) -> usize {
         0

src/blob_tree/mod.rs

@@ -459,6 +459,10 @@ impl AbstractTree for BlobTree {
         self.index.segment_count()
     }
 
+    fn level_segment_count(&self, idx: usize) -> Option<usize> {
+        self.index.level_segment_count(idx)
+    }
+
     fn first_level_segment_count(&self) -> usize {
         self.index.first_level_segment_count()
     }

src/tree/mod.rs

@@ -271,6 +271,15 @@ impl AbstractTree for Tree {
         self.levels.read().expect("lock is poisoned").len()
     }
 
+    fn level_segment_count(&self, idx: usize) -> Option<usize> {
+        self.levels
+            .read()
+            .expect("lock is poisoned")
+            .levels
+            .get(idx)
+            .map(|x| x.len())
+    }
+
     fn first_level_segment_count(&self) -> usize {
         self.levels
             .read()