set concurrency

Signed-off-by: Tim Vaillancourt <[email protected]>
vitessio · Oct 25, 2024 · d237ef0 · d237ef0
1 parent bce3640
commit d237ef0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/go/vt/topo/shard.go b/go/vt/topo/shard.go
@@ -646,11 +646,12 @@ func (ts *Server) GetTabletsByShardCell(ctx context.Context, keyspace, shard str
 	mutex := sync.Mutex{}
 	rec := concurrency.AllErrorRecorder{}
 	tablets := make([]*TabletInfo, 0)
+	concurrency := DefaultConcurrency / len(cells)
 	for _, cell := range cells {
 		wg.Add(1)
 		go func() {
 			t, err := ts.GetTabletsByCell(ctx, cell, &GetTabletsByCellOptions{
-				Concurrency: 1,
+				Concurrency: concurrency,
 				Keyspace:    keyspace,
 				Shard:       shard,
 			})

diff --git a/go/vt/vtorc/logic/tablet_discovery.go b/go/vt/vtorc/logic/tablet_discovery.go
@@ -155,7 +155,7 @@ func refreshTabletsUsing(loader func(tabletAlias string), forceRefresh bool) {
 }
 
 func refreshTabletsInCell(ctx context.Context, cell string, loader func(tabletAlias string), forceRefresh bool) {
-	tablets, err := ts.GetTabletsByCell(ctx, cell, &topo.GetTabletsByCellOptions{Concurrency: 1})
+	tablets, err := ts.GetTabletsByCell(ctx, cell, &topo.GetTabletsByCellOptions{Concurrency: topo.DefaultConcurrency})
 	if err != nil {
 		log.Errorf("Error fetching topo info for cell %v: %v", cell, err)
 		return