parent
7185d4ac9b
commit
fbe99c65ec
@ -1,3 +1,4 @@
|
|||||||
config.yml
|
config.yml
|
||||||
telegabber
|
telegabber
|
||||||
sessions/
|
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