pass cells from backend in refreshTabletsInKeyspaceShard

Signed-off-by: Tim Vaillancourt <[email protected]>
timvaillancourt · Feb 12, 2025 · 6b77931 · 6b77931
1 parent 81cf199
commit 6b77931
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/go/vt/vtorc/logic/tablet_discovery.go b/go/vt/vtorc/logic/tablet_discovery.go
@@ -237,7 +237,12 @@ func refreshTabletInfoOfShard(ctx context.Context, keyspace, shard string) {
 }
 
 func refreshTabletsInKeyspaceShard(ctx context.Context, keyspace, shard string, loader func(tabletAlias string), forceRefresh bool, tabletsToIgnore []string) {
-	tablets, err := ts.GetTabletsByShard(ctx, keyspace, shard)
+	cells, err := inst.ReadCells()
+	if err != nil {
+		log.Errorf("Error fetching cells: %v", err)
+		return
+	}
+	tablets, err := ts.GetTabletsByShardCell(ctx, keyspace, shard, cells)
 	if err != nil {
 		log.Errorf("Error fetching tablets for keyspace/shard %v/%v: %v", keyspace, shard, err)
 		return