Mark preempted jobs as failed

armadaproject · May 3, 2023 · 4ed6ba7 · 4ed6ba7
1 parent 531db48
commit 4ed6ba7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/internal/scheduler/scheduling_algo.go b/internal/scheduler/scheduling_algo.go
@@ -111,9 +111,9 @@ func (l *FairSchedulingAlgo) Schedule(
 
 		// Update jobDb.
 		preemptedJobs := PreemptedJobsFromSchedulerResult[*jobdb.Job](schedulerResult)
+		preemptedJobs = util.Map(preemptedJobs, func(job *jobdb.Job) *jobdb.Job { return job.WithFailed(true) })
 		scheduledJobs := ScheduledJobsFromSchedulerResult[*jobdb.Job](schedulerResult)
 		if err := jobDb.Upsert(txn, preemptedJobs); err != nil {
-			// TODO: We need to do something here to mark the jobs as preempted in the jobDb.
 			return nil, err
 		}
 		if err := jobDb.Upsert(txn, scheduledJobs); err != nil {