fix hot store command (#1244)

tikv · Sep 12, 2018 · 4a87403 · 4a87403
1 parent 65dc1d1
commit 4a87403
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/server/handler.go b/server/handler.go
@@ -115,22 +115,38 @@ func (h *Handler) GetHotReadRegions() *core.StoreHotRegionInfos {
 
 // GetHotBytesWriteStores gets all hot write stores stats.
 func (h *Handler) GetHotBytesWriteStores() map[uint64]uint64 {
-	return h.s.cluster.cachedCluster.getStoresBytesWriteStat()
+	cluster := h.s.GetRaftCluster()
+	if cluster == nil {
+		return nil
+	}
+	return cluster.cachedCluster.getStoresBytesWriteStat()
 }
 
 // GetHotBytesReadStores gets all hot write stores stats.
 func (h *Handler) GetHotBytesReadStores() map[uint64]uint64 {
-	return h.s.cluster.cachedCluster.getStoresBytesReadStat()
+	cluster := h.s.GetRaftCluster()
+	if cluster == nil {
+		return nil
+	}
+	return cluster.cachedCluster.getStoresBytesReadStat()
 }
 
 // GetHotKeysWriteStores gets all hot write stores stats.
 func (h *Handler) GetHotKeysWriteStores() map[uint64]uint64 {
-	return h.s.cluster.cachedCluster.getStoresKeysWriteStat()
+	cluster := h.s.GetRaftCluster()
+	if cluster == nil {
+		return nil
+	}
+	return cluster.cachedCluster.getStoresKeysWriteStat()
 }
 
 // GetHotKeysReadStores gets all hot write stores stats.
 func (h *Handler) GetHotKeysReadStores() map[uint64]uint64 {
-	return h.s.cluster.cachedCluster.getStoresKeysReadStat()
+	cluster := h.s.GetRaftCluster()
+	if cluster == nil {
+		return nil
+	}
+	return cluster.cachedCluster.getStoresKeysReadStat()
 }
 
 // AddScheduler adds a scheduler.