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

This commit is contained in:
2024-01-20 13:04:39 +01:00
parent 4b07a78f29
commit 05eda4213a
22 changed files with 327 additions and 283 deletions

15
utils/constant.go Normal file
View File

@@ -0,0 +1,15 @@
package utils
const Notice = "Please remove --operation flag.\n" +
"Use: \n" +
"- backup for backup operation [eg: bkup backup -d database_name ...]\n" +
"- restore for restore operation [eg. bkup restore -d database_name ...]\n" +
"We are sorry for this inconvenient"
const RestoreExample = "mysql-bkup restore --dbname database --file db_20231219_022941.sql.gz\n" +
"bkup restore --dbname database --storage s3 --path /custom-path --file db_20231219_022941.sql.gz"
const BackupExample = "mysql-bkup backup --dbname database --disable-compression\n" +
"mysql-bkup backup --dbname database --storage s3 --path /custom-path --disable-compression"
const MainExample = "mysql-bkup backup --dbname database --disable-compression\n" +
"mysql-bkup backup --dbname database --storage s3 --path /custom-path\n" +
"mysql-bkup restore --dbname database --file db_20231219_022941.sql.gz"

View File

@@ -8,6 +8,7 @@ package utils
**/
import (
"fmt"
"github.com/spf13/cobra"
"io/fs"
"os"
"os/exec"
@@ -78,3 +79,36 @@ func TestDatabaseConnection() {
}
}
func GetEnv(cmd *cobra.Command, flagName, envName string) string {
value, _ := cmd.Flags().GetString(flagName)
if value != "" {
err := os.Setenv(envName, value)
if err != nil {
return value
}
}
return os.Getenv(envName)
}
func FlagGetString(cmd *cobra.Command, flagName string) string {
value, _ := cmd.Flags().GetString(flagName)
if value != "" {
return value
}
return ""
}
func FlagGetBool(cmd *cobra.Command, flagName string) bool {
value, _ := cmd.Flags().GetBool(flagName)
return value
}
func SetEnv(key, value string) {
err := os.Setenv(key, value)
if err != nil {
return
}
}
func ShowHistory() {
}