package ethdb import ( "fmt" "github.com/ethereum/eth-go/ethutil" ) /* * This is a test memory database. Do not use for any production it does not get persisted */ type MemDatabase struct { db map[string][]byte } func NewMemDatabase() (*MemDatabase, error) { db := &MemDatabase{db: make(map[string][]byte)} 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 (db *MemDatabase) GetKeys() []*ethutil.Key { data, _ := db.Get([]byte("KeyRing")) return []*ethutil.Key{ethutil.NewKeyFromBytes(data)} } func (db *MemDatabase) Delete(key []byte) error { delete(db.db, string(key)) return nil } func (db *MemDatabase) Print() { for key, val := range db.db { fmt.Printf("%x(%d): ", key, len(key)) node := ethutil.NewValueFromBytes(val) fmt.Printf("%q\n", node.Interface()) } } func (db *MemDatabase) Close() { } func (db *MemDatabase) LastKnownTD() []byte { data, _ := db.Get([]byte("LastKnownTotalDifficulty")) if len(data) == 0 || data == nil { data = []byte{0x0} } return data }