vise/persist/fs.go

package persist

import (
	"io/ioutil"
	"log"
	"path"
	"path/filepath"
	"github.com/fxamacker/cbor/v2"

	"git.defalsify.org/vise/cache"
	"git.defalsify.org/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
}