fix usage

Signed-off-by: HunDunDM <[email protected]>
tikv · Jul 26, 2021 · c82c23f · c82c23f
1 parent 0f8a97e
commit c82c23f
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/server/schedulers/shuffle_hot_region.go b/server/schedulers/shuffle_hot_region.go
@@ -132,22 +132,20 @@ func (s *shuffleHotRegionScheduler) Schedule(cluster opt.Cluster) []*operator.Op
 }
 
 func (s *shuffleHotRegionScheduler) dispatch(typ rwType, cluster opt.Cluster) []*operator.Operator {
-	stores := cluster.GetStores()
+	storeInfos := summaryStoreInfos(cluster)
 	storesLoads := cluster.GetStoresLoads()
 	switch typ {
 	case read:
 		s.stLoadInfos[readLeader] = summaryStoresLoad(
-			stores,
+			storeInfos,
 			storesLoads,
-			map[uint64]*Influence{},
 			cluster.RegionReadStats(),
 			read, core.LeaderKind)
 		return s.randomSchedule(cluster, s.stLoadInfos[readLeader])
 	case write:
 		s.stLoadInfos[writeLeader] = summaryStoresLoad(
-			stores,
+			storeInfos,
 			storesLoads,
-			map[uint64]*Influence{},
 			cluster.RegionWriteStats(),
 			write, core.LeaderKind)
 		return s.randomSchedule(cluster, s.stLoadInfos[writeLeader])