frostfs-node/lib/buckets/boltdb/methods.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()
}