fix ci

pingcap · Nov 10, 2020 · 7cf0c64 · 7cf0c64
1 parent 6f141e6
commit 7cf0c64
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pkg/pdutil/pd.go b/pkg/pdutil/pd.go
@@ -509,12 +509,12 @@ func (p *PdController) RemoveSchedulers(ctx context.Context) (undo utils.UndoFun
 			// Ignore non-exist config.
 			continue
 		}
-		switch cfgVal.(type) {
+		switch v := cfgVal.(type) {
 		case bool:
 			disablePDCfg[cfgKey] = false
 		case int:
 			limit := int(value.(float64))
-			disablePDCfg[cfgKey] = int(math.Min(40, float64(limit*len(stores)))) * cfgVal.(int)
+			disablePDCfg[cfgKey] = int(math.Min(40, float64(limit*len(stores)))) * v
 		}
 	}
 	undo = p.makeUndoFunctionByConfig(clusterConfig{scheduleCfg: scheduleCfg})

diff --git a/pkg/pdutil/pd_test.go b/pkg/pdutil/pd_test.go
@@ -38,7 +38,7 @@ func (s *testPDControllerSuite) TestScheduler(c *C) {
 	}
 	schedulerPauseCh := make(chan struct{})
 	pdController := &PdController{addrs: []string{"", ""}, schedulerPauseCh: schedulerPauseCh}
-	_, err := pdController.pauseSchedulersAndConfigWith(ctx, []string{scheduler}, mock)
+	_, err := pdController.pauseSchedulersAndConfigWith(ctx, []string{scheduler}, nil, mock)
 	c.Assert(err, ErrorMatches, "failed")
 
 	go func() {
@@ -53,7 +53,7 @@ func (s *testPDControllerSuite) TestScheduler(c *C) {
 	mock = func(context.Context, string, string, *http.Client, string, io.Reader) ([]byte, error) {
 		return []byte(`["` + scheduler + `"]`), nil
 	}
-	_, err = pdController.pauseSchedulersAndConfigWith(ctx, []string{scheduler}, mock)
+	_, err = pdController.pauseSchedulersAndConfigWith(ctx, []string{scheduler}, nil, mock)
 	c.Assert(err, IsNil)
 
 	go func() {