Merge pull request #2342 from Thor-wl/0708-fix2341

filter the rescheduling strategies which contain victim fucntions
volcano-sh · Jul 8, 2022 · 42702f7 · 42702f7
2 parents e10dac4 + e358f53
commit 42702f7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pkg/scheduler/plugins/rescheduling/rescheduling.go b/pkg/scheduler/plugins/rescheduling/rescheduling.go
@@ -97,11 +97,13 @@ func (rp *reschedulingPlugin) OnSessionOpen(ssn *framework.Session) {
 		return
 	}
 
-	// Get all strategies and register the victim functions
+	// Get all strategies and register the victim functions for each strategy.
 	victimFns := make([]api.VictimTasksFn, 0)
 	for _, strategy := range configs.strategies {
-		klog.V(4).Infof("strategy: %s\n", strategy.Name)
-		victimFns = append(victimFns, VictimFn[strategy.Name])
+		if VictimFn[strategy.Name] != nil {
+			klog.V(4).Infof("strategy: %s\n", strategy.Name)
+			victimFns = append(victimFns, VictimFn[strategy.Name])
+		}
 	}
 	ssn.AddVictimTasksFns(rp.Name(), victimFns)
 }