blender/flamenco
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Manager: mark task as active as soon as it is assigned to a worker

Browse Source 
Move the task to 'active' status so that it won't be assigned to another
worker. This also enables the task timeout monitoring.
This commit is contained in:
Sybren A. Stüvel 2022-06-20 13:00:41 +02:00
parent 2a4c9b2c13
commit 1586c37b32
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
internal/manager/api_impl/workers.go
View File

@ -311,6 +311,12 @@ func (f *Flamenco) ScheduleTask(e echo.Context) error {
return sendAPIError(e, http.StatusInternalServerError, "internal error appending to task log: %v", err) return sendAPIError(e, http.StatusInternalServerError, "internal error appending to task log: %v", err)
} }
// Move the task to 'active' status so that it won't be assigned to another
// worker. This also enables the task timeout monitoring.
if err := f.stateMachine.TaskStatusChange(ctx, dbTask, api.TaskStatusActive); err != nil {
return sendAPIError(e, http.StatusInternalServerError, "internal error marking task as active: %v", err)
}
// Start timeout measurement as soon as the Worker gets the task assigned. // Start timeout measurement as soon as the Worker gets the task assigned.
if err := f.workerPingedTask(ctx, logger, dbTask); err != nil { if err := f.workerPingedTask(ctx, logger, dbTask); err != nil {
return sendAPIError(e, http.StatusInternalServerError, "internal error updating task for timeout calculation: %v", err) return sendAPIError(e, http.StatusInternalServerError, "internal error updating task for timeout calculation: %v", err)