Etcd Backup/restore

cmd/etcd.go

@@ -0,0 +1,136 @@
+package cmd
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/rancher/rke/cluster"
+	"github.com/rancher/rke/hosts"
+	"github.com/rancher/rke/log"
+	"github.com/rancher/rke/pki"
+	"github.com/rancher/types/apis/management.cattle.io/v3"
+	"github.com/urfave/cli"
+)
+
+func EtcdCommand() cli.Command {
+	backupRestoreFlags := []cli.Flag{
+		cli.StringFlag{
+			Name:  "name",
+			Usage: "Specify Backup name",
+		},
+		cli.StringFlag{
+			Name:   "config",
+			Usage:  "Specify an alternate cluster YAML file",
+			Value:  pki.ClusterConfig,
+			EnvVar: "RKE_CONFIG",
+		},
+	}
+
+	backupRestoreFlags = append(backupRestoreFlags, sshCliOptions...)
+
+	return cli.Command{
+		Name:  "etcd",
+		Usage: "etcd backup/restore operations in k8s cluster",
+		Subcommands: []cli.Command{
+			{
+				Name:   "backup",
+				Usage:  "Take backup on all etcd hosts",
+				Flags:  backupRestoreFlags,
+				Action: BackupEtcdHostsFromCli,
+			},
+			{
+				Name:   "restore",
+				Usage:  "Restore existing backup",
+				Flags:  backupRestoreFlags,
+				Action: RestoreEtcdBackupFromCli,
+			},
+		},
+	}
+}
+
+func BackupEtcdHosts(
+	ctx context.Context,
+	rkeConfig *v3.RancherKubernetesEngineConfig,
+	dockerDialerFactory hosts.DialerFactory,
+	configDir, backupName string) error {
+
+	log.Infof(ctx, "Starting Backup on etcd hosts")
+	kubeCluster, err := cluster.ParseCluster(ctx, rkeConfig, clusterFilePath, configDir, dockerDialerFactory, nil, nil)
+	if err != nil {
+		return err
+	}
+
+	if err := kubeCluster.TunnelHosts(ctx, false); err != nil {
+		return err
+	}
+	if err := kubeCluster.BackupEtcd(ctx, backupName); err != nil {
+		return err
+	}
+
+	log.Infof(ctx, "Finished backup on all etcd hosts")
+	return nil
+}
+
+func RestoreEtcdBackup(
+	ctx context.Context,
+	rkeConfig *v3.RancherKubernetesEngineConfig,
+	dockerDialerFactory hosts.DialerFactory,
+	configDir, backupName string) error {
+
+	log.Infof(ctx, "Starting restore on etcd hosts")
+	kubeCluster, err := cluster.ParseCluster(ctx, rkeConfig, clusterFilePath, configDir, dockerDialerFactory, nil, nil)
+	if err != nil {
+		return err
+	}
+
+	if err := kubeCluster.TunnelHosts(ctx, false); err != nil {
+		return err
+	}
+	if err := kubeCluster.RestoreEtcdBackup(ctx, backupName); err != nil {
+		return err
+	}
+
+	log.Infof(ctx, "Finished restoring on all etcd hosts")
+	return nil
+}
+
+func BackupEtcdHostsFromCli(ctx *cli.Context) error {
+	clusterFile, filePath, err := resolveClusterFile(ctx)
+	if err != nil {
+		return fmt.Errorf("Failed to resolve cluster file: %v", err)
+	}
+	clusterFilePath = filePath
+
+	rkeConfig, err := cluster.ParseConfig(clusterFile)
+	if err != nil {
+		return fmt.Errorf("Failed to parse cluster file: %v", err)
+	}
+
+	rkeConfig, err = setOptionsFromCLI(ctx, rkeConfig)
+	if err != nil {
+		return err
+	}
+
+	return BackupEtcdHosts(context.Background(), rkeConfig, nil, "", ctx.String("name"))
+}
+
+func RestoreEtcdBackupFromCli(ctx *cli.Context) error {
+	clusterFile, filePath, err := resolveClusterFile(ctx)
+	if err != nil {
+		return fmt.Errorf("Failed to resolve cluster file: %v", err)
+	}
+	clusterFilePath = filePath
+
+	rkeConfig, err := cluster.ParseConfig(clusterFile)
+	if err != nil {
+		return fmt.Errorf("Failed to parse cluster file: %v", err)
+	}
+
+	rkeConfig, err = setOptionsFromCLI(ctx, rkeConfig)
+	if err != nil {
+		return err
+	}
+
+	return RestoreEtcdBackup(context.Background(), rkeConfig, nil, "", ctx.String("name"))
+
+}