mirror of https://github.com/ethereum/go-ethereum
parent
a926686445
commit
f17930eb46
@ -0,0 +1,50 @@ |
||||
package main |
||||
|
||||
// Database interface
|
||||
type Database interface { |
||||
Put(key []byte, value []byte) |
||||
Get(key []byte) ([]byte, error) |
||||
} |
||||
|
||||
type Trie struct { |
||||
root string |
||||
db Database |
||||
} |
||||
|
||||
func NewTrie(db Database) *Trie { |
||||
return &Trie{db: db, root: ""} |
||||
} |
||||
|
||||
func (t *Trie) Update(key string, value string) { |
||||
k := CompactHexDecode(key) |
||||
|
||||
t.root = t.UpdateState(t.root, k, value) |
||||
} |
||||
|
||||
func (t *Trie) Get(key []byte) ([]byte, error) { |
||||
return nil, nil |
||||
} |
||||
|
||||
// Inserts a new sate or delete a state based on the value
|
||||
func (t *Trie) UpdateState(node string, key []int, value string) string { |
||||
if value != "" { |
||||
return t.InsertState(node, ""/*key*/, value) |
||||
} else { |
||||
// delete it
|
||||
} |
||||
|
||||
return "" |
||||
} |
||||
|
||||
func (t *Trie) InsertState(node, key, value string) string { |
||||
return "" |
||||
} |
||||
|
||||
func (t *Trie) Put(node []byte) []byte { |
||||
enc := Encode(node) |
||||
sha := Sha256Bin(enc) |
||||
|
||||
t.db.Put([]byte(sha), enc) |
||||
|
||||
return sha |
||||
} |
@ -0,0 +1,61 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"testing" |
||||
) |
||||
|
||||
type MemDatabase struct { |
||||
db map[string][]byte |
||||
trie *Trie |
||||
} |
||||
|
||||
func NewMemDatabase() (*MemDatabase, error) { |
||||
db := &MemDatabase{db: make(map[string][]byte)} |
||||
|
||||
db.trie = NewTrie(db) |
||||
|
||||
return db, nil |
||||
} |
||||
|
||||
func (db *MemDatabase) Put(key []byte, value []byte) { |
||||
db.db[string(key)] = value |
||||
} |
||||
|
||||
func (db *MemDatabase) Get(key []byte) ([]byte, error) { |
||||
return db.db[string(key)], nil |
||||
} |
||||
|
||||
func TestTriePut(t *testing.T) { |
||||
db, err := NewMemDatabase() |
||||
|
||||
if err != nil { |
||||
t.Error("Error starting db") |
||||
} |
||||
|
||||
key := db.trie.Put([]byte("testing node")) |
||||
|
||||
data, err := db.Get(key) |
||||
if err != nil { |
||||
t.Error("Nothing at node") |
||||
} |
||||
|
||||
s, _ := Decode(data, 0) |
||||
if str, ok := s.([]byte); ok { |
||||
if string(str) != "testing node" { |
||||
t.Error("Wrong value node", str) |
||||
} |
||||
} else { |
||||
t.Error("Invalid return type") |
||||
} |
||||
} |
||||
|
||||
func TestTrieUpdate(t *testing.T) { |
||||
db, err := NewMemDatabase() |
||||
|
||||
if err != nil { |
||||
t.Error("Error starting db") |
||||
} |
||||
|
||||
db.trie.Update("test", "test") |
||||
} |
||||
|
Loading…
Reference in new issue