pass ut

Signed-off-by: 童剑 <[email protected]>
bufferflies · Jan 13, 2025 · 86ba3da · 86ba3da
1 parent d1da5b5
commit 86ba3da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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