feat: replace --operation flag by backup and restore command

pkg/backup.go

@@ -7,20 +7,72 @@ package pkg
 import (
 	"fmt"
 	"github.com/jkaninda/pg-bkup/utils"
+	"github.com/spf13/cobra"
 	"log"
 	"os"
 	"os/exec"
 	"time"
 )
 
-var (
-	dbName      = ""
-	dbHost      = ""
-	dbPort      = ""
-	dbPassword  = ""
-	dbUserName  = ""
-	storagePath = "/backup"
-)
+func StartBackup(cmd *cobra.Command) {
+	_, _ = cmd.Flags().GetString("operation")
+
+	//Set env
+	utils.SetEnv("STORAGE_PATH", storagePath)
+	utils.GetEnv(cmd, "dbname", "DB_NAME")
+	utils.GetEnv(cmd, "port", "DB_PORT")
+	utils.GetEnv(cmd, "period", "SCHEDULE_PERIOD")
+
+	//Get flag value and set env
+	s3Path = utils.GetEnv(cmd, "path", "S3_PATH")
+	storage = utils.GetEnv(cmd, "storage", "STORAGE")
+	file = utils.GetEnv(cmd, "file", "FILE_NAME")
+	disableCompression, _ = cmd.Flags().GetBool("disable-compression")
+	executionMode, _ = cmd.Flags().GetString("mode")
+
+	if executionMode == "default" {
+		if storage == "s3" {
+			utils.Info("Backup database to s3 storage")
+			s3Backup(disableCompression, s3Path)
+		} else {
+			utils.Info("Backup database to local storage")
+			BackupDatabase(disableCompression)
+
+		}
+	} else if executionMode == "scheduled" {
+		scheduledMode()
+	} else {
+		utils.Fatal("Error, unknown execution mode!")
+	}
+
+}
+
+// Run in scheduled mode
+func scheduledMode() {
+
+	fmt.Println()
+	fmt.Println("**********************************")
+	fmt.Println("     Starting PostgreSQL Bkup...   ")
+	fmt.Println("***********************************")
+	utils.Info("Running in Scheduled mode")
+	utils.Info("Log file in /var/log/pg-bkup.log")
+	utils.Info("Execution period ", os.Getenv("SCHEDULE_PERIOD"))
+
+	//Test database connexion
+	utils.TestDatabaseConnection()
+
+	//Test database connexion
+	utils.TestDatabaseConnection()
+
+	utils.Info("Creating backup job...")
+	CreateCrontabScript(disableCompression, storage)
+
+	//Start Supervisor
+	supervisordCmd := exec.Command("supervisord", "-c", "/etc/supervisor/supervisord.conf")
+	if err := supervisordCmd.Run(); err != nil {
+		utils.Fatalf("Error starting supervisord: %v\n", err)
+	}
+}
 
 // BackupDatabase backup database
 func BackupDatabase(disableCompression bool) {
@@ -38,12 +90,13 @@ func BackupDatabase(disableCompression bool) {
 		if err != nil {
 			return
 		}
-		//Test Database connexion
 		utils.TestDatabaseConnection()
-		// Backup database
+		// Backup Database database
 		utils.Info("Backing up database...")
+		//Generate file name
 		bkFileName := fmt.Sprintf("%s_%s.sql.gz", dbName, time.Now().Format("20060102_150405"))
 
+		// Verify is compression is disabled
 		if disableCompression {
 			bkFileName = fmt.Sprintf("%s_%s.sql", dbName, time.Now().Format("20060102_150405"))
 			// Execute pg_dump
@@ -111,3 +164,9 @@ func BackupDatabase(disableCompression bool) {
 	}
 
 }
+
+func s3Backup(disableCompression bool, s3Path string) {
+	// Backup Database to S3 storage
+	MountS3Storage(s3Path)
+	BackupDatabase(disableCompression)
+}

pkg/restore.go

@@ -3,11 +3,35 @@ package pkg
 import (
 	"fmt"
 	"github.com/jkaninda/pg-bkup/utils"
+	"github.com/spf13/cobra"
 	"os"
 	"os/exec"
 	"path/filepath"
 )
 
+func StartRestore(cmd *cobra.Command) {
+
+	//Set env
+	utils.SetEnv("STORAGE_PATH", storagePath)
+	utils.GetEnv(cmd, "dbname", "DB_NAME")
+	utils.GetEnv(cmd, "port", "DB_PORT")
+
+	//Get flag value and set env
+	s3Path = utils.GetEnv(cmd, "path", "S3_PATH")
+	storage = utils.GetEnv(cmd, "storage", "STORAGE")
+	file = utils.GetEnv(cmd, "file", "FILE_NAME")
+	executionMode, _ = cmd.Flags().GetString("mode")
+
+	if storage == "s3" {
+		utils.Info("Restore database from s3")
+		s3Restore(file, s3Path)
+	} else {
+		utils.Info("Restore database from local")
+		RestoreDatabase(file)
+
+	}
+}
+
 // RestoreDatabase restore database
 func RestoreDatabase(file string) {
 	dbHost = os.Getenv("DB_HOST")
@@ -16,6 +40,9 @@ func RestoreDatabase(file string) {
 	dbName = os.Getenv("DB_NAME")
 	dbPort = os.Getenv("DB_PORT")
 	storagePath = os.Getenv("STORAGE_PATH")
+	if file == "" {
+		utils.Fatal("Error required --file")
+	}
 
 	if os.Getenv("DB_HOST") == "" || os.Getenv("DB_NAME") == "" || os.Getenv("DB_USERNAME") == "" || os.Getenv("DB_PASSWORD") == "" || file == "" {
 		utils.Fatal("Please make sure all required environment variables are set")
@@ -56,3 +83,8 @@ func RestoreDatabase(file string) {
 		}
 	}
 }
+func s3Restore(file, s3Path string) {
+	// Restore database from S3
+	MountS3Storage(s3Path)
+	RestoreDatabase(file)
+}

pkg/s3fs.go

@@ -11,9 +11,6 @@ import (
 	"os/exec"
 )
 
-const s3MountPath string = "/s3mnt"
-const s3fsPasswdFile string = "/etc/passwd-s3fs"
-
 var (
 	accessKey  = ""
 	secretKey  = ""
@@ -21,6 +18,10 @@ var (
 	s3Endpoint = ""
 )
 
+func S3Mount() {
+	MountS3Storage(s3Path)
+}
+
 // MountS3Storage Mount s3 storage using s3fs
 func MountS3Storage(s3Path string) {
 	accessKey = os.Getenv("ACCESS_KEY")
@@ -44,7 +45,9 @@ func MountS3Storage(s3Path string) {
 		}
 		//Change file permission
 		utils.ChangePermission(s3fsPasswdFile, 0600)
-		utils.Info("Mounting Object storage in", s3MountPath)
+
+		//Mount object storage
+		utils.Info("Mounting Object storage in ", s3MountPath)
 		if isEmpty, _ := utils.IsDirEmpty(s3MountPath); isEmpty {
 			cmd := exec.Command("s3fs", bucketName, s3MountPath,
 				"-o", "passwd_file="+s3fsPasswdFile,

pkg/scripts.go

@@ -30,12 +30,12 @@ func CreateCrontabScript(disableCompression bool, storage string) {
 	if storage == "s3" {
 		scriptContent = fmt.Sprintf(`#!/usr/bin/env bash
 set -e
-bkup --operation backup --dbname %s --port %s --storage s3 --path %s %v
+bkup backup --dbname %s --port %s --storage s3 --path %s %v
 `, os.Getenv("DB_NAME"), os.Getenv("DB_PORT"), os.Getenv("S3_PATH"), disableC)
 	} else {
 		scriptContent = fmt.Sprintf(`#!/usr/bin/env bash
 set -e
-bkup --operation backup --dbname %s --port %s %v
+bkup backup --dbname %s --port %s %v
 `, os.Getenv("DB_NAME"), os.Getenv("DB_PORT"), disableC)
 	}
 

pkg/var.go

@@ -0,0 +1,18 @@
+package pkg
+
+const s3MountPath string = "/s3mnt"
+const s3fsPasswdFile string = "/etc/passwd-s3fs"
+
+var (
+	storage            = "local"
+	file               = ""
+	s3Path             = "/pg-bkup"
+	dbPassword         = ""
+	dbUserName         = ""
+	dbName             = ""
+	dbHost             = ""
+	dbPort             = "5432"
+	executionMode      = "default"
+	storagePath        = "/backup"
+	disableCompression = false
+)