-
Notifications
You must be signed in to change notification settings - Fork 285
/
badger.go
41 lines (35 loc) · 999 Bytes
/
badger.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
// Package badger provides a configurable BadgerDB go-datastore for use with
// IPFS Cluster.
package badger
import (
"os"
ds "github.com/ipfs/go-datastore"
badgerds "github.com/ipfs/go-ds-badger"
logging "github.com/ipfs/go-log/v2"
"github.com/pkg/errors"
)
var logger = logging.Logger("badger")
// New returns a BadgerDB datastore configured with the given
// configuration.
func New(cfg *Config) (ds.Datastore, error) {
folder := cfg.GetFolder()
err := os.MkdirAll(folder, 0700)
if err != nil {
return nil, errors.Wrap(err, "creating badger folder")
}
opts := badgerds.Options{
GcDiscardRatio: cfg.GCDiscardRatio,
GcInterval: cfg.GCInterval,
GcSleep: cfg.GCSleep,
Options: cfg.BadgerOptions,
}
return badgerds.NewDatastore(folder, &opts)
}
// Cleanup deletes the badger datastore.
func Cleanup(cfg *Config) error {
folder := cfg.GetFolder()
if _, err := os.Stat(folder); os.IsNotExist(err) {
return nil
}
return os.RemoveAll(cfg.GetFolder())
}