fix: change initialization for cached_vec

This commit is contained in:
Magamedrasul Ibragimov 2022-11-01 20:31:18 +03:00
parent 5f2b9c61ba
commit 4591a9001c
2 changed files with 9 additions and 3 deletions

View File

@ -50,7 +50,7 @@ where
db.put(NEXT_INDEX_KEY, next_index_val); db.put(NEXT_INDEX_KEY, next_index_val);
// Cache nodes // Cache nodes
let mut cache = Vec::with_capacity(depth + 1); let mut cache = vec![H::default_leaf(); depth + 1];
// Initialize one branch of the `Merkle Tree` from bottom to top // Initialize one branch of the `Merkle Tree` from bottom to top
cache[depth] = H::default_leaf(); cache[depth] = H::default_leaf();
@ -82,7 +82,7 @@ where
let next_index = usize::from_be_bytes(next_index); let next_index = usize::from_be_bytes(next_index);
// Load cache vec // Load cache vec
let mut cache = Vec::with_capacity(depth + 1); let mut cache = vec![H::default_leaf(); depth + 1];
cache[depth] = H::default_leaf(); cache[depth] = H::default_leaf();
for i in (0..depth).rev() { for i in (0..depth).rev() {
cache[i] = H::hash(&[cache[i + 1], cache[i + 1]]); cache[i] = H::hash(&[cache[i + 1], cache[i + 1]]);

View File

@ -63,4 +63,10 @@ impl Hasher for MyKeccak {
} }
#[test] #[test]
fn insert_delete() {} fn insert_delete() {
let mt = MerkleTree::<MemoryDB, MyKeccak>::new(3, "abacaba");
assert_eq!(mt.capacity(), 8);
assert_eq!(mt.depth(), 3);
println!("{:?}", mt.root());
}