runner: Respect 403 errors from the API

livepeer · May 24, 2024 · 018480f · 018480f
1 parent 2204bb6
commit 018480f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/task/runner.go b/task/runner.go
@@ -344,6 +344,9 @@ func (r *runner) handleTask(ctx context.Context, taskInfo data.TaskInfo) (out *T
 	if err == api.ErrRateLimited {
 		glog.Warningf("Task execution rate limited type=%q id=%s userID=%s", taskType, taskID, taskCtx.UserID)
 		return nil, r.delayTaskStep(ctx, taskID, taskCtx.Step, taskCtx.StepInput)
+	} else if httpErr := (&api.HTTPStatusError{}); errors.As(err, &httpErr) && httpErr.StatusCode == 403 {
+		glog.Warningf("Task execution forbidden type=%q id=%s userID=%s error=%q", taskType, taskID, taskCtx.UserID, httpErr.Message)
+		return nil, fmt.Errorf("task execution forbidden: %w", httpErr)
 	} else if err != nil {
 		glog.Errorf("Error updating task progress type=%q id=%s err=%q unretriable=%v", taskType, taskID, err, IsUnretriable(err))
 		// execute the task anyway