prevent suddenly scheduling

Signed-off-by: Ryan Leung <rleungx@gmail.com>
tikv · Jan 16, 2024 · b21a599 · b21a599
1 parent 2a4dbc0
commit b21a599
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/schedule/prepare_checker.go b/pkg/schedule/prepare_checker.go
@@ -60,6 +60,12 @@ func (checker *prepareChecker) check(c *core.BasicCluster, collectWaitTime ...ti
 			continue
 		}
 		storeID := store.GetID()
+		// It is used to avoid sudden scheduling when scheduling service is just started.
+		if len(collectWaitTime) > 0 {
+			if float64(store.GetRegionCount())*collectFactor > float64(c.GetNotFromStorageRegionsCntByStore(storeID)) {
+				return false
+			}
+		}
 		// For each store, the number of active regions should be more than total region of the store * collectFactor
 		if float64(c.GetStoreRegionCount(storeID))*collectFactor > float64(c.GetNotFromStorageRegionsCntByStore(storeID)) {
 			return false