57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package cluster
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/costela/wesher/common"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// State keeps track of information needed to rejoin the cluster
|
|
type state struct {
|
|
ClusterKey []byte
|
|
Nodes []common.Node
|
|
}
|
|
|
|
const defaultStatePath = "/var/lib/wesher/state.json"
|
|
|
|
func (s *state) save(statePath string) error {
|
|
if statePath == "" {
|
|
statePath = defaultStatePath
|
|
}
|
|
if err := os.MkdirAll(path.Dir(statePath), 0700); err != nil {
|
|
return err
|
|
}
|
|
|
|
stateOut, err := json.MarshalIndent(s, "", " ")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return ioutil.WriteFile(statePath, stateOut, 0600)
|
|
}
|
|
|
|
func loadState(cs *state, statePath string) {
|
|
if statePath == "" {
|
|
statePath = defaultStatePath
|
|
}
|
|
content, err := ioutil.ReadFile(statePath)
|
|
if err != nil {
|
|
if !os.IsNotExist(err) {
|
|
logrus.Warnf("could not open state in %s: %s", statePath, err)
|
|
}
|
|
return
|
|
}
|
|
|
|
// avoid partially unmarshalled content by using a temp var
|
|
csTmp := &state{}
|
|
if err := json.Unmarshal(content, csTmp); err != nil {
|
|
logrus.Warnf("could not decode state: %s", err)
|
|
} else {
|
|
*cs = *csTmp
|
|
}
|
|
}
|