65 lines
1.1 KiB
Go
65 lines
1.1 KiB
Go
|
package ethdb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/ethereum/eth-go/ethutil"
|
||
|
|
"github.com/syndtr/goleveldb/leveldb"
|
||
|
|
"path"
|
||
|
|
)
|
||
|
|
|
||
|
|
type LDBDatabase struct {
|
||
|
|
db *leveldb.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewLDBDatabase() (*LDBDatabase, error) {
|
||
|
|
dbPath := path.Join(ethutil.Config.ExecPath, "database")
|
||
|
|
|
||
|
|
// Open the db
|
||
|
|
db, err := leveldb.OpenFile(dbPath, nil)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
database := &LDBDatabase{db: db}
|
||
|
|
|
||
|
|
return database, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *LDBDatabase) Put(key []byte, value []byte) {
|
||
|
|
err := db.db.Put(key, value, nil)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Println("Error put", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *LDBDatabase) Get(key []byte) ([]byte, error) {
|
||
|
|
return db.db.Get(key, nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *LDBDatabase) LastKnownTD() []byte {
|
||
|
|
data, _ := db.db.Get([]byte("LastKnownTotalDifficulty"), nil)
|
||
|
|
|
||
|
|
if len(data) == 0 {
|
||
|
|
data = []byte{0x0}
|
||
|
|
}
|
||
|
|
|
||
|
|
return data
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *LDBDatabase) Close() {
|
||
|
|
// Close the leveldb database
|
||
|
|
db.db.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *LDBDatabase) Print() {
|
||
|
|
iter := db.db.NewIterator(nil)
|
||
|
|
for iter.Next() {
|
||
|
|
key := iter.Key()
|
||
|
|
value := iter.Value()
|
||
|
|
|
||
|
|
fmt.Printf("%x(%d): ", key, len(key))
|
||
|
|
node := ethutil.NewValueFromBytes(value)
|
||
|
|
fmt.Printf("%v\n", node)
|
||
|
|
}
|
||
|
|
}
|