Decode from stream directly

This commit is contained in:
obscuren 2014-12-23 15:18:48 +01:00
parent 1382e8d84b
commit 34f72ddb3d
2 changed files with 6 additions and 3 deletions

@ -18,7 +18,6 @@
package main package main
import ( import (
"bytes"
"fmt" "fmt"
"os" "os"
"runtime" "runtime"
@ -120,13 +119,13 @@ func main() {
if len(ImportChain) > 0 { if len(ImportChain) > 0 {
clilogger.Infof("importing chain '%s'\n", ImportChain) clilogger.Infof("importing chain '%s'\n", ImportChain)
c, err := ethutil.ReadAllFile(ImportChain) fh, err := os.OpenFile(ImportChain, os.O_RDONLY, os.ModePerm)
if err != nil { if err != nil {
clilogger.Infoln(err) clilogger.Infoln(err)
return return
} }
var chain types.Blocks var chain types.Blocks
if err := rlp.Decode(bytes.NewReader([]byte(c)), &chain); err != nil { if err := rlp.Decode(fh, &chain); err != nil {
clilogger.Infoln(err) clilogger.Infoln(err)
return return
} }

@ -71,6 +71,10 @@ func (self *LDBDatabase) NewIterator() iterator.Iterator {
return self.db.NewIterator(nil, nil) return self.db.NewIterator(nil, nil)
} }
func (self *LDBDatabase) Write(batch *leveldb.Batch) error {
return self.db.Write(batch, nil)
}
func (self *LDBDatabase) Close() { func (self *LDBDatabase) Close() {
// Close the leveldb database // Close the leveldb database
self.db.Close() self.db.Close()