105 lines
2.2 KiB
Go
105 lines
2.2 KiB
Go
package boltdb
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/bucket"
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/viper"
|
|
"go.etcd.io/bbolt"
|
|
)
|
|
|
|
type (
|
|
boltBucket struct {
|
|
db *bbolt.DB
|
|
name []byte
|
|
}
|
|
|
|
// Options groups the BoltDB bucket's options.
|
|
Options struct {
|
|
bbolt.Options
|
|
Name []byte
|
|
Path string
|
|
Perm os.FileMode
|
|
}
|
|
)
|
|
|
|
const defaultFilePermission = 0777
|
|
|
|
var errEmptyPath = errors.New("database empty path")
|
|
|
|
const name = "boltbucket"
|
|
|
|
func makeCopy(val []byte) []byte {
|
|
tmp := make([]byte, len(val))
|
|
copy(tmp, val)
|
|
|
|
return tmp
|
|
}
|
|
|
|
// NewOptions prepares options for badger instance.
|
|
func NewOptions(v *viper.Viper) (opts Options, err error) {
|
|
opts = Options{
|
|
Options: bbolt.Options{
|
|
// set defaults:
|
|
Timeout: bbolt.DefaultOptions.Timeout,
|
|
FreelistType: bbolt.DefaultOptions.FreelistType,
|
|
|
|
// set config options:
|
|
NoSync: v.GetBool(name + ".no_sync"),
|
|
ReadOnly: v.GetBool(name + ".read_only"),
|
|
NoGrowSync: v.GetBool(name + ".no_grow_sync"),
|
|
NoFreelistSync: v.GetBool(name + ".no_freelist_sync"),
|
|
|
|
PageSize: v.GetInt(name + ".page_size"),
|
|
MmapFlags: v.GetInt(name + ".mmap_flags"),
|
|
InitialMmapSize: v.GetInt(name + ".initial_mmap_size"),
|
|
},
|
|
|
|
Name: []byte(name),
|
|
Perm: defaultFilePermission,
|
|
Path: v.GetString(name + ".path"),
|
|
}
|
|
|
|
if opts.Path == "" {
|
|
return opts, errEmptyPath
|
|
}
|
|
|
|
if tmp := v.GetDuration(name + ".lock_timeout"); tmp > 0 {
|
|
opts.Timeout = tmp
|
|
}
|
|
|
|
if perm := v.GetUint32(name + ".perm"); perm != 0 {
|
|
opts.Perm = os.FileMode(perm)
|
|
}
|
|
|
|
base := path.Dir(opts.Path)
|
|
if err := os.MkdirAll(base, opts.Perm); err != nil {
|
|
return opts, errors.Wrapf(err, "could not use `%s` dir", base)
|
|
}
|
|
|
|
return opts, nil
|
|
}
|
|
|
|
// NewBucket creates badger-bucket instance.
|
|
func NewBucket(opts *Options) (bucket.Bucket, error) {
|
|
log.SetOutput(ioutil.Discard) // disable default logger
|
|
|
|
db, err := bbolt.Open(opts.Path, opts.Perm, &opts.Options)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = db.Update(func(tx *bbolt.Tx) error {
|
|
_, err := tx.CreateBucketIfNotExists(opts.Name)
|
|
return err
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &boltBucket{db: db, name: opts.Name}, nil
|
|
}
|