From 531ffa0c88b1b069618a6ba0ea2cea6602af2d52 Mon Sep 17 00:00:00 2001
From: lhy1024 <admin@liudos.us>
Date: Tue, 20 Jun 2023 22:44:30 +0800
Subject: [PATCH] pd-ctl: fix hot region show

Signed-off-by: lhy1024 <admin@liudos.us>
---
 pkg/schedule/coordinator.go        | 27 ++++++++++++++++++++++-----
 pkg/statistics/hot_peer_cache.go   |  4 ++--
 pkg/statistics/hot_regions_stat.go | 17 ++---------------
 3 files changed, 26 insertions(+), 22 deletions(-)

diff --git a/pkg/schedule/coordinator.go b/pkg/schedule/coordinator.go
index 07e519dac7a..419b6a7adae 100644
--- a/pkg/schedule/coordinator.go
+++ b/pkg/schedule/coordinator.go
@@ -519,11 +519,28 @@ func (c *Coordinator) GetHotRegionsByType(typ statistics.RWType) *statistics.Sto
 	default:
 	}
 	// update params `IsLearner` and `LastUpdateTime`
-	for _, stores := range []statistics.StoreHotPeersStat{infos.AsLeader, infos.AsPeer} {
-		for _, store := range stores {
-			for _, hotPeer := range store.Stats {
-				region := c.cluster.GetRegion(hotPeer.RegionID)
-				hotPeer.UpdateHotPeerStatShow(region)
+	s := []statistics.StoreHotPeersStat{infos.AsLeader, infos.AsPeer}
+	for i, stores := range s {
+		for j, store := range stores {
+			for k := range store.Stats {
+				h := &s[i][j].Stats[k]
+				region := c.cluster.GetRegion(h.RegionID)
+				if region != nil {
+					h.IsLearner = core.IsLearner(region.GetPeer(h.StoreID))
+				}
+				switch typ {
+				case statistics.Write:
+					if region != nil {
+						h.LastUpdateTime = time.Unix(int64(region.GetInterval().GetEndTimestamp()), 0)
+					}
+				case statistics.Read:
+					store := c.cluster.GetStore(h.StoreID)
+					if store != nil {
+						ts := store.GetMeta().GetLastHeartbeat()
+						h.LastUpdateTime = time.Unix(ts/1e9, ts%1e9)
+					}
+				default:
+				}
 			}
 		}
 	}
diff --git a/pkg/statistics/hot_peer_cache.go b/pkg/statistics/hot_peer_cache.go
index 478a9f506d1..16c64b752e0 100644
--- a/pkg/statistics/hot_peer_cache.go
+++ b/pkg/statistics/hot_peer_cache.go
@@ -234,8 +234,8 @@ func (f *hotPeerCache) checkPeerFlow(peer *core.PeerInfo, region *core.RegionInf
 		actionType: Update,
 		stores:     make([]uint64, len(peers)),
 	}
-	for _, peer := range peers {
-		newItem.stores = append(newItem.stores, peer.GetStoreId())
+	for i, peer := range peers {
+		newItem.stores[i] = peer.GetStoreId()
 	}
 
 	if oldItem == nil {
diff --git a/pkg/statistics/hot_regions_stat.go b/pkg/statistics/hot_regions_stat.go
index d606a0d8bb4..d30a153492b 100644
--- a/pkg/statistics/hot_regions_stat.go
+++ b/pkg/statistics/hot_regions_stat.go
@@ -14,11 +14,7 @@
 
 package statistics
 
-import (
-	"time"
-
-	"github.com/tikv/pd/pkg/core"
-)
+import "time"
 
 // HotPeersStat records all hot regions statistics
 type HotPeersStat struct {
@@ -44,14 +40,5 @@ type HotPeerStatShow struct {
 	KeyRate        float64   `json:"flow_keys"`
 	QueryRate      float64   `json:"flow_query"`
 	AntiCount      int       `json:"anti_count"`
-	LastUpdateTime time.Time `json:"last_update_time"`
-}
-
-// UpdateHotPeerStatShow updates the region information, such as `IsLearner` and `LastUpdateTime`.
-func (h *HotPeerStatShow) UpdateHotPeerStatShow(region *core.RegionInfo) {
-	if region == nil {
-		return
-	}
-	h.IsLearner = core.IsLearner(region.GetPeer(h.StoreID))
-	h.LastUpdateTime = time.Unix(int64(region.GetInterval().GetEndTimestamp()), 0)
+	LastUpdateTime time.Time `json:"last_update_time,omitempty"`
 }