65 lines
1 KiB
Go
65 lines
1 KiB
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"path"
|
||
|
|
|
||
|
|
"go.etcd.io/bbolt"
|
||
|
|
)
|
||
|
|
|
||
|
|
type (
|
||
|
|
Bolt struct {
|
||
|
|
db *bbolt.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
Status struct {
|
||
|
|
ID []byte
|
||
|
|
Failed bool
|
||
|
|
Notified bool
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
statusBucket = []byte("status")
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewBolt(filename string) (*Bolt, error) {
|
||
|
|
dbPath := path.Join(filename)
|
||
|
|
|
||
|
|
db, err := bbolt.Open(dbPath, 0600, nil)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("bolot init: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return &Bolt{db}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Bolt) Status(id []byte) (st Status, err error) {
|
||
|
|
return st, b.db.Update(func(tx *bbolt.Tx) error {
|
||
|
|
bkt, err := tx.CreateBucketIfNotExists(statusBucket)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
v := bkt.Get(id)
|
||
|
|
if len(v) == 0 {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return json.Unmarshal(v, &st)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Bolt) SetStatus(id []byte, st Status) error {
|
||
|
|
return b.db.Update(func(tx *bbolt.Tx) error {
|
||
|
|
bkt, err := tx.CreateBucketIfNotExists(statusBucket)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
v, err := json.Marshal(st)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return bkt.Put(id, v)
|
||
|
|
})
|
||
|
|
}
|