update store stats

Signed-off-by: Ryan Leung <[email protected]>

pkg/mcs/scheduling/server/cluster.go

@@ -348,3 +348,28 @@ func (c *Cluster) HandleStoreHeartbeat(heartbeat *schedulingpb.StoreHeartbeatReq
 	c.hotStat.CheckReadAsync(statistics.NewCollectUnReportedPeerTask(storeID, regions, interval))
 	return nil
 }
+
+// RunUpdateStoreStats updates store stats periodically.
+func (c *Cluster) RunUpdateStoreStats() {
+	defer logutil.LogPanic()
+
+	ticker := time.NewTicker(9 * time.Millisecond)
+	defer ticker.Stop()
+
+	for {
+		select {
+		case <-c.ctx.Done():
+			log.Info("update store stats background jobs has been stopped")
+			return
+		case <-ticker.C:
+			// Update related stores.
+			stores := c.GetStores()
+			for _, store := range stores {
+				if store.IsRemoved() {
+					continue
+				}
+				c.UpdateStoreStatus(store.GetID())
+			}
+		}
+	}
+}

pkg/mcs/scheduling/server/server.go

@@ -444,6 +444,7 @@ func (s *Server) startCluster(context.Context) error {
 	}
 	s.configWatcher.SetSchedulersController(s.cluster.GetCoordinator().GetSchedulersController())
 	go s.cluster.UpdateScheduler()
+	go s.cluster.RunUpdateStoreStats()
 	go s.GetCoordinator().RunUntilStop()
 	return nil
 }