[Internal] Update Jobs GetRun API to support paginated responses for jobs and ForEach tasks with >100 runs

service/jobs/ext_api.go

@@ -0,0 +1,36 @@
+package jobs
+
+import "context"
+
+// GetRun retrieves a run based on the provided request.
+// It handles pagination if the run contains multiple iterations or tasks.
+func (a *JobsAPI) GetRun(ctx context.Context, request GetRunRequest) (*Run, error) {
+	run, err := a.jobsImpl.GetRun(ctx, request)
+	if err != nil {
+		return nil, err
+	}
+
+	// When querying a Job run, a page token is returned when there are more than 100 tasks. No iterations are defined for a Job run. Therefore, the next page in the response only includes the next page of tasks.
+	// When querying a ForEach task run, a page token is returned when there are more than 100 iterations. Only a single task is returned, corresponding to the ForEach task itself. Therefore, the client only reads the iterations from the next page and not the tasks.
+	isPaginatingIterations := run.Iterations != nil && len(run.Iterations) > 0
+
+	pageToken := run.NextPageToken
+	for pageToken != "" {
+		request.PageToken = pageToken
+		nextRun, err := a.jobsImpl.GetRun(ctx, request)
+		if err != nil {
+			return nil, err
+		}
+
+		if isPaginatingIterations {
+			run.Iterations = append(run.Iterations, nextRun.Iterations...)
+		} else {
+			run.Tasks = append(run.Tasks, nextRun.Tasks...)
+		}
+		pageToken = nextRun.NextPageToken
+	}
+
+	run.NextPageToken = ""
+	run.PrevPageToken = ""
+	return run, nil
+}