address comment

Signed-off-by: Cabinfever_B <[email protected]>
tikv · Oct 12, 2022 · 24b703d · 24b703d
1 parent d074fe1
commit 24b703d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/server/schedule/plan/plan.go b/server/schedule/plan/plan.go
@@ -88,9 +88,9 @@ func SetResource(resource interface{}) Option {
 	}
 }
 
-// SetResourceAtThirdStep is used to generate Resource for plan
-func SetResourceAtThirdStep(resource interface{}) Option {
+// SetResourceWithStep is used to generate Resource for plan
+func SetResourceWithStep(resource interface{}, step int) Option {
 	return func(plan Plan) {
-		plan.SetResourceWithStep(resource, 2)
+		plan.SetResourceWithStep(resource, step)
 	}
 }