Browse Source

libs/db: Fix the BoltDB Batch.Delete

`Delete` should queue a delete operation that targets the entire
database, and not just keys that happen to be in the current batch.
pull/3720/head
Yawning Angel 6 years ago
parent
commit
f46ed4aac8
2 changed files with 18 additions and 27 deletions
  1. +1
    -0
      CHANGELOG_PENDING.md
  2. +17
    -27
      libs/db/boltdb.go

+ 1
- 0
CHANGELOG_PENDING.md View File

@ -28,3 +28,4 @@
* [rpc] [\#3686](https://github.com/tendermint/tendermint/pull/3686) `HTTPClient#Call` returns wrapped errors, so a caller could use `errors.Cause` to retrieve an error code. * [rpc] [\#3686](https://github.com/tendermint/tendermint/pull/3686) `HTTPClient#Call` returns wrapped errors, so a caller could use `errors.Cause` to retrieve an error code.
### BUG FIXES: ### BUG FIXES:
- [libs/db] Fixed the BoltDB backend's Batch.Delete implementation (@Yawning)

+ 17
- 27
libs/db/boltdb.go View File

@ -149,55 +149,45 @@ func (bdb *BoltDB) Stats() map[string]string {
// boltDBBatch stores key values in sync.Map and dumps them to the underlying // boltDBBatch stores key values in sync.Map and dumps them to the underlying
// DB upon Write call. // DB upon Write call.
type boltDBBatch struct { type boltDBBatch struct {
buffer []struct {
k []byte
v []byte
}
db *BoltDB
db *BoltDB
ops []operation
} }
// NewBatch returns a new batch. // NewBatch returns a new batch.
func (bdb *BoltDB) NewBatch() Batch { func (bdb *BoltDB) NewBatch() Batch {
return &boltDBBatch{ return &boltDBBatch{
buffer: make([]struct {
k []byte
v []byte
}, 0),
db: bdb,
ops: nil,
db: bdb,
} }
} }
// It is safe to modify the contents of the argument after Set returns but not // It is safe to modify the contents of the argument after Set returns but not
// before. // before.
func (bdb *boltDBBatch) Set(key, value []byte) { func (bdb *boltDBBatch) Set(key, value []byte) {
bdb.buffer = append(bdb.buffer, struct {
k []byte
v []byte
}{
key, value,
})
bdb.ops = append(bdb.ops, operation{opTypeSet, key, value})
} }
// It is safe to modify the contents of the argument after Delete returns but // It is safe to modify the contents of the argument after Delete returns but
// not before. // not before.
func (bdb *boltDBBatch) Delete(key []byte) { func (bdb *boltDBBatch) Delete(key []byte) {
for i, elem := range bdb.buffer {
if bytes.Equal(elem.k, key) {
// delete without preserving order
bdb.buffer[i] = bdb.buffer[len(bdb.buffer)-1]
bdb.buffer = bdb.buffer[:len(bdb.buffer)-1]
return
}
}
bdb.ops = append(bdb.ops, operation{opTypeDelete, key, nil})
} }
// NOTE: the operation is synchronous (see BoltDB for reasons) // NOTE: the operation is synchronous (see BoltDB for reasons)
func (bdb *boltDBBatch) Write() { func (bdb *boltDBBatch) Write() {
err := bdb.db.db.Batch(func(tx *bbolt.Tx) error { err := bdb.db.db.Batch(func(tx *bbolt.Tx) error {
b := tx.Bucket(bucket) b := tx.Bucket(bucket)
for _, elem := range bdb.buffer {
if putErr := b.Put(elem.k, elem.v); putErr != nil {
return putErr
for _, op := range bdb.ops {
key := nonEmptyKey(nonNilBytes(op.key))
switch op.opType {
case opTypeSet:
if putErr := b.Put(key, op.value); putErr != nil {
return putErr
}
case opTypeDelete:
if delErr := b.Delete(key); delErr != nil {
return delErr
}
} }
} }
return nil return nil


Loading…
Cancel
Save