VReplication: Add traffic state to vtctldclient workflow status output (

#14280) Signed-off-by: Matt Lord <mattalord@gmail.com>
vitessio · Oct 15, 2023 · c5d149d · c5d149d
1 parent fe31219
commit c5d149d
Show file tree

Hide file tree

Showing 10 changed files with 233 additions and 139 deletions.
diff --git a/go/cmd/vtctldclient/command/vreplication/common/status.go b/go/cmd/vtctldclient/command/vreplication/common/status.go
@@ -40,6 +40,10 @@ func GetStatusCommand(opts *SubCommandsOpts) *cobra.Command {
 }
 
 func commandStatus(cmd *cobra.Command, args []string) error {
+	format, err := GetOutputFormat(cmd)
+	if err != nil {
+		return err
+	}
 	cli.FinishedParsing(cmd)
 
 	req := &vtctldatapb.WorkflowStatusRequest{
@@ -51,7 +55,7 @@ func commandStatus(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
-	if err = OutputStatusResponse(resp, "json"); err != nil {
+	if err = OutputStatusResponse(resp, format); err != nil {
 		return err
 	}
 

diff --git a/go/cmd/vtctldclient/command/vreplication/common/utils.go b/go/cmd/vtctldclient/command/vreplication/common/utils.go
@@ -165,9 +165,11 @@ func OutputStatusResponse(resp *vtctldatapb.WorkflowStatusResponse, format strin
 					shardstream.Id, BaseOptions.TargetKeyspace, tablet, shardstream.Status, shardstream.Info))
 			}
 		}
+		tout.WriteString("\nTraffic State: ")
+		tout.WriteString(resp.TrafficState)
 		output = tout.Bytes()
 	}
-	fmt.Printf("%s\n", output)
+	fmt.Println(string(output))
 	return nil
 }
 

diff --git a/go/test/endtoend/vreplication/migrate_test.go b/go/test/endtoend/vreplication/migrate_test.go
@@ -221,17 +221,17 @@ func TestVtctldMigrate(t *testing.T) {
 	var output, expected string
 
 	t.Run("mount external cluster", func(t *testing.T) {
-		output, err := vc.VtctldClient.ExecuteCommandWithOutput("Mount", "Register", "--topo-type=etcd2",
+		output, err := vc.VtctldClient.ExecuteCommandWithOutput("Mount", "register", "--topo-type=etcd2",
 			fmt.Sprintf("--topo-server=localhost:%d", extVc.ClusterConfig.topoPort), "--topo-root=/vitess/global", "ext1")
 		require.NoError(t, err, "Mount Register command failed with %s", output)
 
-		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Mount", "List")
+		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Mount", "list")
 		require.NoError(t, err, "Mount List command failed with %s", output)
 
 		names := gjson.Get(output, "names")
 		require.Equal(t, 1, len(names.Array()))
 		require.Equal(t, "ext1", names.Array()[0].String())
-		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Mount", "Show", "ext1")
+		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Mount", "show", "ext1")
 		require.NoError(t, err, "Mount command failed with %s\n", output)
 
 		require.Equal(t, "etcd2", gjson.Get(output, "topo_type").String())
@@ -244,7 +244,7 @@ func TestVtctldMigrate(t *testing.T) {
 	t.Run("migrate from external cluster", func(t *testing.T) {
 		if output, err = vc.VtctldClient.ExecuteCommandWithOutput("Migrate",
 			"--target-keyspace", "product", "--workflow", "e1",
-			"Create", "--source-keyspace", "rating", "--mount-name", "ext1", "--all-tables", "--cells=extcell1", "--tablet-types=primary,replica"); err != nil {
+			"create", "--source-keyspace", "rating", "--mount-name", "ext1", "--all-tables", "--cells=extcell1", "--tablet-types=primary,replica"); err != nil {
 			t.Fatalf("Migrate command failed with %+v : %s\n", err, output)
 		}
 		waitForWorkflowState(t, vc, ksWorkflow, binlogdatapb.VReplicationWorkflowState_Running.String())
@@ -258,21 +258,21 @@ func TestVtctldMigrate(t *testing.T) {
 		vdiffSideBySide(t, ksWorkflow, "extcell1")
 
 		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Migrate",
-			"--target-keyspace", "product", "--workflow", "e1", "Show")
+			"--target-keyspace", "product", "--workflow", "e1", "show")
 		require.NoError(t, err, "Migrate command failed with %s", output)
 
 		wf := gjson.Get(output, "workflows").Array()[0]
 		require.Equal(t, "e1", wf.Get("name").String())
 		require.Equal(t, "Migrate", wf.Get("workflow_type").String())
 
 		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Migrate",
-			"--target-keyspace", "product", "--workflow", "e1", "Progress")
+			"--target-keyspace", "product", "--workflow", "e1", "status", "--format=json")
 		require.NoError(t, err, "Migrate command failed with %s", output)
 
 		require.Equal(t, "Running", gjson.Get(output, "shard_streams.product/0.streams.0.status").String())
 
 		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Migrate",
-			"--target-keyspace", "product", "--workflow", "e1", "Complete")
+			"--target-keyspace", "product", "--workflow", "e1", "complete")
 		require.NoError(t, err, "Migrate command failed with %s", output)
 
 		expectNumberOfStreams(t, vtgateConn, "migrate", "e1", "product:0", 0)
@@ -288,7 +288,7 @@ func TestVtctldMigrate(t *testing.T) {
 		waitForRowCount(t, vtgateConn, "product:0", "rating", 0)
 		waitForRowCount(t, vtgateConn, "product:0", "review", 0)
 		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Migrate",
-			"--target-keyspace", "product", "--workflow", "e1", "Cancel")
+			"--target-keyspace", "product", "--workflow", "e1", "cancel")
 		require.NoError(t, err, "Migrate command failed with %s", output)
 
 		expectNumberOfStreams(t, vtgateConn, "migrate", "e1", "product:0", 0)
@@ -302,15 +302,15 @@ func TestVtctldMigrate(t *testing.T) {
 	})
 
 	t.Run("unmount external cluster", func(t *testing.T) {
-		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Mount", "Unregister", "ext1")
+		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Mount", "unregister", "ext1")
 		require.NoError(t, err, "Mount command failed with %s\n", output)
 
-		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Mount", "List")
+		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Mount", "list")
 		require.NoError(t, err, "Mount command failed with %+v : %s\n", output)
 		expected = "{}\n"
 		require.Equal(t, expected, output)
 
-		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Mount", "Show", "ext1")
+		output, err = vc.VtctldClient.ExecuteCommandWithOutput("Mount", "show", "ext1")
 		require.Errorf(t, err, "there is no vitess cluster named ext1")
 	})
 }