chore: add backup app

pkg/core/backup/timer/qiniu/qiniu.go

@@ -0,0 +1,77 @@
+// Package qiniu provides ...
+package qiniu
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"os"
+	"os/exec"
+	"time"
+
+	"github.com/eiblog/eiblog/pkg/config"
+	"github.com/eiblog/eiblog/pkg/internal"
+)
+
+// Storage qiniu storage
+type Storage struct{}
+
+// BackupData implements timer.Storage
+func (s Storage) BackupData(now time.Time) error {
+	switch config.Conf.Database.Source {
+	case "mongodb":
+		return backupFromMongoDB(now)
+	default:
+		return errors.New("unsupported database source backup to qiniu")
+	}
+}
+
+func backupFromMongoDB(now time.Time) error {
+	ctx, cancel := context.WithTimeout(context.Background(), time.Minute*20)
+	defer cancel()
+
+	// dump
+	arg := fmt.Sprintf("mongodump -h %s -d eiblog -o /tmp",
+		config.Conf.Database.Source)
+	cmd := exec.CommandContext(ctx, "sh", "-c", arg)
+	err := cmd.Run()
+	if err != nil {
+		return err
+	}
+	// tar
+	name := fmt.Sprintf("eiblog-%s.tar.gz", now.Format("2006-01-02"))
+	arg = fmt.Sprintf("tar czf %s /tmp/eiblog", name)
+	cmd = exec.CommandContext(ctx, "sh", "-c", arg)
+	err = cmd.Run()
+	if err != nil {
+		return err
+	}
+
+	// upload file
+	f, err := os.Open("/tmp/eiblog/" + name)
+	if err != nil {
+		return err
+	}
+	s, err := f.Stat()
+	if err != nil {
+		return err
+	}
+	uploadParams := internal.UploadParams{
+		Name: name,
+		Size: s.Size(),
+		Data: f,
+
+		Conf: config.Conf.BackupApp.Qiniu,
+	}
+	_, err = internal.QiniuUpload(uploadParams)
+	if err != nil {
+		return err
+	}
+	// after days delete
+	deleteParams := internal.DeleteParams{
+		Name: name,
+
+		Conf: config.Conf.BackupApp.Qiniu,
+	}
+	return internal.QiniuDelete(deleteParams)
+}

pkg/core/backup/timer/timer.go

@@ -0,0 +1,68 @@
+// Package timer provides ...
+package timer
+
+import (
+	"errors"
+	"strconv"
+	"time"
+
+	"github.com/eiblog/eiblog/pkg/config"
+	"github.com/eiblog/eiblog/pkg/core/backup/timer/qiniu"
+
+	"github.com/sirupsen/logrus"
+)
+
+// Start to backup with ticker
+func Start() error {
+	var storage Storage
+	// backup instance
+	switch config.Conf.BackupApp.BackupTo {
+	case "qiniu":
+		storage = qiniu.Storage{}
+
+	default:
+		return errors.New("timer: unknown backup to driver: " +
+			config.Conf.BackupApp.BackupTo)
+	}
+
+	// parse duration
+	interval, err := ParseDuration(config.Conf.BackupApp.Interval)
+	if err != nil {
+		return err
+	}
+
+	t := time.NewTicker(interval)
+	for now := range t.C {
+		err = storage.BackupData(now)
+		if err != nil {
+			logrus.Error("timer: Start.BackupData: ", now, err)
+		}
+	}
+	return nil
+}
+
+// ParseDuration parse string to duration
+func ParseDuration(d string) (time.Duration, error) {
+	if len(d) == 0 {
+		return 0, errors.New("timer: incorrect duration input")
+	}
+
+	length := len(d)
+	switch d[length-1] {
+	case 's', 'm', 'h':
+		return time.ParseDuration(d)
+	case 'd':
+		di, err := strconv.Atoi(d[:length-1])
+		if err != nil {
+			return 0, err
+		}
+		return time.Duration(di) * time.Hour * 24, nil
+	}
+
+	return 0, errors.New("timer: unsupported duration:" + d)
+}
+
+// Storage backup backend
+type Storage interface {
+	BackupData(now time.Time) error
+}