From 6902b5681293b22558061fd7399e328c5fe49055 Mon Sep 17 00:00:00 2001 From: Ryan Leung Date: Mon, 29 Jan 2024 10:47:38 +0800 Subject: [PATCH] fix store statistics Signed-off-by: Ryan Leung --- tools/pd-heartbeat-bench/main.go | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/tools/pd-heartbeat-bench/main.go b/tools/pd-heartbeat-bench/main.go index 2a81c3e01df..52591b05770 100644 --- a/tools/pd-heartbeat-bench/main.go +++ b/tools/pd-heartbeat-bench/main.go @@ -401,7 +401,14 @@ func (s *Stores) update(rs *Regions) { }, } } - for _, region := range rs.regions { + var toUpdate []*pdpb.RegionHeartbeatRequest + updatedRegions := rs.awakenRegions.Load() + if updatedRegions == nil { + toUpdate = rs.regions + } else { + toUpdate = updatedRegions.([]*pdpb.RegionHeartbeatRequest) + } + for _, region := range toUpdate { for _, peer := range region.Region.Peers { store := stats[peer.StoreId] store.UsedSize += region.ApproximateSize