pass ut

Signed-off-by: 童剑 <[email protected]>
tikv · Jan 10, 2025 · 1e6d628 · 1e6d628
1 parent 23ff7d0
commit 1e6d628
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 19 deletions.
diff --git a/pkg/mcs/scheduling/server/cluster.go b/pkg/mcs/scheduling/server/cluster.go
@@ -314,7 +314,10 @@ func (c *Cluster) updateScheduler() {
 		)
 		// Create the newly added schedulers.
 		for _, scheduler := range latestSchedulersConfig {
-			schedulerType := types.ConvertOldStrToType[scheduler.Type]
+			schedulerType, ok := types.ConvertOldStrToType[scheduler.Type]
+			if !ok {
+				log.Warn("scheduler not found  ", zap.String("type", scheduler.Type))
+			}
 			s, err := schedulers.CreateScheduler(
 				schedulerType,
 				c.coordinator.GetOperatorController(),

diff --git a/pkg/schedule/types/type.go b/pkg/schedule/types/type.go
@@ -106,24 +106,24 @@ var (
 	//
 	//	It is used to convert the old scheduler type to `CheckerSchedulerType`.
 	ConvertOldStrToType = map[string]CheckerSchedulerType{
-		"balance-leader":              BalanceLeaderScheduler,
-		"balance-region":              BalanceRegionScheduler,
-		"balance-witness":             BalanceWitnessScheduler,
-		"evict-leader":                EvictLeaderScheduler,
-		"evict-slow-store":            EvictSlowStoreScheduler,
-		"evict-slow-trend":            EvictSlowTrendScheduler,
-		"grant-leader":                GrantLeaderScheduler,
-		"grant-hot-region":            GrantHotRegionScheduler,
-		"hot-region":                  BalanceHotRegionScheduler,
-		"random-merge":                RandomMergeScheduler,
-		"scatter-range":               ScatterRangeScheduler,
-		"shuffle-hot-region":          ShuffleHotRegionScheduler,
-		"shuffle-leader":              ShuffleLeaderScheduler,
-		"shuffle-region":              ShuffleRegionScheduler,
-		"split-bucket":                SplitBucketScheduler,
-		"transfer-witness-leader":     TransferWitnessLeaderScheduler,
-		"label":                       LabelScheduler,
-		"balance-key-range-scheduler": BalanceKeyRangeScheduler,
+		"balance-leader":          BalanceLeaderScheduler,
+		"balance-region":          BalanceRegionScheduler,
+		"balance-witness":         BalanceWitnessScheduler,
+		"evict-leader":            EvictLeaderScheduler,
+		"evict-slow-store":        EvictSlowStoreScheduler,
+		"evict-slow-trend":        EvictSlowTrendScheduler,
+		"grant-leader":            GrantLeaderScheduler,
+		"grant-hot-region":        GrantHotRegionScheduler,
+		"hot-region":              BalanceHotRegionScheduler,
+		"random-merge":            RandomMergeScheduler,
+		"scatter-range":           ScatterRangeScheduler,
+		"shuffle-hot-region":      ShuffleHotRegionScheduler,
+		"shuffle-leader":          ShuffleLeaderScheduler,
+		"shuffle-region":          ShuffleRegionScheduler,
+		"split-bucket":            SplitBucketScheduler,
+		"transfer-witness-leader": TransferWitnessLeaderScheduler,
+		"label":                   LabelScheduler,
+		"balance-key-range":       BalanceKeyRangeScheduler,
 	}
 
 	// StringToSchedulerType is a map to convert the scheduler string to the CheckerSchedulerType.