forked from cerc-io/plugeth
61 lines
1.1 KiB
Go
61 lines
1.1 KiB
Go
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
|
|
}
|