-
Notifications
You must be signed in to change notification settings - Fork 0
/
badger.go
33 lines (28 loc) · 807 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
// 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"
"github.com/pkg/errors"
)
// 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{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())
}