diff --git a/pkg/schedule/schedulers/balance_key_range.go b/pkg/schedule/schedulers/balance_key_range.go index aace3cc057bb..7bb73a81117f 100644 --- a/pkg/schedule/schedulers/balance_key_range.go +++ b/pkg/schedule/schedulers/balance_key_range.go @@ -17,6 +17,7 @@ import ( "github.com/tikv/pd/pkg/schedule/operator" "github.com/tikv/pd/pkg/schedule/plan" "github.com/tikv/pd/pkg/schedule/types" + "github.com/tikv/pd/pkg/utils/syncutil" ) const ( @@ -52,7 +53,8 @@ func (handler *balanceKeyRangeSchedulerHandler) listConfig(w http.ResponseWriter } type balanceKeyRangeSchedulerConfig struct { - baseDefaultSchedulerConfig + syncutil.RWMutex + schedulerConfig balanceKeyRangeSchedulerParam } diff --git a/pkg/schedule/schedulers/init.go b/pkg/schedule/schedulers/init.go index 4734b162203f..d0ea4e75c297 100644 --- a/pkg/schedule/schedulers/init.go +++ b/pkg/schedule/schedulers/init.go @@ -581,7 +581,7 @@ func schedulersRegister() { RegisterScheduler(types.BalanceKeyRangeScheduler, func(opController *operator.Controller, storage endpoint.ConfigStorage, decoder ConfigDecoder, _ ...func(string) error) (Scheduler, error) { conf := &balanceKeyRangeSchedulerConfig{ - baseDefaultSchedulerConfig: newBaseDefaultSchedulerConfig(), + schedulerConfig: newBaseDefaultSchedulerConfig(), } if err := decoder(conf); err != nil { return nil, err