support preempting task with bound status

Signed-off-by: lili <lili_9309@163.com>
volcano-sh · Nov 16, 2023 · 8093238 · 8093238
1 parent 5683dbe
commit 8093238
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/pkg/scheduler/actions/preempt/preempt.go b/pkg/scheduler/actions/preempt/preempt.go
@@ -114,7 +114,7 @@ func (pmpt *Action) Execute(ssn *framework.Session) {
 
 				if preempted, _ := preempt(ssn, stmt, preemptor, func(task *api.TaskInfo) bool {
 					// Ignore non running task.
-					if task.Status != api.Running {
+					if !api.PreemptibleStatus(task.Status) {
 						return false
 					}
 					// Ignore task with empty resource request.
@@ -169,7 +169,7 @@ func (pmpt *Action) Execute(ssn *framework.Session) {
 				stmt := framework.NewStatement(ssn)
 				assigned, _ := preempt(ssn, stmt, preemptor, func(task *api.TaskInfo) bool {
 					// Ignore non running task.
-					if task.Status != api.Running {
+					if !api.PreemptibleStatus(task.Status) {
 						return false
 					}
 					// Ignore task with empty resource request.

diff --git a/pkg/scheduler/api/helpers.go b/pkg/scheduler/api/helpers.go
@@ -60,6 +60,16 @@ func getTaskStatus(pod *v1.Pod) TaskStatus {
 	return Unknown
 }
 
+// PreemptibleStatus checks whether the task can be preempted
+func PreemptibleStatus(status TaskStatus) bool {
+	switch status {
+	case Bound, Running:
+		return true
+	default:
+		return false
+	}
+}
+
 // AllocatedStatus checks whether the tasks has AllocatedStatus
 func AllocatedStatus(status TaskStatus) bool {
 	switch status {