Add support for function.paused_at in ScheduleBatchPayload/RetrieveAn…

…dScheduleBatch (#1353)
inngest · May 13, 2024 · 9ea844b · 9ea844b
1 parent 3af47ce
commit 9ea844b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 13 deletions.
diff --git a/pkg/execution/batch/batch.go b/pkg/execution/batch/batch.go
@@ -78,10 +78,11 @@ type ScheduleBatchOpts struct {
 }
 
 type ScheduleBatchPayload struct {
-	BatchID         ulid.ULID `json:"batchID"`
-	AccountID       uuid.UUID `json:"acctID"`
-	WorkspaceID     uuid.UUID `json:"wsID"`
-	AppID           uuid.UUID `json:"appID"`
-	FunctionID      uuid.UUID `json:"fnID"`
-	FunctionVersion int       `json:"fnV"`
+	BatchID          ulid.ULID  `json:"batchID"`
+	AccountID        uuid.UUID  `json:"acctID"`
+	WorkspaceID      uuid.UUID  `json:"wsID"`
+	AppID            uuid.UUID  `json:"appID"`
+	FunctionID       uuid.UUID  `json:"fnID"`
+	FunctionVersion  int        `json:"fnV"`
+	FunctionPausedAt *time.Time `json:"fpAt,omitempty"`
 }
diff --git a/pkg/execution/executor/executor.go b/pkg/execution/executor/executor.go
@@ -2528,13 +2528,14 @@ func (e executor) RetrieveAndScheduleBatch(ctx context.Context, fn inngest.Funct
 
 	key := fmt.Sprintf("%s-%s", fn.ID, payload.BatchID)
 	identifier, err := e.Schedule(ctx, execution.ScheduleRequest{
-		AccountID:      payload.AccountID,
-		WorkspaceID:    payload.WorkspaceID,
-		AppID:          payload.AppID,
-		Function:       fn,
-		Events:         events,
-		BatchID:        &payload.BatchID,
-		IdempotencyKey: &key,
+		AccountID:        payload.AccountID,
+		WorkspaceID:      payload.WorkspaceID,
+		AppID:            payload.AppID,
+		Function:         fn,
+		Events:           events,
+		BatchID:          &payload.BatchID,
+		IdempotencyKey:   &key,
+		FunctionPausedAt: payload.FunctionPausedAt,
 	})
 	if err != nil {
 		span.SetStatus(codes.Error, err.Error())