Merge pull request #60 from HerodotusDev/develop

v0.4.5 - Stacked MMR bug fix
HerodotusDev · Aug 27, 2024 · 5d51808 · 5d51808
2 parents dd228bc + eeb5ee2
commit 5d51808
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "accumulators"
-version = "0.4.4"
+version = "0.4.5"
 edition = "2021"
 license-file = "LICENSE"
 description = "Complete package of multiple Accumulators with Stores and hashing functions (Hashers)"

diff --git a/src/mmr/mmrs/stacked.rs b/src/mmr/mmrs/stacked.rs
@@ -139,8 +139,12 @@ impl MMR {
                 _ => return table.default_get_stores_and_full_keys(sub_keys),
             };
 
+            //? Sort the sub MMRs by size in ascending order
+            let mut sub_mmrs = table.sub_mmrs.as_ref().unwrap().clone();
+            sub_mmrs.sort_by(|a, b| a.size.cmp(&b.size));
+
             let mut use_mmr: Option<SubMMR> = None;
-            for sub_mmr in table.sub_mmrs.as_ref().unwrap().iter() {
+            for sub_mmr in sub_mmrs.iter() {
                 if *element_index <= sub_mmr.size {
                     use_mmr = Some(sub_mmr.clone());
                     break;