95 lines
1.9 KiB
Go
95 lines
1.9 KiB
Go
|
package boltdb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/mr-tron/base58"
|
||
|
|
"github.com/nspcc-dev/neofs-node/lib/core"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
"go.etcd.io/bbolt"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Get value by key or return error.
|
||
|
|
func (b *bucket) Get(key []byte) (data []byte, err error) {
|
||
|
|
err = b.db.View(func(txn *bbolt.Tx) error {
|
||
|
|
txn.Bucket(b.name).Cursor().Seek(key)
|
||
|
|
val := txn.Bucket(b.name).Get(key)
|
||
|
|
if val == nil {
|
||
|
|
return errors.Wrapf(core.ErrNotFound, "key=%s", base58.Encode(key))
|
||
|
|
}
|
||
|
|
|
||
|
|
data = makeCopy(val)
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// Set value for key.
|
||
|
|
func (b *bucket) Set(key, value []byte) error {
|
||
|
|
return b.db.Update(func(txn *bbolt.Tx) error {
|
||
|
|
k, v := makeCopy(key), makeCopy(value)
|
||
|
|
return txn.Bucket(b.name).Put(k, v)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
// Del removes item from bucket by key.
|
||
|
|
func (b *bucket) Del(key []byte) error {
|
||
|
|
return b.db.Update(func(txn *bbolt.Tx) error {
|
||
|
|
return txn.Bucket(b.name).Delete(key)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
// Has checks key exists.
|
||
|
|
func (b *bucket) Has(key []byte) bool {
|
||
|
|
_, err := b.Get(key)
|
||
|
|
return !errors.Is(errors.Cause(err), core.ErrNotFound)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Size returns size of database.
|
||
|
|
func (b *bucket) Size() int64 {
|
||
|
|
info, err := os.Stat(b.db.Path())
|
||
|
|
if err != nil {
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
|
||
|
|
return info.Size()
|
||
|
|
}
|
||
|
|
|
||
|
|
// List all items in bucket.
|
||
|
|
func (b *bucket) List() ([][]byte, error) {
|
||
|
|
var items [][]byte
|
||
|
|
|
||
|
|
if err := b.db.View(func(txn *bbolt.Tx) error {
|
||
|
|
return txn.Bucket(b.name).ForEach(func(k, _ []byte) error {
|
||
|
|
items = append(items, makeCopy(k))
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
}); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return items, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Filter elements by filter closure.
|
||
|
|
func (b *bucket) Iterate(handler core.FilterHandler) error {
|
||
|
|
if handler == nil {
|
||
|
|
return core.ErrNilFilterHandler
|
||
|
|
}
|
||
|
|
|
||
|
|
return b.db.View(func(txn *bbolt.Tx) error {
|
||
|
|
return txn.Bucket(b.name).ForEach(func(k, v []byte) error {
|
||
|
|
if !handler(makeCopy(k), makeCopy(v)) {
|
||
|
|
return core.ErrIteratingAborted
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
// Close bucket database.
|
||
|
|
func (b *bucket) Close() error {
|
||
|
|
return b.db.Close()
|
||
|
|
}
|