@trigger.dev/redis-worker 4.0.0-v4-beta.24 → 4.0.0-v4-beta.26

package/dist/index.cjs

@@ -9472,13 +9472,17 @@ var SimpleQueue = class {
     this.#registerCommands();
     this.schema = schema;
   }
+  async cancel(cancellationKey) {
+    await this.redis.set(`cancellationKey:${cancellationKey}`, "1", "EX", 60 * 60 * 24);
+  }
   async enqueue({
     id,
     job,
     item,
     attempt,
     availableAt,
-    visibilityTimeoutMs
+    visibilityTimeoutMs,
+    cancellationKey
   }) {
     try {
       const score = availableAt ? availableAt.getTime() : Date.now();
@@ -9490,13 +9494,14 @@ var SimpleQueue = class {
         attempt,
         deduplicationKey
       });
-      const result = await this.redis.enqueueItem(
+      const result = cancellationKey ? await this.redis.enqueueItemWithCancellationKey(
         `queue`,
         `items`,
+        `cancellationKey:${cancellationKey}`,
         id ?? nanoid(),
         score,
         serializedItem
-      );
+      ) : await this.redis.enqueueItem(`queue`, `items`, id ?? nanoid(), score, serializedItem);
       if (result !== 1) {
         throw new Error("Enqueue operation failed");
       }
@@ -9749,6 +9754,28 @@ var SimpleQueue = class {
         return 1
       `
     });
+    this.redis.defineCommand("enqueueItemWithCancellationKey", {
+      numberOfKeys: 3,
+      lua: `
+        local queue = KEYS[1]
+        local items = KEYS[2]
+        local cancellationKey = KEYS[3]
+
+        local id = ARGV[1]
+        local score = ARGV[2]
+        local serializedItem = ARGV[3]
+
+        -- if the cancellation key exists, return 1
+        if redis.call('EXISTS', cancellationKey) == 1 then
+          return 1
+        end
+
+        redis.call('ZADD', queue, score, id)
+        redis.call('HSET', items, id, serializedItem)
+
+        return 1
+      `
+    });
     this.redis.defineCommand("dequeueItems", {
       numberOfKeys: 2,
       lua: `
@@ -11201,7 +11228,8 @@ var Worker = class _Worker {
     job,
     payload,
     visibilityTimeoutMs,
-    availableAt
+    availableAt,
+    cancellationKey
   }) {
     return startSpan(
       this.tracer,
@@ -11219,7 +11247,8 @@ var Worker = class _Worker {
             job,
             item: payload,
             visibilityTimeoutMs: timeout,
-            availableAt
+            availableAt,
+            cancellationKey
           }),
           {
             job_type: String(job),
@@ -11308,6 +11337,16 @@ var Worker = class _Worker {
       }
     );
   }
+  /**
+   * Cancels a job before it's enqueued.
+   * @param cancellationKey - The cancellation key to cancel.
+   * @returns A promise that resolves when the job is cancelled.
+   *
+   * Any jobs enqueued with the same cancellation key will be not be enqueued.
+   */
+  cancel(cancellationKey) {
+    return startSpan(this.tracer, "cancel", () => this.queue.cancel(cancellationKey));
+  }
   ack(id) {
     return startSpan(
       this.tracer,