mirrors/remark42
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add auto-backup

Browse Source
This commit is contained in:
Umputun
2017-12-24 19:59:09 -06:00
parent 74b67ae2d6
commit 6150fecd81
2 changed files with 33 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/main.go
+31 -5
View File
@@ -1,9 +1,11 @@
package main
import (
"compress/gzip"
"fmt"
"log"
"os"
"time"
"github.com/gorilla/sessions"
"github.com/hashicorp/logutils"
@@ -17,11 +19,12 @@ import (
)
var opts struct {
DBFile string `long:"db" env:"BOLTDB_FILE" default:"/tmp/remark.db" description:"bolt file name"`
SiteURL string `long:"site-url" env:"REMARK_URL" default:"http://remark.umputun.com:8080" description:"url to remark site"`
Admins []string `long:"admin" env:"ADMIN" default:"umputun@gmail.com" description:"admin(s) names" env-delim:","`
DevMode bool `long:"dev" env:"DEV" description:"development mode, no auth enforced"`
Dbg bool `long:"dbg" env:"DEBUG" description:"debug mode"`
DBFile string `long:"db" env:"BOLTDB_FILE" default:"/tmp/remark.db" description:"bolt file name"`
SiteURL string `long:"site-url" env:"REMARK_URL" default:"http://remark.umputun.com:8080" description:"url to remark site"`
Admins []string `long:"admin" env:"ADMIN" default:"umputun@gmail.com" description:"admin(s) names" env-delim:","`
DevMode bool `long:"dev" env:"DEV" description:"development mode, no auth enforced"`
Dbg bool `long:"dbg" env:"DEBUG" description:"debug mode"`
BackupLocation string `long:"backup" env:"BACKUP_PATH" default:"/tmp" description:"backups location"`
ServerCommand struct {
SessionStore string `long:"session" env:"SESSION_STORE" default:"/tmp" description:"path to session store directory"`
@@ -65,6 +68,7 @@ func main() {
}
sessionStore := sessions.NewFilesystemStore(opts.ServerCommand.SessionStore, []byte(opts.ServerCommand.StoreKey))
exporter := migrator.Remark{DataStore: dataStore}
srv := rest.Server{
Version: revision,
@@ -72,6 +76,7 @@ func main() {
SessionStore: sessionStore,
Admins: opts.Admins,
DevMode: opts.DevMode,
Exporter: &exporter,
AuthGoogle: auth.NewGoogle(auth.Params{
Cid: opts.ServerCommand.GoogleCID,
Csecret: opts.ServerCommand.GoogleCSEC,
@@ -90,6 +95,7 @@ func main() {
log.Printf("[WARN] running in dev mode, no auth!")
}
go autoBackup(&exporter)
srv.Run()
}
@@ -112,6 +118,26 @@ func importComments(dataStore store.Interface) error {
return importer.Import(fh, opts.ImportCommand.SiteID)
}
func autoBackup(exporter migrator.Exporter) {
log.Print("[INFO] activate auto-backup")
tick := time.NewTicker(24 * time.Hour)
for _ = range tick.C {
log.Print("[DEBUG] make backup")
fh, err := os.Create(fmt.Sprintf("%s/backup-%s.gz", opts.BackupLocation, time.Now().Format("20060102")))
if err != nil {
log.Printf("[WARN] can't create backup file, %s", err)
continue
}
gz := gzip.NewWriter(fh)
if err = exporter.Export(gz, ""); err != nil {
log.Printf("[WARN] export failed, %+v", err)
}
_ = gz.Close()
_ = fh.Close()
}
}
func setupLog(dbg bool) {
filter := &logutils.LevelFilter{
Levels: []logutils.LogLevel{"DEBUG", "INFO", "WARN", "ERROR"},
app/rest/server.go
+2 -1
View File
@@ -29,6 +29,7 @@ type Server struct {
AuthGoogle *auth.Provider
AuthGithub *auth.Provider
SessionStore *sessions.FilesystemStore
Exporter migrator.Exporter
DevMode bool
mod admin
@@ -61,7 +62,7 @@ func (s *Server) Run() {
})
rapi.With(Auth(s.SessionStore, s.Admins, s.DevMode)).Group(func(radmin chi.Router) {
s.mod = admin{dataStore: s.Store, exporter: &migrator.Remark{DataStore: s.Store}}
s.mod = admin{dataStore: s.Store, exporter: s.Exporter}
radmin.Mount("/admin", s.mod.routes())
})
})