chore: add store_mut fn to MMRBatch

nervosnetwork · Jul 27, 2023 · 267702a · 267702a
1 parent f2a3211
commit 267702a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/mmr.rs b/src/mmr.rs
@@ -45,9 +45,17 @@ impl<T, M, S> MMR<T, M, S> {
         &self.batch
     }
 
+    pub fn batch_mut(&mut self) -> &mut MMRBatch<T, S> {
+        &mut self.batch
+    }
+
     pub fn store(&self) -> &S {
         self.batch.store()
     }
+
+    pub fn store_mut(&mut self) -> &mut S {
+        self.batch.store_mut()
+    }
 }
 
 impl<T: Clone + PartialEq, M: Merge<Item = T>, S: MMRStoreReadOps<T>> MMR<T, M, S> {

diff --git a/src/mmr_store.rs b/src/mmr_store.rs
@@ -27,6 +27,10 @@ impl<Elem, Store> MMRBatch<Elem, Store> {
     pub fn store(&self) -> &Store {
         &self.store
     }
+
+    pub fn store_mut(&mut self) -> &mut Store {
+        &mut self.store
+    }
 }
 
 impl<Elem: Clone, Store: MMRStoreReadOps<Elem>> MMRBatch<Elem, Store> {