aboutsummaryrefslogtreecommitdiffstats
path: root/ethutil/trie.go
diff options
context:
space:
mode:
authorobscuren <geffobscura@gmail.com>2014-05-27 07:08:51 +0800
committerobscuren <geffobscura@gmail.com>2014-05-27 07:08:51 +0800
commit5cdfee51437532ccfb49e874fdbbea2702c3d13f (patch)
treefe7178f2c877da1fca17dc3529f8a9d934ea2617 /ethutil/trie.go
parent4c7bd75c1a4ccbed09a10c81dbfbbcffdc66c411 (diff)
downloaddexon-5cdfee51437532ccfb49e874fdbbea2702c3d13f.tar
dexon-5cdfee51437532ccfb49e874fdbbea2702c3d13f.tar.gz
dexon-5cdfee51437532ccfb49e874fdbbea2702c3d13f.tar.bz2
dexon-5cdfee51437532ccfb49e874fdbbea2702c3d13f.tar.lz
dexon-5cdfee51437532ccfb49e874fdbbea2702c3d13f.tar.xz
dexon-5cdfee51437532ccfb49e874fdbbea2702c3d13f.tar.zst
dexon-5cdfee51437532ccfb49e874fdbbea2702c3d13f.zip
New Trie iterator
Diffstat (limited to 'ethutil/trie.go')
-rw-r--r--ethutil/trie.go46
1 files changed, 46 insertions, 0 deletions
diff --git a/ethutil/trie.go b/ethutil/trie.go
index c993e4d8f..18d0a5f0a 100644
--- a/ethutil/trie.go
+++ b/ethutil/trie.go
@@ -442,6 +442,8 @@ type TrieIterator struct {
shas [][]byte
values []string
+
+ lastNode []byte
}
func (t *Trie) NewIterator() *TrieIterator {
@@ -513,3 +515,47 @@ func (it *TrieIterator) Key() string {
func (it *TrieIterator) Value() string {
return ""
}
+
+type EachCallback func(key string, node *Value)
+
+func (it *TrieIterator) Each(cb EachCallback) {
+ it.fetchNode(nil, NewValue(it.trie.Root).Bytes(), cb)
+}
+
+func (it *TrieIterator) fetchNode(key []int, node []byte, cb EachCallback) {
+ it.iterateNode(key, it.trie.cache.Get(node), cb)
+}
+
+func (it *TrieIterator) iterateNode(key []int, currentNode *Value, cb EachCallback) {
+ if currentNode.Len() == 2 {
+ k := CompactDecode(currentNode.Get(0).Str())
+
+ if currentNode.Get(1).Str() == "" {
+ it.iterateNode(key, currentNode.Get(1), cb)
+ } else {
+ pk := append(key, k...)
+
+ if k[len(k)-1] == 16 {
+ cb(DecodeCompact(pk), currentNode.Get(1))
+ } else {
+ it.fetchNode(pk, currentNode.Get(1).Bytes(), cb)
+ }
+ }
+ } else {
+ for i := 0; i < currentNode.Len(); i++ {
+ pk := append(key, i)
+ if i == 16 && currentNode.Get(i).Len() != 0 {
+ cb(DecodeCompact(pk), currentNode.Get(i))
+ } else {
+ if currentNode.Get(i).Str() == "" {
+ it.iterateNode(pk, currentNode.Get(i), cb)
+ } else {
+ val := currentNode.Get(i).Str()
+ if val != "" {
+ it.fetchNode(pk, []byte(val), cb)
+ }
+ }
+ }
+ }
+ }
+}