forked from RoganDawes/P4wnP1_aloa
-
Notifications
You must be signed in to change notification settings - Fork 0
/
cmd_db.go
99 lines (81 loc) · 2.07 KB
/
cmd_db.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package cli_client
import (
"encoding/json"
"fmt"
"github.com/spf13/cobra"
"os"
)
var (
// deploy
tmpDBBackupName = ""
)
func init() {
cmdDB := &cobra.Command{
Use: "db",
Short: "Database backup and restore",
}
cmdDBBackup := &cobra.Command{
Use: "backup",
Short: "Backup DB",
Run: func(cmd *cobra.Command, args []string) {
if len(tmpDBBackupName) == 0 {
fmt.Println("A name for the backup has to be provided with the '--name' flag")
os.Exit(-1)
}
fmt.Print("Creating backup ...")
err := ClientDBBackup(TIMEOUT_LONG, StrRemoteHost, StrRemotePort, tmpDBBackupName)
if err != nil {
fmt.Println(" failed")
fmt.Println(err.Error())
os.Exit(-1)
}
fmt.Println(" success")
},
}
cmdDBRestore := &cobra.Command{
Use: "restore",
Short: "Restore DB",
Run: func(cmd *cobra.Command, args []string) {
if len(tmpDBBackupName) == 0 {
fmt.Println("A name for the backup has to be provided with the '--name' flag")
os.Exit(-1)
}
fmt.Print("Restoring ...")
err := ClientDBRestore(TIMEOUT_LONG, StrRemoteHost, StrRemotePort, tmpDBBackupName)
if err != nil {
fmt.Println(" failed")
fmt.Println(err.Error())
os.Exit(-1)
}
fmt.Println(" success")
},
}
cmdDBList := &cobra.Command{
Use: "list",
Short: "List backups",
Long: ``,
Run: func(cmd *cobra.Command, args []string) {
list,err := ClientDBList(TIMEOUT_LONG, StrRemoteHost, StrRemotePort)
if err != nil {
fmt.Println(err.Error())
os.Exit(-1)
}
if BoolJson {
b, err := json.Marshal(list)
if err == nil {
fmt.Println(string(b))
}
} else {
fmt.Println("Database backups:")
for _, item := range list {
fmt.Println(item)
}
}
},
}
rootCmd.AddCommand(cmdDB)
cmdDB.AddCommand(cmdDBBackup, cmdDBList, cmdDBRestore)
cmdDBBackup.Flags().StringVarP(&tmpDBBackupName, "name", "n", "","Name of backup")
cmdDBRestore.Flags().StringVarP(&tmpDBBackupName, "name", "n", "","Name of backup")
cmdDBList.Flags().BoolVar(&BoolJson, "json", false, "Output results as JSON if applicable")
}