shardtree: Add ShardTree::store_mut

zcash · Mar 11, 2024 · fa147c8 · fa147c8
1 parent 25cb189
commit fa147c8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/shardtree/CHANGELOG.md b/shardtree/CHANGELOG.md
@@ -8,7 +8,7 @@ and this project adheres to Rust's notion of
 ## Unreleased
 
 ## Added
-* `Shardtree::store`
+* `Shardtree::{store, store_mut}`
 - `ShardTree::insert_frontier`
 
 ## Changed

diff --git a/shardtree/src/lib.rs b/shardtree/src/lib.rs
@@ -97,6 +97,11 @@ impl<
         &self.store
     }
 
+    /// Returns a mutable reference to the underlying [`ShardStore`].
+    pub fn store_mut(&mut self) -> &mut S {
+        &mut self.store
+    }
+
     /// Returns the root address of the tree.
     pub fn root_addr() -> Address {
         Address::from_parts(Level::from(DEPTH), 0)