pg-workflows 0.13.1 → 0.14.0

package/dist/index.js

@@ -1,6 +1,7 @@
 import {
   DEFAULT_PGBOSS_SCHEMA,
   PAUSE_EVENT_NAME,
+  STEP_BASE_METHOD_TYPES,
   WORKFLOW_RUN_DLQ_QUEUE_NAME,
   WORKFLOW_RUN_QUEUE_NAME,
   WorkflowClient,
@@ -14,6 +15,7 @@ import {
   insertWorkflowRun,
   invokeChildWorkflowTimelineKey,
   isInvokeChildWorkflowTimelineEntry,
+  resolvePriority,
   runMigrations,
   scheduleQueueNameFor,
   updateWorkflowRun,
@@ -22,7 +24,7 @@ import {
   waitForTimelineKey,
   withPostgresTransaction,
   workflow
-} from "./shared/chunk-5xswmve7.js";
+} from "./shared/chunk-w3qcrrwx.js";
 // src/engine.ts
 import { merge } from "es-toolkit";
 import pg from "pg";
@@ -73,11 +75,11 @@ function parseWorkflowHandler(handler) {
       const propertyAccess = node.expression;
       const objectName = propertyAccess.expression.getText(sourceFile);
       const methodName = propertyAccess.name.text;
-      if (objectName === "step" && (methodName === "run" || methodName === "waitFor" || methodName === "pause" || methodName === "waitUntil" || methodName === "delay" || methodName === "sleep" || methodName === "poll" || methodName === "invokeChildWorkflow")) {
+      const stepType = objectName === "step" ? STEP_BASE_METHOD_TYPES.get(methodName) : undefined;
+      if (stepType !== undefined) {
         const firstArg = node.arguments[0];
         if (firstArg) {
           const { id, isDynamic } = extractStepId(firstArg);
-          const stepType = methodName === "sleep" ? "delay" /* DELAY */ : methodName;
           const stepDefinition = {
             id,
             type: stepType,
@@ -381,6 +383,9 @@ class WorkflowEngine {
     });
     return run;
   }
+  async createChildWorkflowRun(params) {
+    return this.createWorkflowRun(params);
+  }
   async createWorkflowRun({
     workflowId,
     input,
@@ -390,6 +395,7 @@ class WorkflowEngine {
     parentRunId,
     parentStepId,
     parentResourceId,
+    parentPriority,
     scheduledAt,
     enqueue = true,
     db
@@ -420,6 +426,7 @@ class WorkflowEngine {
       status: "running" /* RUNNING */,
       input,
       maxRetries: options?.retries ?? workflow2.retries ?? 0,
+      priority: resolvePriority(options?.priority, workflow2.priority, parentPriority),
       timeoutAt,
       idempotencyKey,
       parentRunId,
@@ -447,6 +454,7 @@ class WorkflowEngine {
     await this.boss.send(WORKFLOW_RUN_QUEUE_NAME, job, {
       startAfter: new Date,
       expireInSeconds: options?.expireInSeconds ?? defaultExpireInSeconds,
+      priority: run.priority,
       ...retrySendOptions(run.maxRetries),
       ...db ? { db } : {}
     });
@@ -596,6 +604,7 @@ class WorkflowEngine {
     };
     await this.boss.send(WORKFLOW_RUN_QUEUE_NAME, job, {
       expireInSeconds: options?.expireInSeconds ?? defaultExpireInSeconds,
+      priority: run.priority,
       ...retrySendOptions(run.maxRetries)
     });
     this.logger.log(`event ${eventName} sent for workflow run with id ${runId}`);
@@ -767,75 +776,7 @@ class WorkflowEngine {
           }, { db });
         }, this.pool);
       }
-      const baseStep = {
-        run: async (stepId, handler) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          return this.runStep({ stepId, run, handler });
-        },
-        waitFor: async (stepId, { eventName, timeout }) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          const timeoutDate = timeout ? new Date(Date.now() + timeout) : undefined;
-          return this.waitStep({ run, stepId, eventName, timeoutDate });
-        },
-        waitUntil: async (stepId, dateOrOptions) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          const date = dateOrOptions instanceof Date ? dateOrOptions : typeof dateOrOptions === "string" ? new Date(dateOrOptions) : dateOrOptions.date instanceof Date ? dateOrOptions.date : new Date(dateOrOptions.date);
-          await this.waitStep({ run, stepId, timeoutDate: date });
-        },
-        pause: async (stepId) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          await this.waitStep({ run, stepId, eventName: PAUSE_EVENT_NAME });
-        },
-        delay: async (stepId, duration) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          await this.waitStep({
-            run,
-            stepId,
-            timeoutDate: new Date(Date.now() + parseDuration(duration))
-          });
-        },
-        get sleep() {
-          return this.delay;
-        },
-        poll: async (stepId, conditionFn, options) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          const intervalMs = parseDuration(options?.interval ?? "30s");
-          if (intervalMs < 30000) {
-            throw new WorkflowEngineError(`step.poll interval must be at least 30s (got ${intervalMs}ms)`, workflowId, runId);
-          }
-          const timeoutMs = options?.timeout ? parseDuration(options.timeout) : undefined;
-          return this.pollStep({ run, stepId, conditionFn, intervalMs, timeoutMs });
-        },
-        invokeChildWorkflow: async (stepId, refOrParams, inputArg, optionsArg) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          const resolvedChildCall = this.resolveWorkflowRunParameters(refOrParams, inputArg, optionsArg);
-          const childWorkflowInvocation = {
-            run,
-            stepId,
-            workflowId: resolvedChildCall.workflowId,
-            input: resolvedChildCall.input,
-            options: resolvedChildCall.options,
-            resourceId: resolvedChildCall.resourceId,
-            idempotencyKey: resolvedChildCall.idempotencyKey
-          };
-          return this.invokeChildWorkflowStep(childWorkflowInvocation);
-        }
-      };
-      let step = { ...baseStep };
+      let step = this.createBaseStep(run, workflowId, runId);
       const plugins = workflow2.plugins ?? [];
       const context = {
         input: run.input,
@@ -925,6 +866,55 @@ class WorkflowEngine {
       workflowId: run.workflowId
     });
   }
