Add max retries to SubmitJob

tilebox · May 7, 2024 · c41e6c9 · c41e6c9
1 parent ad585e9
commit c41e6c9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/workflows/v1/jobs.go b/workflows/v1/jobs.go
@@ -27,7 +27,7 @@ func NewJobService(client workflowsv1connect.JobServiceClient) *JobService {
 	}
 }
 
-func (js *JobService) Submit(ctx context.Context, jobName, clusterSlug string, tasks ...Task) (*workflowsv1.Job, error) {
+func (js *JobService) Submit(ctx context.Context, jobName, clusterSlug string, maxRetries int, tasks ...Task) (*workflowsv1.Job, error) {
 	if len(tasks) == 0 {
 		return nil, errors.New("no tasks to submit")
 	}
@@ -63,8 +63,9 @@ func (js *JobService) Submit(ctx context.Context, jobName, clusterSlug string, t
 				Name:    identifier.Name(),
 				Version: identifier.Version(),
 			},
-			Input:   subtaskInput,
-			Display: identifier.Display(),
+			Input:      subtaskInput,
+			Display:    identifier.Display(),
+			MaxRetries: int64(maxRetries),
 		})
 	}
 

diff --git a/workflows/v1/runner_test.go b/workflows/v1/runner_test.go
@@ -26,7 +26,7 @@ type badIdentifierTask struct {
 }
 
 func (t *badIdentifierTask) Identifier() TaskIdentifier {
-	return TaskIdentifier{Name: "", Version: ""}
+	return NewTaskIdentifier("", "")
 }
 
 func TestTaskRunner_RegisterTask(t *testing.T) {