asim: also trace StoreRebalancer

tbg · tbg · commit 8cb593032fdd · 2025-10-01T11:53:37.000+02:00
diff --git a/pkg/kv/kvserver/asim/asim.go b/pkg/kv/kvserver/asim/asim.go
@@ -404,7 +404,9 @@ func (s *Simulator) tickStoreRebalancers(ctx context.Context, tick time.Time, st
 	stores := s.state.Stores()
 	s.shuffler(len(stores), func(i, j int) { stores[i], stores[j] = stores[j], stores[i] })
 	for _, store := range stores {
-		s.srs[store.StoreID()].Tick(ctx, tick, state)
+		s.doAndMaybeTrace(ctx, store.StoreID(), tick, "StoreRebalancer", func(ctx context.Context) {
+			s.srs[store.StoreID()].Tick(ctx, tick, state)
+		})
 	}
 }
 

Original file line number	Diff line number	Diff line change
`@@ -404,7 +404,9 @@ func (s *Simulator) tickStoreRebalancers(ctx context.Context, tick time.Time, st`
`404`	`404`	`stores := s.state.Stores()`
`405`	`405`	`s.shuffler(len(stores), func(i, j int) { stores[i], stores[j] = stores[j], stores[i] })`
`406`	`406`	`for _, store := range stores {`
`407`		`- s.srs[store.StoreID()].Tick(ctx, tick, state)`
	`407`	`+ s.doAndMaybeTrace(ctx, store.StoreID(), tick, "StoreRebalancer", func(ctx context.Context) {`
	`408`	`+ s.srs[store.StoreID()].Tick(ctx, tick, state)`
	`409`	`+ })`
`408`	`410`	`}`
`409`	`411`	`}`
`410`	`412`