fix: public getter

This commit is contained in:
rymnc 2023-05-10 04:39:04 +05:30
parent b3a02216ce
commit 8b38443637
No known key found for this signature in database
GPG Key ID: AAA088D5C68ECD34
1 changed files with 10 additions and 1 deletions

View File

@ -136,7 +136,7 @@ where
}
// Recalculates `Merkle Tree` from the specified key
fn recalculate_from(&mut self, key: usize) -> PmtreeResult<()> {
pub fn recalculate_from(&mut self, key: usize) -> PmtreeResult<()> {
let mut depth = self.depth;
let mut i = key;
@ -174,6 +174,15 @@ where
Ok(res)
}
// Returns the leaf by the key
pub fn get(&self, key: usize) -> PmtreeResult<H::Fr> {
if key >= self.capacity() {
return Err(PmtreeErrorKind::TreeError(TreeErrorKind::IndexOutOfBounds));
}
self.get_elem(Key(self.depth, key))
}
/// Deletes a leaf at the `key` by setting it to its default value
pub fn delete(&mut self, key: usize) -> PmtreeResult<()> {
if key >= self.next_index {