50 lines
944 B
Go
50 lines
944 B
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"errors"
|
||
|
|
|
||
|
|
badger "github.com/dgraph-io/badger/v3"
|
||
|
|
"github.com/eternal-flame-AD/yoake/config"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DB interface {
|
||
|
|
NewTransaction(update bool) DBTxn
|
||
|
|
}
|
||
|
|
|
||
|
|
type DBTxn interface {
|
||
|
|
Set(key, value []byte) error
|
||
|
|
Delete(key []byte) error
|
||
|
|
Get(key []byte) ([]byte, error)
|
||
|
|
Commit() error
|
||
|
|
Discard()
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetJSON(t DBTxn, key []byte, v interface{}) error {
|
||
|
|
if data, err := t.Get(key); err != nil {
|
||
|
|
return err
|
||
|
|
} else {
|
||
|
|
return json.Unmarshal(data, v)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func SetJSON(t DBTxn, key []byte, v interface{}) error {
|
||
|
|
if data, err := json.Marshal(v); err != nil {
|
||
|
|
return err
|
||
|
|
} else {
|
||
|
|
return t.Set(key, data)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func New(conf config.C) (DB, error) {
|
||
|
|
if conf.DB.Badger.Dir != "" {
|
||
|
|
opts := badger.DefaultOptions(conf.DB.Badger.Dir)
|
||
|
|
if db, err := badger.Open(opts); err != nil {
|
||
|
|
return nil, err
|
||
|
|
} else {
|
||
|
|
return &BadgerDB{db}, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil, errors.New("no database configured")
|
||
|
|
}
|