From c605f38104c0e77089229e7b1797488cf7d2b48d Mon Sep 17 00:00:00 2001 From: Rohit Nayak Date: Mon, 9 Oct 2023 13:02:54 +0200 Subject: [PATCH] Cache workflow for Reshard as well since same workflow name is used in e2e tests for reshard and movetables Signed-off-by: Rohit Nayak --- go/vt/wrangler/resharder.go | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/go/vt/wrangler/resharder.go b/go/vt/wrangler/resharder.go index a81c3e8d598..49563374a3f 100644 --- a/go/vt/wrangler/resharder.go +++ b/go/vt/wrangler/resharder.go @@ -23,6 +23,8 @@ import ( "sync" "time" + "vitess.io/vitess/go/vt/proto/topodata" + "google.golang.org/protobuf/encoding/prototext" "vitess.io/vitess/go/vt/log" @@ -95,6 +97,17 @@ func (wr *Wrangler) Reshard(ctx context.Context, keyspace, workflow string, sour return vterrors.Wrap(err, "createStreams") } + wm := &topodata.WorkflowMetadata{ + Name: workflow, + Type: "Reshard", + TargetShards: targets, + SourceShards: sources, + } + log.Infof("Going to save workflow metadata for workflow %s: %+v", workflow, wm) + if err := wr.ts.SaveWorkflowMetadata(ctx, keyspace, wm); err != nil { + return err + } + if autoStart { if err := rs.startStreams(ctx); err != nil { return vterrors.Wrap(err, "startStreams")