node.go 1.31 KB
Newer Older
1
package trie
obscuren's avatar
obscuren committed
2 3 4 5 6 7 8

import "fmt"

var indices = []string{"0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "a", "b", "c", "d", "e", "f", "[17]"}

type Node interface {
	Value() Node
9
	Copy(*Trie) Node // All nodes, for now, return them self
obscuren's avatar
obscuren committed
10 11 12 13 14 15 16 17 18 19
	Dirty() bool
	fstring(string) string
	Hash() interface{}
	RlpData() interface{}
}

// Value node
func (self *ValueNode) String() string            { return self.fstring("") }
func (self *FullNode) String() string             { return self.fstring("") }
func (self *ShortNode) String() string            { return self.fstring("") }
20
func (self *ValueNode) fstring(ind string) string { return fmt.Sprintf("%x ", self.data) }
21 22 23 24 25

//func (self *HashNode) fstring(ind string) string  { return fmt.Sprintf("< %x > ", self.key) }
func (self *HashNode) fstring(ind string) string {
	return fmt.Sprintf("%v", self.trie.trans(self))
}
obscuren's avatar
obscuren committed
26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42

// Full node
func (self *FullNode) fstring(ind string) string {
	resp := fmt.Sprintf("[\n%s  ", ind)
	for i, node := range self.nodes {
		if node == nil {
			resp += fmt.Sprintf("%s: <nil> ", indices[i])
		} else {
			resp += fmt.Sprintf("%s: %v", indices[i], node.fstring(ind+"  "))
		}
	}

	return resp + fmt.Sprintf("\n%s] ", ind)
}

// Short node
func (self *ShortNode) fstring(ind string) string {
43
	return fmt.Sprintf("[ %x: %v ] ", self.key, self.value.fstring(ind+"  "))
obscuren's avatar
obscuren committed
44
}