Add optional skipRetrying on runTask errors

nicktrn · nicktrn · commit cb6e260430fd · 2023-09-19T12:45:43.000Z
diff --git a/packages/trigger-sdk/src/io.ts b/packages/trigger-sdk/src/io.ts
@@ -56,7 +56,11 @@ export type RunTaskErrorCallback = (
   error: unknown,
   task: IOTask,
   io: IO
-) => { retryAt: Date; error?: Error; jitter?: number } | Error | undefined | void;
+) =>
+  | { retryAt?: Date; error?: Error; jitter?: number; skipRetrying?: boolean }
+  | Error
+  | undefined
+  | void;
 
 export class IO {
   private _id: string;
@@ -574,6 +578,8 @@ export class IO {
           throw error;
         }
 
+        let skipRetrying = false;
+
         if (onError) {
           try {
             const onErrorResult = onError(error, task, this);
@@ -582,13 +588,17 @@ export class IO {
               if (onErrorResult instanceof Error) {
                 error = onErrorResult;
               } else {
-                const parsedError = ErrorWithStackSchema.safeParse(onErrorResult.error);
+                skipRetrying = !!onErrorResult.skipRetrying;
+
+                if (onErrorResult.retryAt && !skipRetrying) {
+                  const parsedError = ErrorWithStackSchema.safeParse(onErrorResult.error);
 
-                throw new RetryWithTaskError(
-                  parsedError.success ? parsedError.data : { message: "Unknown error" },
-                  task,
-                  onErrorResult.retryAt
-                );
+                  throw new RetryWithTaskError(
+                    parsedError.success ? parsedError.data : { message: "Unknown error" },
+                    task,
+                    onErrorResult.retryAt
+                  );
+                }
               }
             }
           } catch (innerError) {
@@ -602,7 +612,7 @@ export class IO {
 
         const parsedError = ErrorWithStackSchema.safeParse(error);
 
-        if (options?.retry) {
+        if (options?.retry && !skipRetrying) {
           const retryAt = calculateRetryAt(options.retry, task.attempts - 1);
 
           if (retryAt) {
