Vtctldserver backup

go/cmd/vtctldclient/command/backups.go

@@ -18,6 +18,7 @@ package command
 
 import (
 	"fmt"
+	"io"
 	"strings"
 
 	"github.com/spf13/cobra"
@@ -28,11 +29,56 @@ import (
 	vtctldatapb "vitess.io/vitess/go/vt/proto/vtctldata"
 )
 
-// GetBackups makes a GetBackups gRPC call to a vtctld.
-var GetBackups = &cobra.Command{
-	Use:  "GetBackups <keyspace/shard>",
-	Args: cobra.ExactArgs(1),
-	RunE: commandGetBackups,
+var (
+	// Backup makes a Backup gRPC call to a vtctld.
+	Backup = &cobra.Command{
+		Use:                   "Backup [--concurrency <concurrency>] [--allow-primary] <tablet_alias>",
+		Short:                 "Uses the BackupStorage service on the given tablet to create and store a new backup.",
+		DisableFlagsInUseLine: true,
+		Args:                  cobra.ExactArgs(1),
+		RunE:                  commandBackup,
+	}
+	// GetBackups makes a GetBackups gRPC call to a vtctld.
+	GetBackups = &cobra.Command{
+		Use:  "GetBackups <keyspace/shard>",
+		Args: cobra.ExactArgs(1),
+		RunE: commandGetBackups,
+	}
+)
+
+var backupOptions = struct {
+	AllowPrimary bool
+	Concurrency  uint64
+}{}
+
+func commandBackup(cmd *cobra.Command, args []string) error {
+	tabletAlias, err := topoproto.ParseTabletAlias(cmd.Flags().Arg(0))
+	if err != nil {
+		return err
+	}
+
+	cli.FinishedParsing(cmd)
+
+	stream, err := client.Backup(commandCtx, &vtctldatapb.BackupRequest{
+		TabletAlias:  tabletAlias,
+		AllowPrimary: backupOptions.AllowPrimary,
+		Concurrency:  backupOptions.Concurrency,
+	})
+	if err != nil {
+		return err
+	}
+
+	for {
+		resp, err := stream.Recv()
+		switch err {
+		case nil:
+			fmt.Printf("%s/%s (%s): %v\n", resp.Keyspace, resp.Shard, topoproto.TabletAliasString(resp.TabletAlias), resp.Event)
+		case io.EOF:
+			return nil
+		default:
+			return err
+		}
+	}
 }
 
 var getBackupsOptions = struct {
@@ -78,6 +124,10 @@ func commandGetBackups(cmd *cobra.Command, args []string) error {
 }
 
 func init() {
+	Backup.Flags().BoolVar(&backupOptions.AllowPrimary, "allow-primary", false, "Allow the primary of a shard to be used for the backup. WARNING: If using the builtin backup engine, this will shutdown mysqld on the primary and stop writes for the duration of the backup.")
+	Backup.Flags().Uint64Var(&backupOptions.Concurrency, "concurrency", 4, "Specifies the number of compression/checksum jobs to run simultaneously.")
+	Root.AddCommand(Backup)
+
 	GetBackups.Flags().Uint32VarP(&getBackupsOptions.Limit, "limit", "l", 0, "Retrieve only the most recent N backups")
 	GetBackups.Flags().BoolVarP(&getBackupsOptions.OutputJSON, "json", "j", false, "Output backup info in JSON format rather than a list of backups")
 	Root.AddCommand(GetBackups)