mirror of https://github.com/qwc/backive.git
45 lines
892 B
Go
45 lines
892 B
Go
|
package backive
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"path"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Database is a simple string to string mapping, where arbitrary strings can be stored and safed to disk or loaded
|
||
|
|
type Database struct {
|
||
|
|
data map[string]string
|
||
|
|
}
|
||
|
|
|
||
|
|
var dbPath string = "/var/lib/backive/data.json"
|
||
|
|
|
||
|
|
// Save saves the database
|
||
|
|
func (d *Database) Save() {
|
||
|
|
jsonstr, merr := json.Marshal(d.data)
|
||
|
|
if merr != nil {
|
||
|
|
panic(merr)
|
||
|
|
}
|
||
|
|
log.Printf("Writing database output to file: %s", jsonstr)
|
||
|
|
saveDir, _ := path.Split(dbPath)
|
||
|
|
CreateDirectoryIfNotExists(saveDir)
|
||
|
|
err := os.WriteFile(dbPath, []byte(jsonstr), 0644)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// LoadDb loads the database
|
||
|
|
func (d *Database) Load() {
|
||
|
|
if _, err := os.Stat(dbPath); err == nil {
|
||
|
|
data, rferr := os.ReadFile(dbPath)
|
||
|
|
if rferr != nil {
|
||
|
|
panic(rferr)
|
||
|
|
}
|
||
|
|
json.Unmarshal(data, &d.data)
|
||
|
|
} else if os.IsNotExist(err) {
|
||
|
|
// no data
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|