grassrootseconomics/openethereum
5
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

printing trie

Browse Source
This commit is contained in:
debris 2015-12-01 19:19:16 +01:00
parent 4f5399c558
commit b8b8e4c61a
1 changed files with 28 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
src/trie.rs
View File

@ -146,22 +146,8 @@ pub struct TrieDB {
impl fmt::Debug for TrieDB { impl fmt::Debug for TrieDB {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
try!(writeln!(f, "[")); try!(writeln!(f, "["));
{
let print_indent = | fo: &mut fmt::Formatter, times | { for i in 0..times { write!(fo, " "); }};
let root_rlp = self.db.lookup(&self.root).expect("Trie root not found!"); let root_rlp = self.db.lookup(&self.root).expect("Trie root not found!");
self.fmt_all(root_rlp, f, 0);
self.apply_to_all(root_rlp, &mut | node, deepness, index | {
print_indent(f, deepness);
match *node {
Node::Leaf(ref slice, ref value) => { writeln!(f, "{:x}: Leaf {:?}, {:?}", index, slice, value); },
Node::ExtensionRaw(ref slice, ref item) => { write!(f, "Extension (raw): "); }
Node::ExtensionSha3(ref slice, ref sha3) => { write!(f, "Extension (sha3): "); }
Node::Branch(Some(ref nodes), ref value) => { writeln!(f, "{:x} Branch: ", index); }
_ => { writeln!(f, "node"); }
}
}, 0, 0);
}
writeln!(f, "]") writeln!(f, "]")
} }
} }
@ -197,30 +183,47 @@ impl TrieDB {
} }
} }
fn apply_to_all<F>(&self, node: &[u8], f: &mut F, deepness: usize, index: usize) where F: FnMut(&Node, usize, usize) -> () { fn fmt_indent(&self, f: &mut fmt::Formatter, size: usize) -> fmt::Result {
for i in 0..size {
try!(write!(f, " "));
}
Ok(())
}
fn fmt_all(&self, node: &[u8], f: &mut fmt::Formatter, deepness: usize) -> fmt::Result {
let node = Node::decoded(node); let node = Node::decoded(node);
match node { match node {
Node::Leaf(_, _) => f(&node, deepness, index), Node::Leaf(slice, value) => try!(writeln!(f, "Leaf {:?}, {:?}", slice, value.pretty())),
Node::ExtensionRaw(_, ref item) => { Node::ExtensionRaw(_, ref item) => {
f(&node, deepness, index); self.fmt_indent(f, deepness);
self.apply_to_all(item, f, deepness + 1, 0); try!(write!(f, "Extension (raw): "));
self.fmt_all(item, f, deepness + 1);
}, },
Node::ExtensionSha3(_, sha3) => { Node::ExtensionSha3(_, sha3) => {
f(&node, deepness, index); self.fmt_indent(f, deepness);
try!(write!(f, "Extension (sha3): "));
let rlp = self.db.lookup(&H256::from_slice(sha3)).expect("sha3 not found!"); let rlp = self.db.lookup(&H256::from_slice(sha3)).expect("sha3 not found!");
self.apply_to_all(rlp, f, deepness + 1, 0); self.fmt_all(rlp, f, deepness + 1);
}, },
Node::Branch(Some(ref nodes), ref value) => { Node::Branch(Some(ref nodes), ref value) => {
f(&node, deepness, index); try!(writeln!(f, "Branch: "));
for i in 0..16 { for i in 0..16 {
self.apply_to_all(nodes[i], f, deepness + 1, i); match Node::decoded(nodes[i]) {
Node::Branch(None, _) => (),
_ => {
self.fmt_indent(f, deepness + 1);
try!(write!(f, "{:x}: ", i));
self.fmt_all(nodes[i], f, deepness + 1);
}
}
} }
}, },
// empty // empty
n @ Node::Branch(_, _) => { n @ Node::Branch(_, _) => {
// do nothing // do nothing
} }
} };
Ok(())
} }
fn get<'a>(&'a self, key: &NibbleSlice<'a>) -> Option<&'a [u8]> { fn get<'a>(&'a self, key: &NibbleSlice<'a>) -> Option<&'a [u8]> {