87 lines
1.9 KiB
Go
87 lines
1.9 KiB
Go
package persist
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"log"
|
|
"path"
|
|
"path/filepath"
|
|
|
|
"github.com/fxamacker/cbor/v2"
|
|
|
|
"git.grassecon.net/kamikazechaser/vise/cache"
|
|
"git.grassecon.net/kamikazechaser/vise/state"
|
|
)
|
|
|
|
// FsPersister is an implementation of Persister that saves state to the file system.
|
|
type FsPersister struct {
|
|
State *state.State
|
|
Memory *cache.Cache
|
|
dir string
|
|
}
|
|
|
|
// NewFsPersister creates a new FsPersister.
|
|
//
|
|
// The filesystem store will be at the given directory. The directory must exist.
|
|
func NewFsPersister(dir string) *FsPersister {
|
|
fp, err := filepath.Abs(dir)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return &FsPersister{
|
|
dir: fp,
|
|
}
|
|
}
|
|
|
|
// WithContent sets a current State and Cache object.
|
|
//
|
|
// This method is normally called before Serialize / Save.
|
|
func (p *FsPersister) WithContent(st *state.State, ca *cache.Cache) *FsPersister {
|
|
p.State = st
|
|
p.Memory = ca
|
|
return p
|
|
}
|
|
|
|
// GetState implements the Persister interface.
|
|
func (p *FsPersister) GetState() *state.State {
|
|
return p.State
|
|
}
|
|
|
|
// GetState implements the Persister interface.
|
|
func (p *FsPersister) GetMemory() cache.Memory {
|
|
return p.Memory
|
|
}
|
|
|
|
// GetState implements the Persister interface.
|
|
func (p *FsPersister) Serialize() ([]byte, error) {
|
|
return cbor.Marshal(p)
|
|
}
|
|
|
|
// GetState implements the Persister interface.
|
|
func (p *FsPersister) Deserialize(b []byte) error {
|
|
err := cbor.Unmarshal(b, p)
|
|
return err
|
|
}
|
|
|
|
// GetState implements the Persister interface.
|
|
func (p *FsPersister) Save(key string) error {
|
|
b, err := p.Serialize()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fp := path.Join(p.dir, key)
|
|
log.Printf("saved key %v state %x", key, p.State.Code)
|
|
return ioutil.WriteFile(fp, b, 0600)
|
|
}
|
|
|
|
// GetState implements the Persister interface.
|
|
func (p *FsPersister) Load(key string) error {
|
|
fp := path.Join(p.dir, key)
|
|
b, err := ioutil.ReadFile(fp)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = p.Deserialize(b)
|
|
log.Printf("loaded key %v state %x", key, p.State.Code)
|
|
return err
|
|
}
|