parent
7185d4ac9b
commit
fbe99c65ec
@ -1,3 +1,4 @@
|
||||
config.yml
|
||||
telegabber
|
||||
sessions/
|
||||
session.dat
|
||||
|
@ -0,0 +1,69 @@
|
||||
package persistence
|
||||
|
||||
import (
|
||||
"github.com/pkg/errors"
|
||||
"io/ioutil"
|
||||
|
||||
"dev.narayana.im/narayana/telegabber/yamldb"
|
||||
|
||||
"gopkg.in/yaml.v2"
|
||||
)
|
||||
|
||||
// SessionsYamlDB wraps YamlDB with Session
|
||||
type SessionsYamlDB struct {
|
||||
yamldb.YamlDB
|
||||
Data *SessionsMap
|
||||
}
|
||||
|
||||
// SessionsMap is for :sessions: subtree
|
||||
type SessionsMap struct {
|
||||
Sessions map[string]Session `yaml:":sessions"`
|
||||
}
|
||||
|
||||
// Session is a key-values subtree
|
||||
type Session struct {
|
||||
Login string `yaml:":login"`
|
||||
}
|
||||
|
||||
var sessionDB SessionsYamlDB
|
||||
|
||||
// Marshaller implementation for YamlDB
|
||||
func Marshaller() ([]byte, error) {
|
||||
return yaml.Marshal(sessionDB.Data)
|
||||
}
|
||||
|
||||
// LoadSessions restores TDlib sessions from the previous run
|
||||
func LoadSessions(path string) (SessionsYamlDB, error) {
|
||||
var sessionData SessionsMap
|
||||
|
||||
sessionDB, err := initYamlDB(path, &sessionData)
|
||||
if err != nil {
|
||||
return sessionDB, errors.Wrap(err, "Sessions restore error")
|
||||
}
|
||||
|
||||
sessionDB.Transaction(func() {
|
||||
}, Marshaller)
|
||||
|
||||
return sessionDB, nil
|
||||
}
|
||||
|
||||
func initYamlDB(path string, dataPtr *SessionsMap) (SessionsYamlDB, error) {
|
||||
file, err := ioutil.ReadFile(path)
|
||||
if err == nil {
|
||||
err = yaml.Unmarshal(file, dataPtr)
|
||||
if err != nil {
|
||||
return SessionsYamlDB{}, errors.Wrap(err, "YamlDB is corrupted")
|
||||
}
|
||||
} else {
|
||||
// DB file does not exist, create an empty DB
|
||||
dataPtr.Sessions = make(map[string]Session)
|
||||
}
|
||||
|
||||
return SessionsYamlDB{
|
||||
YamlDB: yamldb.YamlDB{
|
||||
Path: path,
|
||||
PathNew: path + ".new",
|
||||
},
|
||||
Data: dataPtr,
|
||||
}, nil
|
||||
}
|
@ -0,0 +1,47 @@
|
||||
package yamldb
|
||||
|
||||
import (
|
||||
"github.com/pkg/errors"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"sync"
|
||||
|
||||
log "github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
// YamlDB represents a YAML file database instance
|
||||
type YamlDB struct {
|
||||
Path string
|
||||
PathNew string
|
||||
lock sync.Mutex
|
||||
}
|
||||
|
||||
// Transaction executes the given callback and safely saves
|
||||
// the data after they are modified within the callback
|
||||
func (db *YamlDB) Transaction(callback func(), marshaller func() ([]byte, error)) error {
|
||||
var err error
|
||||
|
||||
log.Debug("Enter transaction")
|
||||
db.lock.Lock()
|
||||
defer func() {
|
||||
db.lock.Unlock()
|
||||
log.Debug("Exit transaction")
|
||||
}()
|
||||
|
||||
callback()
|
||||
|
||||
yamlData, err := marshaller()
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "Data marshalling error")
|
||||
}
|
||||
err = ioutil.WriteFile(db.PathNew, yamlData, 0644)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "YamlDB write failure")
|
||||
}
|
||||
err = os.Rename(db.PathNew, db.Path)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "Couldn't rewrite an old YamlDB file")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
Loading…
Reference in new issue