+  createBaseStep(run, workflowId, runId) {
+    return {
+      run: async (stepId, handler) => {
+        return this.runStep({ stepId, run, handler });
+      },
+      waitFor: async (stepId, { eventName, timeout }) => {
+        const timeoutDate = timeout ? new Date(Date.now() + timeout) : undefined;
+        return this.waitStep({ run, stepId, eventName, timeoutDate });
+      },
+      waitUntil: async (stepId, dateOrOptions) => {
+        const date = dateOrOptions instanceof Date ? dateOrOptions : typeof dateOrOptions === "string" ? new Date(dateOrOptions) : dateOrOptions.date instanceof Date ? dateOrOptions.date : new Date(dateOrOptions.date);
+        await this.waitStep({ run, stepId, timeoutDate: date });
+      },
+      pause: async (stepId) => {
+        await this.waitStep({ run, stepId, eventName: PAUSE_EVENT_NAME });
+      },
+      delay: async (stepId, duration) => {
+        await this.waitStep({
+          run,
+          stepId,
+          timeoutDate: new Date(Date.now() + parseDuration(duration))
+        });
+      },
+      get sleep() {
+        return this.delay;
+      },
+      poll: async (stepId, conditionFn, options) => {
+        const intervalMs = parseDuration(options?.interval ?? "30s");
+        if (intervalMs < 30000) {
+          throw new WorkflowEngineError(`step.poll interval must be at least 30s (got ${intervalMs}ms)`, workflowId, runId);
+        }
+        const timeoutMs = options?.timeout ? parseDuration(options.timeout) : undefined;
+        return this.pollStep({ run, stepId, conditionFn, intervalMs, timeoutMs });
+      },
+      invokeChildWorkflow: async (stepId, refOrParams, inputArg, optionsArg) => {
+        const resolvedChildCall = this.resolveWorkflowRunParameters(refOrParams, inputArg, optionsArg);
+        const childWorkflowInvocation = {
+          run,
+          stepId,
+          workflowId: resolvedChildCall.workflowId,
+          input: resolvedChildCall.input,
+          options: resolvedChildCall.options,
+          resourceId: resolvedChildCall.resourceId,
+          idempotencyKey: resolvedChildCall.idempotencyKey
+        };
+        return this.invokeChildWorkflowStep(childWorkflowInvocation);
+      }
+    };
+  }
   getCachedStepEntry(timeline, stepId) {
     const stepEntry = timeline[stepId];
     return stepEntry && typeof stepEntry === "object" && "output" in stepEntry ? stepEntry : null;
@@ -1015,7 +1005,7 @@ class WorkflowEngine {
         });
         return;
       }
-      const result = await this.createWorkflowRun({
+      const result = await this.createChildWorkflowRun({
         workflowId,
         input,
         resourceId: childResourceId,
@@ -1024,6 +1014,7 @@ class WorkflowEngine {
         parentRunId: run.id,
         parentStepId: stepId,
         parentResourceId: run.resourceId ?? undefined,
+        parentPriority: run.priority,
         enqueue: true,
         db
       });
@@ -1222,6 +1213,7 @@ ${error.stack}` : String(error)
         await this.boss.send(WORKFLOW_RUN_QUEUE_NAME, job, {
           startAfter: timeoutDate.getTime() <= Date.now() ? new Date : timeoutDate,
           expireInSeconds: defaultExpireInSeconds,
+          priority: run.priority,
           ...retrySendOptions(run.maxRetries)
         });
       } catch (error) {
@@ -1340,6 +1332,7 @@ ${error.stack}` : String(error)
       }, {
         startAfter: new Date(Date.now() + intervalMs),
         expireInSeconds: defaultExpireInSeconds,
+        priority: run.priority,
         ...retrySendOptions(run.maxRetries)
       });
     } catch (error) {
@@ -1581,5 +1574,5 @@ export {
   WorkflowClient
 };
 
-//# debugId=C2CB4B181900270764756E2164756E21
+//# debugId=A6C8F39FA22E601264756E2164756E21
 //# sourceMappingURL=index.js.map