forked from flanksource/karina
/
consul.go
64 lines (54 loc) · 1.94 KB
/
consul.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
package cmd
import (
"github.com/moshloop/platform-cli/pkg/phases/consul"
log "github.com/sirupsen/logrus"
"github.com/spf13/cobra"
)
var Consul = &cobra.Command{
Use: "consul",
}
func init() {
backupCmd := &cobra.Command{
Use: "backup",
Short: "Create a new consul backup",
Args: cobra.MinimumNArgs(0),
Run: func(cmd *cobra.Command, args []string) {
name, _ := cmd.Flags().GetString("name")
namespace, _ := cmd.Flags().GetString("namespace")
schedule, _ := cmd.Flags().GetString("schedule")
cs := consul.NewBackupRestore(getPlatform(cmd), name, namespace)
if schedule != "" {
log.Infof("Creating consul backup schedule: %s: %s", schedule, cs.Name)
if err := cs.ScheduleBackup(schedule); err != nil {
log.Fatalf("Failed to create backup schedule: %v", err)
}
} else {
log.Infof("Backing up consul %s", cs.Name)
if err := cs.Backup(); err != nil {
log.Fatalf("Error backing up consul %s: %v\n", cs.Name, err)
}
}
},
}
backupCmd.Flags().String("name", "", "Name of the consul deployment")
backupCmd.Flags().String("namespace", "", "Namespace where consul runs")
backupCmd.Flags().String("schedule", "", "A cron schedule to backup on a recurring basis")
restoreCmd := &cobra.Command{
Use: "restore [backup path]",
Short: "Restore consul from backups",
Args: cobra.MinimumNArgs(1),
Run: func(cmd *cobra.Command, args []string) {
name, _ := cmd.Flags().GetString("name")
namespace, _ := cmd.Flags().GetString("namespace")
cs := consul.NewBackupRestore(getPlatform(cmd), name, namespace)
log.Infof("Restoring consul %s from %s", cs.Name, args[0])
if err := cs.Restore(args[0]); err != nil {
log.Fatalf("Error Restore up consul %s: %v\n", cs.Name, err)
}
},
}
restoreCmd.Flags().String("name", "", "Name of the consul deployment")
restoreCmd.Flags().String("namespace", "", "Namespace where consul runs")
Consul.AddCommand(backupCmd)
Consul.AddCommand(restoreCmd)
}