Match shard names instead so as to e.g. support "0"

Signed-off-by: Matt Lord <mattalord@gmail.com>
vitessio · Aug 27, 2024 · 747d454 · 747d454
1 parent b802c14
commit 747d454
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/go/vt/vtctl/workflow/server.go b/go/vt/vtctl/workflow/server.go
@@ -1927,9 +1927,9 @@ func (s *Server) VDiffResume(ctx context.Context, req *vtctldatapb.VDiffResumeRe
 	}
 
 	if len(targetShards) > 0 {
-		for k, target := range ts.targets {
-			if !slices.Contains(targetShards, key.KeyRangeString(target.GetShard().GetKeyRange())) {
-				delete(ts.targets, k)
+		for key, target := range ts.targets {
+			if !slices.Contains(targetShards, target.GetShard().ShardName()) {
+				delete(ts.targets, key)
 			}
 		}
 	}
@@ -2012,9 +2012,9 @@ func (s *Server) VDiffStop(ctx context.Context, req *vtctldatapb.VDiffStopReques
 	}
 
 	if len(targetShards) > 0 {
-		for k, target := range ts.targets {
-			if !slices.Contains(targetShards, key.KeyRangeString(target.GetShard().GetKeyRange())) {
-				delete(ts.targets, k)
+		for key, target := range ts.targets {
+			if !slices.Contains(targetShards, target.GetShard().ShardName()) {
+				delete(ts.targets, key)
 			}
 		}
 	}