69 lines
1.6 KiB
Go
69 lines
1.6 KiB
Go
package database
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"fmt"
|
|
"time"
|
|
|
|
bolt "go.etcd.io/bbolt"
|
|
)
|
|
|
|
// VspDatabase wraps an instance of bbolt DB and provides VSP specific
|
|
// convenience functions.
|
|
type VspDatabase struct {
|
|
db *bolt.DB
|
|
}
|
|
|
|
var (
|
|
// vspBkt is the main parent bucket of the VSP. All values and other buckets
|
|
// are nested within it.
|
|
vspBkt = []byte("vspbkt")
|
|
ticketsBkt = []byte("ticketsbkt")
|
|
versionK = []byte("version")
|
|
version = 1
|
|
)
|
|
|
|
// New initialises and returns a database connection. If no database file is
|
|
// found at the provided path, a new one will be created. Returns an open
|
|
// database connection which should be closed after use.
|
|
func New(dbFile string) (*VspDatabase, error) {
|
|
db, err := bolt.Open(dbFile, 0600, &bolt.Options{Timeout: 1 * time.Second})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("unable to open db file: %v", err)
|
|
}
|
|
|
|
err = createBuckets(db)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &VspDatabase{db: db}, nil
|
|
}
|
|
|
|
// createBuckets creates all storage buckets of the VSP if they don't already
|
|
// exist.
|
|
func createBuckets(db *bolt.DB) error {
|
|
return db.Update(func(tx *bolt.Tx) error {
|
|
if tx.Bucket(vspBkt) == nil {
|
|
parentBkt, err := tx.CreateBucket(vspBkt)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create %s bucket: %v", string(vspBkt), err)
|
|
}
|
|
|
|
vbytes := make([]byte, 4)
|
|
binary.LittleEndian.PutUint32(vbytes, uint32(version))
|
|
err = parentBkt.Put(versionK, vbytes)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = parentBkt.CreateBucket(ticketsBkt)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create %s bucket: %v", string(ticketsBkt), err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|