ethdb: add NewBatch

This commit is contained in:
Felix Lange 2015-08-18 13:42:21 +02:00
parent 8c4dab77ba
commit 8b32f10f16
3 changed files with 49 additions and 0 deletions

View File

@ -268,3 +268,23 @@ func (self *LDBDatabase) meter(refresh time.Duration) {
} }
} }
} }
// TODO: remove this stuff and expose leveldb directly
func (db *LDBDatabase) NewBatch() Batch {
return &ldbBatch{db: db.db, b: new(leveldb.Batch)}
}
type ldbBatch struct {
db *leveldb.DB
b *leveldb.Batch
}
func (b *ldbBatch) Put(key, value []byte) error {
b.b.Put(key, value)
return nil
}
func (b *ldbBatch) Write() error {
return b.db.Write(b.b, nil)
}

View File

@ -22,4 +22,10 @@ type Database interface {
Delete(key []byte) error Delete(key []byte) error
Close() Close()
Flush() error Flush() error
NewBatch() Batch
}
type Batch interface {
Put(key, value []byte) error
Write() error
} }

View File

@ -95,3 +95,26 @@ func (db *MemDatabase) LastKnownTD() []byte {
func (db *MemDatabase) Flush() error { func (db *MemDatabase) Flush() error {
return nil return nil
} }
func (db *MemDatabase) NewBatch() Batch {
return &memBatch{db: db}
}
type kv struct{ k, v []byte }
type memBatch struct {
db *MemDatabase
writes []kv
}
func (w *memBatch) Put(key, value []byte) error {
w.writes = append(w.writes, kv{key, common.CopyBytes(value)})
return nil
}
func (w *memBatch) Write() error {
for _, kv := range w.writes {
w.db.db[string(kv.k)] = kv.v
}
return nil
}