107 lines
1.8 KiB
Go
107 lines
1.8 KiB
Go
//nolint:dupl // maybe we'll figure out how to make DB resource accessors generic in the future
|
|
package badger
|
|
|
|
import (
|
|
"encoding/json"
|
|
"path"
|
|
|
|
"github.com/cirruslabs/orchard/pkg/resource/v1"
|
|
"github.com/dgraph-io/badger/v3"
|
|
)
|
|
|
|
const SpaceVMs = "/vms"
|
|
|
|
func VMKey(name string) []byte {
|
|
return []byte(path.Join(SpaceVMs, name))
|
|
}
|
|
|
|
func (txn *Transaction) GetVM(name string) (_ *v1.VM, err error) {
|
|
defer func() {
|
|
err = mapErr(err)
|
|
}()
|
|
|
|
key := VMKey(name)
|
|
|
|
item, err := txn.badgerTxn.Get(key)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
valueBytes, err := item.ValueCopy(nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var vm v1.VM
|
|
|
|
err = json.Unmarshal(valueBytes, &vm)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
vm.Version = item.Version()
|
|
|
|
return &vm, nil
|
|
}
|
|
|
|
func (txn *Transaction) SetVM(vm v1.VM) (err error) {
|
|
defer func() {
|
|
err = mapErr(err)
|
|
}()
|
|
|
|
key := VMKey(vm.Name)
|
|
|
|
valueBytes, err := json.Marshal(vm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return txn.badgerTxn.Set(key, valueBytes)
|
|
}
|
|
|
|
func (txn *Transaction) DeleteVM(name string) (err error) {
|
|
defer func() {
|
|
err = mapErr(err)
|
|
}()
|
|
|
|
key := VMKey(name)
|
|
|
|
return txn.badgerTxn.Delete(key)
|
|
}
|
|
|
|
func (txn *Transaction) ListVMs() (_ []v1.VM, err error) {
|
|
defer func() {
|
|
err = mapErr(err)
|
|
}()
|
|
|
|
// Declare an empty, non-nil slice to
|
|
// return [] when no VMs are found
|
|
result := []v1.VM{}
|
|
|
|
it := txn.badgerTxn.NewIterator(badger.IteratorOptions{
|
|
Prefix: []byte(SpaceVMs),
|
|
})
|
|
defer it.Close()
|
|
|
|
for it.Rewind(); it.Valid(); it.Next() {
|
|
item := it.Item()
|
|
|
|
vmBytes, err := item.ValueCopy(nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var vm v1.VM
|
|
|
|
if err := json.Unmarshal(vmBytes, &vm); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
vm.Version = item.Version()
|
|
|
|
result = append(result, vm)
|
|
}
|
|
|
|
return result, nil
|
|
}
|