statistics: support hotPeerCache maintained in Peer level

server/core/peer.go

@@ -72,3 +72,75 @@ func CountInJointState(peers ...*metapb.Peer) int {
 	}
 	return count
 }
+
+// PeerInfo provides peer information
+type PeerInfo struct {
+	peerID       uint64
+	StoreID      uint64
+	writtenBytes uint64
+	writtenKeys  uint64
+	readBytes    uint64
+	readKeys     uint64
+}
+
+// GetKeysWritten provides peer written keys
+func (p *PeerInfo) GetKeysWritten() uint64 {
+	return p.writtenKeys
+}
+
+// GetBytesWritten provides peer written bytes
+func (p *PeerInfo) GetBytesWritten() uint64 {
+	return p.writtenBytes
+}
+
+// GetBytesRead provides peer read bytes
+func (p *PeerInfo) GetBytesRead() uint64 {
+	return p.readBytes
+}
+
+// GetKeysRead provides read keys
+func (p *PeerInfo) GetKeysRead() uint64 {
+	return p.readKeys
+}
+
+// GetStoreID provides located storeID
+func (p *PeerInfo) GetStoreID() uint64 {
+	return p.StoreID
+}
+
+// GetPeerID provides peer id
+func (p *PeerInfo) GetPeerID() uint64 {
+	return p.peerID
+}
+
+// FromMetaPeer provides PeerInfo from metapb.Peer
+func FromMetaPeer(peer *metapb.Peer) *PeerInfo {
+	return &PeerInfo{
+		peerID:  peer.GetId(),
+		StoreID: peer.GetStoreId(),
+	}
+}
+
+// SetBytesRead sets read bytes
+func (p *PeerInfo) SetBytesRead(b uint64) *PeerInfo {
+	p.readBytes = b
+	return p
+}
+
+// SetKeysRead sets read keys
+func (p *PeerInfo) SetKeysRead(k uint64) *PeerInfo {
+	p.readKeys = k
+	return p
+}
+
+// SetBytesWrite sets write bytes
+func (p *PeerInfo) SetBytesWrite(b uint64) *PeerInfo {
+	p.writtenBytes = b
+	return p
+}
+
+// SetKeysWrite sets write keys
+func (p *PeerInfo) SetKeysWrite(k uint64) *PeerInfo {
+	p.writtenKeys = k
+	return p
+}

server/statistics/hot_peer.go

@@ -88,9 +88,10 @@ type HotPeerStat struct {
 	// LastUpdateTime used to calculate average write
 	LastUpdateTime time.Time `json:"last_update_time"`
 
-	needDelete             bool
-	isLeader               bool
-	isNew                  bool
+	needDelete bool
+	isLeader   bool
+	isNew      bool
+	//TODO: remove it when we send peer stat by store info
 	justTransferLeader     bool
 	interval               uint64
 	thresholds             []float64
@@ -128,7 +129,7 @@ func (stat *HotPeerStat) Log(str string, level func(msg string, fields ...zap.Fi
 
 // IsNeedCoolDownTransferLeader use cooldown time after transfer leader to avoid unnecessary schedule
 func (stat *HotPeerStat) IsNeedCoolDownTransferLeader(minHotDegree int) bool {
-	return time.Since(stat.lastTransferLeaderTime).Seconds() < float64(minHotDegree*RegionHeartBeatReportInterval)
+	return time.Since(stat.lastTransferLeaderTime).Seconds() < float64(minHotDegree*HotStatReportInterval)
 }
 
 // IsNeedDelete to delete the item in cache.