scheduler: use right check for evict/grant leader scheduler (tikv#8758)…

… (tikv#8778) ref tikv#8756 Signed-off-by: ti-chi-bot <ti-community-prow-bot@tidb.io> Signed-off-by: okJiang <819421878@qq.com> Co-authored-by: okJiang <819421878@qq.com>
ti-chi-bot · Nov 7, 2024 · 7fe011a · 7fe011a
1 parent 959b578
commit 7fe011a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/plugin/scheduler_example/evict_leader.go b/plugin/scheduler_example/evict_leader.go
@@ -46,7 +46,7 @@ const (
 func init() {
 	schedule.RegisterSliceDecoderBuilder(EvictLeaderType, func(args []string) schedule.ConfigDecoder {
 		return func(v interface{}) error {
-			if len(args) != 1 {
+			if len(args) < 1 {
 				return errors.New("should specify the store-id")
 			}
 			conf, ok := v.(*evictLeaderSchedulerConfig)
@@ -98,7 +98,7 @@ type evictLeaderSchedulerConfig struct {
 }
 
 func (conf *evictLeaderSchedulerConfig) BuildWithArgs(args []string) error {
-	if len(args) != 1 {
+	if len(args) < 1 {
 		return errors.New("should specify the store-id")
 	}
 

diff --git a/server/schedulers/evict_leader.go b/server/schedulers/evict_leader.go
@@ -98,7 +98,7 @@ func (conf *evictLeaderSchedulerConfig) getStores() []uint64 {
 }
 
 func (conf *evictLeaderSchedulerConfig) BuildWithArgs(args []string) error {
-	if len(args) != 1 {
+	if len(args) < 1 {
 		return errs.ErrSchedulerConfig.FastGenByArgs("id")
 	}
 

diff --git a/server/schedulers/grant_leader.go b/server/schedulers/grant_leader.go
@@ -83,7 +83,7 @@ type grantLeaderSchedulerConfig struct {
 }
 
 func (conf *grantLeaderSchedulerConfig) BuildWithArgs(args []string) error {
-	if len(args) != 1 {
+	if len(args) < 1 {
 		return errs.ErrSchedulerConfig.FastGenByArgs("id")
 	}