Fix: keep the last invoked task alive even when it triggered drop cache

sbip-sg · Apr 9, 2024 · ac1757e · ac1757e
1 parent 25d6f2c
commit ac1757e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/task_executor/executor.go b/task_executor/executor.go
@@ -316,6 +316,10 @@ func (meca *MecaExecutor) handleOneRequest(req *MecaRequestHandle) {
 			return
 		}
 		log.Printf("task with config %v added: %v ", req.taskCfg, h.task)
+		// if cleared cache, we add back the reference
+		if retryDueToResourceShortage > 0 {
+			meca.tracker.add(taskId, h.task)
+		}
 		go func() {
 			defer req.Done()
 			req.output, req.err = h.task.Execute(req.ctx, req.input)