@dudousxd/nestjs-durable-transport-sqs 0.1.0 → 0.2.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Davide Carvalho
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/index.cjs

@@ -0,0 +1,218 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  SqsTransport: () => SqsTransport
+});
+module.exports = __toCommonJS(index_exports);
+
+// src/sqs-transport.ts
+var import_client_sqs = require("@aws-sdk/client-sqs");
+var import_nestjs_durable_core = require("@dudousxd/nestjs-durable-core");
+var SqsTransport = class {
+  static {
+    __name(this, "SqsTransport");
+  }
+  client;
+  ownsClient;
+  group;
+  prefix;
+  autoCreate;
+  marker;
+  waitTimeSeconds;
+  visibilityTimeoutSec;
+  taskQueueUrlResolver;
+  resultsQueueUrlOption;
+  handlers = /* @__PURE__ */ new Map();
+  urlCache = /* @__PURE__ */ new Map();
+  running = true;
+  abort = new AbortController();
+  taskLoop;
+  resultLoop;
+  constructor(options) {
+    this.client = options.client ?? new import_client_sqs.SQSClient(options.clientConfig ?? {});
+    this.ownsClient = !options.client;
+    this.group = options.group;
+    this.prefix = options.prefix ?? "durable";
+    this.autoCreate = options.autoCreate ?? false;
+    this.marker = options.marker;
+    this.waitTimeSeconds = options.waitTimeSeconds ?? 20;
+    this.visibilityTimeoutSec = options.visibilityTimeoutSec ?? 60;
+    this.taskQueueUrlResolver = options.taskQueueUrl;
+    this.resultsQueueUrlOption = options.resultsQueueUrl;
+  }
+  // ── queue resolution ───────────────────────────────────────────────────────────────────────
+  async resolveByName(name) {
+    const cached = this.urlCache.get(name);
+    if (cached) return cached;
+    let url;
+    try {
+      const out = await this.client.send(new import_client_sqs.GetQueueUrlCommand({
+        QueueName: name
+      }));
+      url = out.QueueUrl;
+    } catch (err) {
+      if (this.autoCreate) {
+        const created = await this.client.send(new import_client_sqs.CreateQueueCommand({
+          QueueName: name
+        }));
+        url = created.QueueUrl;
+      } else {
+        throw err;
+      }
+    }
+    this.urlCache.set(name, url);
+    return url;
+  }
+  async tasksUrl(group) {
+    if (this.taskQueueUrlResolver) return this.taskQueueUrlResolver(group);
+    return this.resolveByName(`${this.prefix}-tasks-${group}`);
+  }
+  async resultsUrl() {
+    const opt = this.resultsQueueUrlOption;
+    if (typeof opt === "string") return opt;
+    if (typeof opt === "function") return opt();
+    return this.resolveByName(`${this.prefix}-results`);
+  }
+  // ── engine → worker ────────────────────────────────────────────────────────────────────────
+  async dispatch(task) {
+    const QueueUrl = await this.tasksUrl(task.group);
+    await this.client.send(new import_client_sqs.SendMessageCommand({
+      QueueUrl,
+      MessageBody: JSON.stringify(task),
+      MessageAttributes: this.marker ? {
+        [this.marker]: {
+          DataType: "String",
+          StringValue: "1"
+        }
+      } : void 0
+    }));
+  }
+  // ── worker → engine ────────────────────────────────────────────────────────────────────────
+  /** Register a step handler (worker side). Starts the group's task consumer on first call. */
+  handle(name, fn) {
+    if (!this.group) throw new Error("SqsTransport needs a `group` to register handlers");
+    this.handlers.set(name, fn);
+    if (!this.taskLoop) this.taskLoop = this.pollTasks(this.group);
+  }
+  onResult(handler) {
+    if (!this.resultLoop) this.resultLoop = this.pollResults(handler);
+  }
+  onHeartbeat(_handler) {
+  }
+  // ── polling loops ──────────────────────────────────────────────────────────────────────────
+  async pollTasks(group) {
+    const QueueUrl = await this.tasksUrl(group);
+    while (this.running) {
+      const messages = await this.receive(QueueUrl, this.visibilityTimeoutSec);
+      for (const msg of messages) {
+        if (this.shouldSkip(msg)) {
+          await this.release(QueueUrl, msg.ReceiptHandle);
+          continue;
+        }
+        const task = JSON.parse(msg.Body ?? "{}");
+        const result = await (0, import_nestjs_durable_core.runStepHandler)(task, this.handlers.get(task.name));
+        await this.client.send(new import_client_sqs.SendMessageCommand({
+          QueueUrl: await this.resultsUrl(),
+          MessageBody: JSON.stringify(result),
+          MessageAttributes: this.marker ? {
+            [this.marker]: {
+              DataType: "String",
+              StringValue: "1"
+            }
+          } : void 0
+        }));
+        await this.del(QueueUrl, msg.ReceiptHandle);
+      }
+    }
+  }
+  async pollResults(handler) {
+    const QueueUrl = await this.resultsUrl();
+    while (this.running) {
+      const messages = await this.receive(QueueUrl, 30);
+      for (const msg of messages) {
+        if (this.shouldSkip(msg)) {
+          await this.release(QueueUrl, msg.ReceiptHandle);
+          continue;
+        }
+        await handler(JSON.parse(msg.Body ?? "{}"));
+        await this.del(QueueUrl, msg.ReceiptHandle);
+      }
+    }
+  }
+  /** One long-poll receive. Swallows the abort thrown on `close()`; logs and backs off otherwise. */
+  async receive(QueueUrl, visibility) {
+    try {
+      const out = await this.client.send(new import_client_sqs.ReceiveMessageCommand({
+        QueueUrl,
+        MaxNumberOfMessages: 10,
+        WaitTimeSeconds: this.waitTimeSeconds,
+        VisibilityTimeout: visibility,
+        MessageAttributeNames: this.marker ? [
+          "All"
+        ] : void 0
+      }), {
+        abortSignal: this.abort.signal
+      });
+      return out.Messages ?? [];
+    } catch (err) {
+      if (!this.running) return [];
+      console.error("[SqsTransport] receive failed", err);
+      await new Promise((r) => setTimeout(r, 1e3));
+      return [];
+    }
+  }
+  shouldSkip(msg) {
+    return !!this.marker && !msg.MessageAttributes?.[this.marker];
+  }
+  async del(QueueUrl, ReceiptHandle) {
+    if (!ReceiptHandle) return;
+    await this.client.send(new import_client_sqs.DeleteMessageCommand({
+      QueueUrl,
+      ReceiptHandle
+    }));
+  }
+  async release(QueueUrl, ReceiptHandle) {
+    if (!ReceiptHandle) return;
+    await this.client.send(new import_client_sqs.ChangeMessageVisibilityCommand({
+      QueueUrl,
+      ReceiptHandle,
+      VisibilityTimeout: 0
+    })).catch(() => {
+    });
+  }
+  /** Stop the pollers and (if we created it) destroy the SQS client so the process can exit. */
+  async close() {
+    this.running = false;
+    this.abort.abort();
+    await Promise.allSettled([
+      this.taskLoop,
+      this.resultLoop
+    ]);
+    if (this.ownsClient) this.client.destroy();
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  SqsTransport
+});
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts","../src/sqs-transport.ts"],"sourcesContent":["export * from './sqs-transport';\n","import {\n  ChangeMessageVisibilityCommand,\n  CreateQueueCommand,\n  DeleteMessageCommand,\n  GetQueueUrlCommand,\n  ReceiveMessageCommand,\n  SendMessageCommand,\n  SQSClient,\n  type SQSClientConfig,\n} from '@aws-sdk/client-sqs';\nimport {\n  type Heartbeat,\n  type RemoteTask,\n  type StepHandler,\n  type StepResult,\n  type Transport,\n  runStepHandler,\n} from '@dudousxd/nestjs-durable-core';\n\n/** Resolve a group's tasks queue URL. Sync or async; resolved once and cached. */\nexport type QueueUrlResolver = (group: string) => string | Promise<string>;\n\nexport interface SqsTransportOptions {\n  /**\n   * Reuse an existing SQS client (recommended — share the app's). When omitted, one is built\n   * from `clientConfig` and destroyed on `close()`.\n   */\n  client?: SQSClient;\n  /** Used only when `client` is omitted — e.g. `{ region, endpoint, credentials }` for ElasticMQ. */\n  clientConfig?: SQSClientConfig;\n\n  /** The worker group this instance serves. Required to register `handle()` consumers. */\n  group?: string;\n\n  /**\n   * Map a group → an **existing** tasks queue URL. This is the seam for reusing queues you already\n   * have (e.g. `extraction`, `ingestion`) instead of letting the transport mint its own. When\n   * omitted, the transport resolves `${prefix}-tasks-${group}` by name (and creates it if\n   * `autoCreate`). Return value is awaited and cached.\n   */\n  taskQueueUrl?: QueueUrlResolver;\n  /**\n   * The results queue URL (worker → engine). A literal URL or a resolver. When omitted, the\n   * transport resolves `${prefix}-results` by name (and creates it if `autoCreate`).\n   */\n  resultsQueueUrl?: string | (() => string | Promise<string>);\n\n  /** Prefix for the fallback `${prefix}-tasks-*` / `${prefix}-results` names. Default `durable`. */\n  prefix?: string;\n  /**\n   * Create the fallback-named queues on first use if they don't exist (handy for ElasticMQ / dev).\n   * Ignored for queues you supply via `taskQueueUrl` / `resultsQueueUrl` — those must already exist.\n   */\n  autoCreate?: boolean;\n\n  /**\n   * When set, durable messages carry this SQS message attribute and pollers **skip** messages\n   * that lack it (releasing them for a co-resident legacy consumer). Enables sharing one physical\n   * queue with a non-durable consumer — but only if that consumer likewise ignores tagged messages.\n   * Leave unset for the clean case: a queue dedicated to durable traffic.\n   */\n  marker?: string;\n\n  /** SQS receive long-poll seconds (0–20). Default 20. */\n  waitTimeSeconds?: number;\n  /** Visibility timeout for an in-flight task, in seconds. Must exceed the step's runtime. Default 60. */\n  visibilityTimeoutSec?: number;\n}\n\n/**\n * An SQS-backed `Transport` — dispatch remote steps over Amazon SQS (or ElasticMQ). Unlike the\n * BullMQ transport, it does **not** mint its own queue names: pass `taskQueueUrl` / `resultsQueueUrl`\n * to ride the queues you already operate. The wire payload is the documented `RemoteTask` /\n * `StepResult` JSON, so non-Node workers on the same queues interoperate.\n *\n * Run one instance engine-side (calls `onResult`, dispatches tasks) and one per worker process\n * (calls `handle()` for its group). SQS is at-least-once: a task whose process crashes before the\n * result is sent reappears after the visibility timeout (durable retry); a duplicate `StepResult`\n * is harmless because the engine resolves a step's `pending` exactly once by `stepId`.\n */\nexport class SqsTransport implements Transport {\n  private readonly client: SQSClient;\n  private readonly ownsClient: boolean;\n  private readonly group?: string | undefined;\n  private readonly prefix: string;\n  private readonly autoCreate: boolean;\n  private readonly marker?: string | undefined;\n  private readonly waitTimeSeconds: number;\n  private readonly visibilityTimeoutSec: number;\n  private readonly taskQueueUrlResolver?: QueueUrlResolver | undefined;\n  private readonly resultsQueueUrlOption?: string | (() => string | Promise<string>) | undefined;\n\n  private readonly handlers = new Map<string, StepHandler>();\n  private readonly urlCache = new Map<string, string>();\n  private running = true;\n  private readonly abort = new AbortController();\n  private taskLoop?: Promise<void>;\n  private resultLoop?: Promise<void>;\n\n  constructor(options: SqsTransportOptions) {\n    this.client = options.client ?? new SQSClient(options.clientConfig ?? {});\n    this.ownsClient = !options.client;\n    this.group = options.group;\n    this.prefix = options.prefix ?? 'durable';\n    this.autoCreate = options.autoCreate ?? false;\n    this.marker = options.marker;\n    this.waitTimeSeconds = options.waitTimeSeconds ?? 20;\n    this.visibilityTimeoutSec = options.visibilityTimeoutSec ?? 60;\n    this.taskQueueUrlResolver = options.taskQueueUrl;\n    this.resultsQueueUrlOption = options.resultsQueueUrl;\n  }\n\n  // ── queue resolution ───────────────────────────────────────────────────────────────────────\n\n  private async resolveByName(name: string): Promise<string> {\n    const cached = this.urlCache.get(name);\n    if (cached) return cached;\n    let url: string;\n    try {\n      const out = await this.client.send(new GetQueueUrlCommand({ QueueName: name }));\n      url = out.QueueUrl as string;\n    } catch (err) {\n      if (this.autoCreate) {\n        const created = await this.client.send(new CreateQueueCommand({ QueueName: name }));\n        url = created.QueueUrl as string;\n      } else {\n        throw err;\n      }\n    }\n    this.urlCache.set(name, url);\n    return url;\n  }\n\n  private async tasksUrl(group: string): Promise<string> {\n    if (this.taskQueueUrlResolver) return this.taskQueueUrlResolver(group);\n    return this.resolveByName(`${this.prefix}-tasks-${group}`);\n  }\n\n  private async resultsUrl(): Promise<string> {\n    const opt = this.resultsQueueUrlOption;\n    if (typeof opt === 'string') return opt;\n    if (typeof opt === 'function') return opt();\n    return this.resolveByName(`${this.prefix}-results`);\n  }\n\n  // ── engine → worker ────────────────────────────────────────────────────────────────────────\n\n  async dispatch(task: RemoteTask): Promise<void> {\n    const QueueUrl = await this.tasksUrl(task.group);\n    await this.client.send(\n      new SendMessageCommand({\n        QueueUrl,\n        MessageBody: JSON.stringify(task),\n        MessageAttributes: this.marker\n          ? { [this.marker]: { DataType: 'String', StringValue: '1' } }\n          : undefined,\n      }),\n    );\n  }\n\n  // ── worker → engine ────────────────────────────────────────────────────────────────────────\n\n  /** Register a step handler (worker side). Starts the group's task consumer on first call. */\n  handle(name: string, fn: StepHandler): void {\n    if (!this.group) throw new Error('SqsTransport needs a `group` to register handlers');\n    this.handlers.set(name, fn);\n    if (!this.taskLoop) this.taskLoop = this.pollTasks(this.group);\n  }\n\n  onResult(handler: (result: StepResult) => Promise<void>): void {\n    if (!this.resultLoop) this.resultLoop = this.pollResults(handler);\n  }\n\n  onHeartbeat(_handler: (beat: Heartbeat) => Promise<void>): void {\n    // Not modelled over SQS; the queue's visibility-timeout redelivery is the liveness mechanism.\n  }\n\n  // ── polling loops ──────────────────────────────────────────────────────────────────────────\n\n  private async pollTasks(group: string): Promise<void> {\n    const QueueUrl = await this.tasksUrl(group);\n    while (this.running) {\n      const messages = await this.receive(QueueUrl, this.visibilityTimeoutSec);\n      for (const msg of messages) {\n        if (this.shouldSkip(msg)) {\n          // Not ours (shared queue): release immediately so the legacy consumer can take it.\n          await this.release(QueueUrl, msg.ReceiptHandle);\n          continue;\n        }\n        const task = JSON.parse(msg.Body ?? '{}') as RemoteTask;\n        const result = await runStepHandler(task, this.handlers.get(task.name));\n        await this.client.send(\n          new SendMessageCommand({\n            QueueUrl: await this.resultsUrl(),\n            MessageBody: JSON.stringify(result),\n            MessageAttributes: this.marker\n              ? { [this.marker]: { DataType: 'String', StringValue: '1' } }\n              : undefined,\n          }),\n        );\n        await this.del(QueueUrl, msg.ReceiptHandle);\n      }\n    }\n  }\n\n  private async pollResults(handler: (result: StepResult) => Promise<void>): Promise<void> {\n    const QueueUrl = await this.resultsUrl();\n    while (this.running) {\n      const messages = await this.receive(QueueUrl, 30);\n      for (const msg of messages) {\n        if (this.shouldSkip(msg)) {\n          await this.release(QueueUrl, msg.ReceiptHandle);\n          continue;\n        }\n        await handler(JSON.parse(msg.Body ?? '{}') as StepResult);\n        await this.del(QueueUrl, msg.ReceiptHandle);\n      }\n    }\n  }\n\n  /** One long-poll receive. Swallows the abort thrown on `close()`; logs and backs off otherwise. */\n  private async receive(QueueUrl: string, visibility: number) {\n    try {\n      const out = await this.client.send(\n        new ReceiveMessageCommand({\n          QueueUrl,\n          MaxNumberOfMessages: 10,\n          WaitTimeSeconds: this.waitTimeSeconds,\n          VisibilityTimeout: visibility,\n          MessageAttributeNames: this.marker ? ['All'] : undefined,\n        }),\n        { abortSignal: this.abort.signal },\n      );\n      return out.Messages ?? [];\n    } catch (err) {\n      if (!this.running) return [];\n      console.error('[SqsTransport] receive failed', err);\n      await new Promise((r) => setTimeout(r, 1000));\n      return [];\n    }\n  }\n\n  private shouldSkip(msg: { MessageAttributes?: Record<string, unknown> | undefined }): boolean {\n    return !!this.marker && !msg.MessageAttributes?.[this.marker];\n  }\n\n  private async del(QueueUrl: string, ReceiptHandle?: string): Promise<void> {\n    if (!ReceiptHandle) return;\n    await this.client.send(new DeleteMessageCommand({ QueueUrl, ReceiptHandle }));\n  }\n\n  private async release(QueueUrl: string, ReceiptHandle?: string): Promise<void> {\n    if (!ReceiptHandle) return;\n    await this.client\n      .send(new ChangeMessageVisibilityCommand({ QueueUrl, ReceiptHandle, VisibilityTimeout: 0 }))\n      .catch(() => {});\n  }\n\n  /** Stop the pollers and (if we created it) destroy the SQS client so the process can exit. */\n  async close(): Promise<void> {\n    this.running = false;\n    this.abort.abort();\n    await Promise.allSettled([this.taskLoop, this.resultLoop]);\n    if (this.ownsClient) this.client.destroy();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAAA;;;;;;;ACAA,wBASO;AACP,iCAOO;AA+DA,IAAMA,eAAN,MAAMA;EAhFb,OAgFaA;;;EACMC;EACAC;EACAC;EACAC;EACAC;EACAC;EACAC;EACAC;EACAC;EACAC;EAEAC,WAAW,oBAAIC,IAAAA;EACfC,WAAW,oBAAID,IAAAA;EACxBE,UAAU;EACDC,QAAQ,IAAIC,gBAAAA;EACrBC;EACAC;EAER,YAAYC,SAA8B;AACxC,SAAKlB,SAASkB,QAAQlB,UAAU,IAAImB,4BAAUD,QAAQE,gBAAgB,CAAC,CAAA;AACvE,SAAKnB,aAAa,CAACiB,QAAQlB;AAC3B,SAAKE,QAAQgB,QAAQhB;AACrB,SAAKC,SAASe,QAAQf,UAAU;AAChC,SAAKC,aAAac,QAAQd,cAAc;AACxC,SAAKC,SAASa,QAAQb;AACtB,SAAKC,kBAAkBY,QAAQZ,mBAAmB;AAClD,SAAKC,uBAAuBW,QAAQX,wBAAwB;AAC5D,SAAKC,uBAAuBU,QAAQG;AACpC,SAAKZ,wBAAwBS,QAAQI;EACvC;;EAIA,MAAcC,cAAcC,MAA+B;AACzD,UAAMC,SAAS,KAAKb,SAASc,IAAIF,IAAAA;AACjC,QAAIC,OAAQ,QAAOA;AACnB,QAAIE;AACJ,QAAI;AACF,YAAMC,MAAM,MAAM,KAAK5B,OAAO6B,KAAK,IAAIC,qCAAmB;QAAEC,WAAWP;MAAK,CAAA,CAAA;AAC5EG,YAAMC,IAAII;IACZ,SAASC,KAAK;AACZ,UAAI,KAAK7B,YAAY;AACnB,cAAM8B,UAAU,MAAM,KAAKlC,OAAO6B,KAAK,IAAIM,qCAAmB;UAAEJ,WAAWP;QAAK,CAAA,CAAA;AAChFG,cAAMO,QAAQF;MAChB,OAAO;AACL,cAAMC;MACR;IACF;AACA,SAAKrB,SAASwB,IAAIZ,MAAMG,GAAAA;AACxB,WAAOA;EACT;EAEA,MAAcU,SAASnC,OAAgC;AACrD,QAAI,KAAKM,qBAAsB,QAAO,KAAKA,qBAAqBN,KAAAA;AAChE,WAAO,KAAKqB,cAAc,GAAG,KAAKpB,MAAM,UAAUD,KAAAA,EAAO;EAC3D;EAEA,MAAcoC,aAA8B;AAC1C,UAAMC,MAAM,KAAK9B;AACjB,QAAI,OAAO8B,QAAQ,SAAU,QAAOA;AACpC,QAAI,OAAOA,QAAQ,WAAY,QAAOA,IAAAA;AACtC,WAAO,KAAKhB,cAAc,GAAG,KAAKpB,MAAM,UAAU;EACpD;;EAIA,MAAMqC,SAASC,MAAiC;AAC9C,UAAMT,WAAW,MAAM,KAAKK,SAASI,KAAKvC,KAAK;AAC/C,UAAM,KAAKF,OAAO6B,KAChB,IAAIa,qCAAmB;MACrBV;MACAW,aAAaC,KAAKC,UAAUJ,IAAAA;MAC5BK,mBAAmB,KAAKzC,SACpB;QAAE,CAAC,KAAKA,MAAM,GAAG;UAAE0C,UAAU;UAAUC,aAAa;QAAI;MAAE,IAC1DC;IACN,CAAA,CAAA;EAEJ;;;EAKAC,OAAO1B,MAAc2B,IAAuB;AAC1C,QAAI,CAAC,KAAKjD,MAAO,OAAM,IAAIkD,MAAM,mDAAA;AACjC,SAAK1C,SAAS0B,IAAIZ,MAAM2B,EAAAA;AACxB,QAAI,CAAC,KAAKnC,SAAU,MAAKA,WAAW,KAAKqC,UAAU,KAAKnD,KAAK;EAC/D;EAEAoD,SAASC,SAAsD;AAC7D,QAAI,CAAC,KAAKtC,WAAY,MAAKA,aAAa,KAAKuC,YAAYD,OAAAA;EAC3D;EAEAE,YAAYC,UAAoD;EAEhE;;EAIA,MAAcL,UAAUnD,OAA8B;AACpD,UAAM8B,WAAW,MAAM,KAAKK,SAASnC,KAAAA;AACrC,WAAO,KAAKW,SAAS;AACnB,YAAM8C,WAAW,MAAM,KAAKC,QAAQ5B,UAAU,KAAKzB,oBAAoB;AACvE,iBAAWsD,OAAOF,UAAU;AAC1B,YAAI,KAAKG,WAAWD,GAAAA,GAAM;AAExB,gBAAM,KAAKE,QAAQ/B,UAAU6B,IAAIG,aAAa;AAC9C;QACF;AACA,cAAMvB,OAAOG,KAAKqB,MAAMJ,IAAIK,QAAQ,IAAA;AACpC,cAAMC,SAAS,UAAMC,2CAAe3B,MAAM,KAAK/B,SAASgB,IAAIe,KAAKjB,IAAI,CAAA;AACrE,cAAM,KAAKxB,OAAO6B,KAChB,IAAIa,qCAAmB;UACrBV,UAAU,MAAM,KAAKM,WAAU;UAC/BK,aAAaC,KAAKC,UAAUsB,MAAAA;UAC5BrB,mBAAmB,KAAKzC,SACpB;YAAE,CAAC,KAAKA,MAAM,GAAG;cAAE0C,UAAU;cAAUC,aAAa;YAAI;UAAE,IAC1DC;QACN,CAAA,CAAA;AAEF,cAAM,KAAKoB,IAAIrC,UAAU6B,IAAIG,aAAa;MAC5C;IACF;EACF;EAEA,MAAcR,YAAYD,SAA+D;AACvF,UAAMvB,WAAW,MAAM,KAAKM,WAAU;AACtC,WAAO,KAAKzB,SAAS;AACnB,YAAM8C,WAAW,MAAM,KAAKC,QAAQ5B,UAAU,EAAA;AAC9C,iBAAW6B,OAAOF,UAAU;AAC1B,YAAI,KAAKG,WAAWD,GAAAA,GAAM;AACxB,gBAAM,KAAKE,QAAQ/B,UAAU6B,IAAIG,aAAa;AAC9C;QACF;AACA,cAAMT,QAAQX,KAAKqB,MAAMJ,IAAIK,QAAQ,IAAA,CAAA;AACrC,cAAM,KAAKG,IAAIrC,UAAU6B,IAAIG,aAAa;MAC5C;IACF;EACF;;EAGA,MAAcJ,QAAQ5B,UAAkBsC,YAAoB;AAC1D,QAAI;AACF,YAAM1C,MAAM,MAAM,KAAK5B,OAAO6B,KAC5B,IAAI0C,wCAAsB;QACxBvC;QACAwC,qBAAqB;QACrBC,iBAAiB,KAAKnE;QACtBoE,mBAAmBJ;QACnBK,uBAAuB,KAAKtE,SAAS;UAAC;YAAS4C;MACjD,CAAA,GACA;QAAE2B,aAAa,KAAK9D,MAAM+D;MAAO,CAAA;AAEnC,aAAOjD,IAAIkD,YAAY,CAAA;IACzB,SAAS7C,KAAK;AACZ,UAAI,CAAC,KAAKpB,QAAS,QAAO,CAAA;AAC1BkE,cAAQC,MAAM,iCAAiC/C,GAAAA;AAC/C,YAAM,IAAIgD,QAAQ,CAACC,MAAMC,WAAWD,GAAG,GAAA,CAAA;AACvC,aAAO,CAAA;IACT;EACF;EAEQpB,WAAWD,KAA2E;AAC5F,WAAO,CAAC,CAAC,KAAKxD,UAAU,CAACwD,IAAIf,oBAAoB,KAAKzC,MAAM;EAC9D;EAEA,MAAcgE,IAAIrC,UAAkBgC,eAAuC;AACzE,QAAI,CAACA,cAAe;AACpB,UAAM,KAAKhE,OAAO6B,KAAK,IAAIuD,uCAAqB;MAAEpD;MAAUgC;IAAc,CAAA,CAAA;EAC5E;EAEA,MAAcD,QAAQ/B,UAAkBgC,eAAuC;AAC7E,QAAI,CAACA,cAAe;AACpB,UAAM,KAAKhE,OACR6B,KAAK,IAAIwD,iDAA+B;MAAErD;MAAUgC;MAAeU,mBAAmB;IAAE,CAAA,CAAA,EACxFY,MAAM,MAAA;IAAO,CAAA;EAClB;;EAGA,MAAMC,QAAuB;AAC3B,SAAK1E,UAAU;AACf,SAAKC,MAAMA,MAAK;AAChB,UAAMmE,QAAQO,WAAW;MAAC,KAAKxE;MAAU,KAAKC;KAAW;AACzD,QAAI,KAAKhB,WAAY,MAAKD,OAAOyF,QAAO;EAC1C;AACF;","names":["SqsTransport","client","ownsClient","group","prefix","autoCreate","marker","waitTimeSeconds","visibilityTimeoutSec","taskQueueUrlResolver","resultsQueueUrlOption","handlers","Map","urlCache","running","abort","AbortController","taskLoop","resultLoop","options","SQSClient","clientConfig","taskQueueUrl","resultsQueueUrl","resolveByName","name","cached","get","url","out","send","GetQueueUrlCommand","QueueName","QueueUrl","err","created","CreateQueueCommand","set","tasksUrl","resultsUrl","opt","dispatch","task","SendMessageCommand","MessageBody","JSON","stringify","MessageAttributes","DataType","StringValue","undefined","handle","fn","Error","pollTasks","onResult","handler","pollResults","onHeartbeat","_handler","messages","receive","msg","shouldSkip","release","ReceiptHandle","parse","Body","result","runStepHandler","del","visibility","ReceiveMessageCommand","MaxNumberOfMessages","WaitTimeSeconds","VisibilityTimeout","MessageAttributeNames","abortSignal","signal","Messages","console","error","Promise","r","setTimeout","DeleteMessageCommand","ChangeMessageVisibilityCommand","catch","close","allSettled","destroy"]}

package/dist/{sqs-transport.d.ts → index.d.cts}

@@ -1,8 +1,9 @@
-import { SQSClient, type SQSClientConfig } from '@aws-sdk/client-sqs';
-import { type Heartbeat, type RemoteTask, type StepHandler, type StepResult, type Transport } from '@dudousxd/nestjs-durable-core';
+import { SQSClient, SQSClientConfig } from '@aws-sdk/client-sqs';
+import { Transport, RemoteTask, StepHandler, StepResult, Heartbeat } from '@dudousxd/nestjs-durable-core';
+
 /** Resolve a group's tasks queue URL. Sync or async; resolved once and cached. */
-export type QueueUrlResolver = (group: string) => string | Promise<string>;
-export interface SqsTransportOptions {
+type QueueUrlResolver = (group: string) => string | Promise<string>;
+interface SqsTransportOptions {
     /**
      * Reuse an existing SQS client (recommended — share the app's). When omitted, one is built
      * from `clientConfig` and destroyed on `close()`.
@@ -54,7 +55,7 @@ export interface SqsTransportOptions {
  * result is sent reappears after the visibility timeout (durable retry); a duplicate `StepResult`
  * is harmless because the engine resolves a step's `pending` exactly once by `stepId`.
  */
-export declare class SqsTransport implements Transport {
+declare class SqsTransport implements Transport {
     private readonly client;
     private readonly ownsClient;
     private readonly group?;
@@ -90,4 +91,5 @@ export declare class SqsTransport implements Transport {
     /** Stop the pollers and (if we created it) destroy the SQS client so the process can exit. */
     close(): Promise<void>;
 }
-//# sourceMappingURL=sqs-transport.d.ts.map
+
+export { type QueueUrlResolver, SqsTransport, type SqsTransportOptions };

package/dist/index.d.ts

@@ -1,2 +1,95 @@
-export * from './sqs-transport';
-//# sourceMappingURL=index.d.ts.map
+import { SQSClient, SQSClientConfig } from '@aws-sdk/client-sqs';
+import { Transport, RemoteTask, StepHandler, StepResult, Heartbeat } from '@dudousxd/nestjs-durable-core';
+
+/** Resolve a group's tasks queue URL. Sync or async; resolved once and cached. */
+type QueueUrlResolver = (group: string) => string | Promise<string>;
+interface SqsTransportOptions {
+    /**
+     * Reuse an existing SQS client (recommended — share the app's). When omitted, one is built
+     * from `clientConfig` and destroyed on `close()`.
+     */
+    client?: SQSClient;
+    /** Used only when `client` is omitted — e.g. `{ region, endpoint, credentials }` for ElasticMQ. */
+    clientConfig?: SQSClientConfig;
+    /** The worker group this instance serves. Required to register `handle()` consumers. */
+    group?: string;
+    /**
+     * Map a group → an **existing** tasks queue URL. This is the seam for reusing queues you already
+     * have (e.g. `extraction`, `ingestion`) instead of letting the transport mint its own. When
+     * omitted, the transport resolves `${prefix}-tasks-${group}` by name (and creates it if
+     * `autoCreate`). Return value is awaited and cached.
+     */
+    taskQueueUrl?: QueueUrlResolver;
+    /**
+     * The results queue URL (worker → engine). A literal URL or a resolver. When omitted, the
+     * transport resolves `${prefix}-results` by name (and creates it if `autoCreate`).
+     */
+    resultsQueueUrl?: string | (() => string | Promise<string>);
+    /** Prefix for the fallback `${prefix}-tasks-*` / `${prefix}-results` names. Default `durable`. */
+    prefix?: string;
+    /**
+     * Create the fallback-named queues on first use if they don't exist (handy for ElasticMQ / dev).
+     * Ignored for queues you supply via `taskQueueUrl` / `resultsQueueUrl` — those must already exist.
+     */
+    autoCreate?: boolean;
+    /**
+     * When set, durable messages carry this SQS message attribute and pollers **skip** messages
+     * that lack it (releasing them for a co-resident legacy consumer). Enables sharing one physical
+     * queue with a non-durable consumer — but only if that consumer likewise ignores tagged messages.
+     * Leave unset for the clean case: a queue dedicated to durable traffic.
+     */
+    marker?: string;
+    /** SQS receive long-poll seconds (0–20). Default 20. */
+    waitTimeSeconds?: number;
+    /** Visibility timeout for an in-flight task, in seconds. Must exceed the step's runtime. Default 60. */
+    visibilityTimeoutSec?: number;
+}
+/**
+ * An SQS-backed `Transport` — dispatch remote steps over Amazon SQS (or ElasticMQ). Unlike the
+ * BullMQ transport, it does **not** mint its own queue names: pass `taskQueueUrl` / `resultsQueueUrl`
+ * to ride the queues you already operate. The wire payload is the documented `RemoteTask` /
+ * `StepResult` JSON, so non-Node workers on the same queues interoperate.
+ *
+ * Run one instance engine-side (calls `onResult`, dispatches tasks) and one per worker process
+ * (calls `handle()` for its group). SQS is at-least-once: a task whose process crashes before the
+ * result is sent reappears after the visibility timeout (durable retry); a duplicate `StepResult`
+ * is harmless because the engine resolves a step's `pending` exactly once by `stepId`.
+ */
+declare class SqsTransport implements Transport {
+    private readonly client;
+    private readonly ownsClient;
+    private readonly group?;
+    private readonly prefix;
+    private readonly autoCreate;
+    private readonly marker?;
+    private readonly waitTimeSeconds;
+    private readonly visibilityTimeoutSec;
+    private readonly taskQueueUrlResolver?;
+    private readonly resultsQueueUrlOption?;
+    private readonly handlers;
+    private readonly urlCache;
+    private running;
+    private readonly abort;
+    private taskLoop?;
+    private resultLoop?;
+    constructor(options: SqsTransportOptions);
+    private resolveByName;
+    private tasksUrl;
+    private resultsUrl;
+    dispatch(task: RemoteTask): Promise<void>;
+    /** Register a step handler (worker side). Starts the group's task consumer on first call. */
+    handle(name: string, fn: StepHandler): void;
+    onResult(handler: (result: StepResult) => Promise<void>): void;
+    onHeartbeat(_handler: (beat: Heartbeat) => Promise<void>): void;
+    private pollTasks;
+    private pollResults;
+    /** One long-poll receive. Swallows the abort thrown on `close()`; logs and backs off otherwise. */
+    private receive;
+    private shouldSkip;
+    private del;
+    private release;
+    /** Stop the pollers and (if we created it) destroy the SQS client so the process can exit. */
+    close(): Promise<void>;
+}
+
+export { type QueueUrlResolver, SqsTransport, type SqsTransportOptions };

package/dist/index.js

@@ -1,18 +1,193 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+
+// src/sqs-transport.ts
+import { ChangeMessageVisibilityCommand, CreateQueueCommand, DeleteMessageCommand, GetQueueUrlCommand, ReceiveMessageCommand, SendMessageCommand, SQSClient } from "@aws-sdk/client-sqs";
+import { runStepHandler } from "@dudousxd/nestjs-durable-core";
+var SqsTransport = class {
+  static {
+    __name(this, "SqsTransport");
+  }
+  client;
+  ownsClient;
+  group;
+  prefix;
+  autoCreate;
+  marker;
+  waitTimeSeconds;
+  visibilityTimeoutSec;
+  taskQueueUrlResolver;
+  resultsQueueUrlOption;
+  handlers = /* @__PURE__ */ new Map();
+  urlCache = /* @__PURE__ */ new Map();
+  running = true;
+  abort = new AbortController();
+  taskLoop;
+  resultLoop;
+  constructor(options) {
+    this.client = options.client ?? new SQSClient(options.clientConfig ?? {});
+    this.ownsClient = !options.client;
+    this.group = options.group;
+    this.prefix = options.prefix ?? "durable";
+    this.autoCreate = options.autoCreate ?? false;
+    this.marker = options.marker;
+    this.waitTimeSeconds = options.waitTimeSeconds ?? 20;
+    this.visibilityTimeoutSec = options.visibilityTimeoutSec ?? 60;
+    this.taskQueueUrlResolver = options.taskQueueUrl;
+    this.resultsQueueUrlOption = options.resultsQueueUrl;
+  }
+  // ── queue resolution ───────────────────────────────────────────────────────────────────────
+  async resolveByName(name) {
+    const cached = this.urlCache.get(name);
+    if (cached) return cached;
+    let url;
+    try {
+      const out = await this.client.send(new GetQueueUrlCommand({
+        QueueName: name
+      }));
+      url = out.QueueUrl;
+    } catch (err) {
+      if (this.autoCreate) {
+        const created = await this.client.send(new CreateQueueCommand({
+          QueueName: name
+        }));
+        url = created.QueueUrl;
+      } else {
+        throw err;
+      }
     }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+    this.urlCache.set(name, url);
+    return url;
+  }
+  async tasksUrl(group) {
+    if (this.taskQueueUrlResolver) return this.taskQueueUrlResolver(group);
+    return this.resolveByName(`${this.prefix}-tasks-${group}`);
+  }
+  async resultsUrl() {
+    const opt = this.resultsQueueUrlOption;
+    if (typeof opt === "string") return opt;
+    if (typeof opt === "function") return opt();
+    return this.resolveByName(`${this.prefix}-results`);
+  }
+  // ── engine → worker ────────────────────────────────────────────────────────────────────────
+  async dispatch(task) {
+    const QueueUrl = await this.tasksUrl(task.group);
+    await this.client.send(new SendMessageCommand({
+      QueueUrl,
+      MessageBody: JSON.stringify(task),
+      MessageAttributes: this.marker ? {
+        [this.marker]: {
+          DataType: "String",
+          StringValue: "1"
+        }
+      } : void 0
+    }));
+  }
+  // ── worker → engine ────────────────────────────────────────────────────────────────────────
+  /** Register a step handler (worker side). Starts the group's task consumer on first call. */
+  handle(name, fn) {
+    if (!this.group) throw new Error("SqsTransport needs a `group` to register handlers");
+    this.handlers.set(name, fn);
+    if (!this.taskLoop) this.taskLoop = this.pollTasks(this.group);
+  }
+  onResult(handler) {
+    if (!this.resultLoop) this.resultLoop = this.pollResults(handler);
+  }
+  onHeartbeat(_handler) {
+  }
+  // ── polling loops ──────────────────────────────────────────────────────────────────────────
+  async pollTasks(group) {
+    const QueueUrl = await this.tasksUrl(group);
+    while (this.running) {
+      const messages = await this.receive(QueueUrl, this.visibilityTimeoutSec);
+      for (const msg of messages) {
+        if (this.shouldSkip(msg)) {
+          await this.release(QueueUrl, msg.ReceiptHandle);
+          continue;
+        }
+        const task = JSON.parse(msg.Body ?? "{}");
+        const result = await runStepHandler(task, this.handlers.get(task.name));
+        await this.client.send(new SendMessageCommand({
+          QueueUrl: await this.resultsUrl(),
+          MessageBody: JSON.stringify(result),
+          MessageAttributes: this.marker ? {
+            [this.marker]: {
+              DataType: "String",
+              StringValue: "1"
+            }
+          } : void 0
+        }));
+        await this.del(QueueUrl, msg.ReceiptHandle);
+      }
+    }
+  }
+  async pollResults(handler) {
+    const QueueUrl = await this.resultsUrl();
+    while (this.running) {
+      const messages = await this.receive(QueueUrl, 30);
+      for (const msg of messages) {
+        if (this.shouldSkip(msg)) {
+          await this.release(QueueUrl, msg.ReceiptHandle);
+          continue;
+        }
+        await handler(JSON.parse(msg.Body ?? "{}"));
+        await this.del(QueueUrl, msg.ReceiptHandle);
+      }
+    }
+  }
+  /** One long-poll receive. Swallows the abort thrown on `close()`; logs and backs off otherwise. */
+  async receive(QueueUrl, visibility) {
+    try {
+      const out = await this.client.send(new ReceiveMessageCommand({
+        QueueUrl,
+        MaxNumberOfMessages: 10,
+        WaitTimeSeconds: this.waitTimeSeconds,
+        VisibilityTimeout: visibility,
+        MessageAttributeNames: this.marker ? [
+          "All"
+        ] : void 0
+      }), {
+        abortSignal: this.abort.signal
+      });
+      return out.Messages ?? [];
+    } catch (err) {
+      if (!this.running) return [];
+      console.error("[SqsTransport] receive failed", err);
+      await new Promise((r) => setTimeout(r, 1e3));
+      return [];
+    }
+  }
+  shouldSkip(msg) {
+    return !!this.marker && !msg.MessageAttributes?.[this.marker];
+  }
+  async del(QueueUrl, ReceiptHandle) {
+    if (!ReceiptHandle) return;
+    await this.client.send(new DeleteMessageCommand({
+      QueueUrl,
+      ReceiptHandle
+    }));
+  }
+  async release(QueueUrl, ReceiptHandle) {
+    if (!ReceiptHandle) return;
+    await this.client.send(new ChangeMessageVisibilityCommand({
+      QueueUrl,
+      ReceiptHandle,
+      VisibilityTimeout: 0
+    })).catch(() => {
+    });
+  }
+  /** Stop the pollers and (if we created it) destroy the SQS client so the process can exit. */
+  async close() {
+    this.running = false;
+    this.abort.abort();
+    await Promise.allSettled([
+      this.taskLoop,
+      this.resultLoop
+    ]);
+    if (this.ownsClient) this.client.destroy();
+  }
+};
+export {
+  SqsTransport
 };
-Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./sqs-transport"), exports);
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,kDAAgC"}
+{"version":3,"sources":["../src/sqs-transport.ts"],"sourcesContent":["import {\n  ChangeMessageVisibilityCommand,\n  CreateQueueCommand,\n  DeleteMessageCommand,\n  GetQueueUrlCommand,\n  ReceiveMessageCommand,\n  SendMessageCommand,\n  SQSClient,\n  type SQSClientConfig,\n} from '@aws-sdk/client-sqs';\nimport {\n  type Heartbeat,\n  type RemoteTask,\n  type StepHandler,\n  type StepResult,\n  type Transport,\n  runStepHandler,\n} from '@dudousxd/nestjs-durable-core';\n\n/** Resolve a group's tasks queue URL. Sync or async; resolved once and cached. */\nexport type QueueUrlResolver = (group: string) => string | Promise<string>;\n\nexport interface SqsTransportOptions {\n  /**\n   * Reuse an existing SQS client (recommended — share the app's). When omitted, one is built\n   * from `clientConfig` and destroyed on `close()`.\n   */\n  client?: SQSClient;\n  /** Used only when `client` is omitted — e.g. `{ region, endpoint, credentials }` for ElasticMQ. */\n  clientConfig?: SQSClientConfig;\n\n  /** The worker group this instance serves. Required to register `handle()` consumers. */\n  group?: string;\n\n  /**\n   * Map a group → an **existing** tasks queue URL. This is the seam for reusing queues you already\n   * have (e.g. `extraction`, `ingestion`) instead of letting the transport mint its own. When\n   * omitted, the transport resolves `${prefix}-tasks-${group}` by name (and creates it if\n   * `autoCreate`). Return value is awaited and cached.\n   */\n  taskQueueUrl?: QueueUrlResolver;\n  /**\n   * The results queue URL (worker → engine). A literal URL or a resolver. When omitted, the\n   * transport resolves `${prefix}-results` by name (and creates it if `autoCreate`).\n   */\n  resultsQueueUrl?: string | (() => string | Promise<string>);\n\n  /** Prefix for the fallback `${prefix}-tasks-*` / `${prefix}-results` names. Default `durable`. */\n  prefix?: string;\n  /**\n   * Create the fallback-named queues on first use if they don't exist (handy for ElasticMQ / dev).\n   * Ignored for queues you supply via `taskQueueUrl` / `resultsQueueUrl` — those must already exist.\n   */\n  autoCreate?: boolean;\n\n  /**\n   * When set, durable messages carry this SQS message attribute and pollers **skip** messages\n   * that lack it (releasing them for a co-resident legacy consumer). Enables sharing one physical\n   * queue with a non-durable consumer — but only if that consumer likewise ignores tagged messages.\n   * Leave unset for the clean case: a queue dedicated to durable traffic.\n   */\n  marker?: string;\n\n  /** SQS receive long-poll seconds (0–20). Default 20. */\n  waitTimeSeconds?: number;\n  /** Visibility timeout for an in-flight task, in seconds. Must exceed the step's runtime. Default 60. */\n  visibilityTimeoutSec?: number;\n}\n\n/**\n * An SQS-backed `Transport` — dispatch remote steps over Amazon SQS (or ElasticMQ). Unlike the\n * BullMQ transport, it does **not** mint its own queue names: pass `taskQueueUrl` / `resultsQueueUrl`\n * to ride the queues you already operate. The wire payload is the documented `RemoteTask` /\n * `StepResult` JSON, so non-Node workers on the same queues interoperate.\n *\n * Run one instance engine-side (calls `onResult`, dispatches tasks) and one per worker process\n * (calls `handle()` for its group). SQS is at-least-once: a task whose process crashes before the\n * result is sent reappears after the visibility timeout (durable retry); a duplicate `StepResult`\n * is harmless because the engine resolves a step's `pending` exactly once by `stepId`.\n */\nexport class SqsTransport implements Transport {\n  private readonly client: SQSClient;\n  private readonly ownsClient: boolean;\n  private readonly group?: string | undefined;\n  private readonly prefix: string;\n  private readonly autoCreate: boolean;\n  private readonly marker?: string | undefined;\n  private readonly waitTimeSeconds: number;\n  private readonly visibilityTimeoutSec: number;\n  private readonly taskQueueUrlResolver?: QueueUrlResolver | undefined;\n  private readonly resultsQueueUrlOption?: string | (() => string | Promise<string>) | undefined;\n\n  private readonly handlers = new Map<string, StepHandler>();\n  private readonly urlCache = new Map<string, string>();\n  private running = true;\n  private readonly abort = new AbortController();\n  private taskLoop?: Promise<void>;\n  private resultLoop?: Promise<void>;\n\n  constructor(options: SqsTransportOptions) {\n    this.client = options.client ?? new SQSClient(options.clientConfig ?? {});\n    this.ownsClient = !options.client;\n    this.group = options.group;\n    this.prefix = options.prefix ?? 'durable';\n    this.autoCreate = options.autoCreate ?? false;\n    this.marker = options.marker;\n    this.waitTimeSeconds = options.waitTimeSeconds ?? 20;\n    this.visibilityTimeoutSec = options.visibilityTimeoutSec ?? 60;\n    this.taskQueueUrlResolver = options.taskQueueUrl;\n    this.resultsQueueUrlOption = options.resultsQueueUrl;\n  }\n\n  // ── queue resolution ───────────────────────────────────────────────────────────────────────\n\n  private async resolveByName(name: string): Promise<string> {\n    const cached = this.urlCache.get(name);\n    if (cached) return cached;\n    let url: string;\n    try {\n      const out = await this.client.send(new GetQueueUrlCommand({ QueueName: name }));\n      url = out.QueueUrl as string;\n    } catch (err) {\n      if (this.autoCreate) {\n        const created = await this.client.send(new CreateQueueCommand({ QueueName: name }));\n        url = created.QueueUrl as string;\n      } else {\n        throw err;\n      }\n    }\n    this.urlCache.set(name, url);\n    return url;\n  }\n\n  private async tasksUrl(group: string): Promise<string> {\n    if (this.taskQueueUrlResolver) return this.taskQueueUrlResolver(group);\n    return this.resolveByName(`${this.prefix}-tasks-${group}`);\n  }\n\n  private async resultsUrl(): Promise<string> {\n    const opt = this.resultsQueueUrlOption;\n    if (typeof opt === 'string') return opt;\n    if (typeof opt === 'function') return opt();\n    return this.resolveByName(`${this.prefix}-results`);\n  }\n\n  // ── engine → worker ────────────────────────────────────────────────────────────────────────\n\n  async dispatch(task: RemoteTask): Promise<void> {\n    const QueueUrl = await this.tasksUrl(task.group);\n    await this.client.send(\n      new SendMessageCommand({\n        QueueUrl,\n        MessageBody: JSON.stringify(task),\n        MessageAttributes: this.marker\n          ? { [this.marker]: { DataType: 'String', StringValue: '1' } }\n          : undefined,\n      }),\n    );\n  }\n\n  // ── worker → engine ────────────────────────────────────────────────────────────────────────\n\n  /** Register a step handler (worker side). Starts the group's task consumer on first call. */\n  handle(name: string, fn: StepHandler): void {\n    if (!this.group) throw new Error('SqsTransport needs a `group` to register handlers');\n    this.handlers.set(name, fn);\n    if (!this.taskLoop) this.taskLoop = this.pollTasks(this.group);\n  }\n\n  onResult(handler: (result: StepResult) => Promise<void>): void {\n    if (!this.resultLoop) this.resultLoop = this.pollResults(handler);\n  }\n\n  onHeartbeat(_handler: (beat: Heartbeat) => Promise<void>): void {\n    // Not modelled over SQS; the queue's visibility-timeout redelivery is the liveness mechanism.\n  }\n\n  // ── polling loops ──────────────────────────────────────────────────────────────────────────\n\n  private async pollTasks(group: string): Promise<void> {\n    const QueueUrl = await this.tasksUrl(group);\n    while (this.running) {\n      const messages = await this.receive(QueueUrl, this.visibilityTimeoutSec);\n      for (const msg of messages) {\n        if (this.shouldSkip(msg)) {\n          // Not ours (shared queue): release immediately so the legacy consumer can take it.\n          await this.release(QueueUrl, msg.ReceiptHandle);\n          continue;\n        }\n        const task = JSON.parse(msg.Body ?? '{}') as RemoteTask;\n        const result = await runStepHandler(task, this.handlers.get(task.name));\n        await this.client.send(\n          new SendMessageCommand({\n            QueueUrl: await this.resultsUrl(),\n            MessageBody: JSON.stringify(result),\n            MessageAttributes: this.marker\n              ? { [this.marker]: { DataType: 'String', StringValue: '1' } }\n              : undefined,\n          }),\n        );\n        await this.del(QueueUrl, msg.ReceiptHandle);\n      }\n    }\n  }\n\n  private async pollResults(handler: (result: StepResult) => Promise<void>): Promise<void> {\n    const QueueUrl = await this.resultsUrl();\n    while (this.running) {\n      const messages = await this.receive(QueueUrl, 30);\n      for (const msg of messages) {\n        if (this.shouldSkip(msg)) {\n          await this.release(QueueUrl, msg.ReceiptHandle);\n          continue;\n        }\n        await handler(JSON.parse(msg.Body ?? '{}') as StepResult);\n        await this.del(QueueUrl, msg.ReceiptHandle);\n      }\n    }\n  }\n\n  /** One long-poll receive. Swallows the abort thrown on `close()`; logs and backs off otherwise. */\n  private async receive(QueueUrl: string, visibility: number) {\n    try {\n      const out = await this.client.send(\n        new ReceiveMessageCommand({\n          QueueUrl,\n          MaxNumberOfMessages: 10,\n          WaitTimeSeconds: this.waitTimeSeconds,\n          VisibilityTimeout: visibility,\n          MessageAttributeNames: this.marker ? ['All'] : undefined,\n        }),\n        { abortSignal: this.abort.signal },\n      );\n      return out.Messages ?? [];\n    } catch (err) {\n      if (!this.running) return [];\n      console.error('[SqsTransport] receive failed', err);\n      await new Promise((r) => setTimeout(r, 1000));\n      return [];\n    }\n  }\n\n  private shouldSkip(msg: { MessageAttributes?: Record<string, unknown> | undefined }): boolean {\n    return !!this.marker && !msg.MessageAttributes?.[this.marker];\n  }\n\n  private async del(QueueUrl: string, ReceiptHandle?: string): Promise<void> {\n    if (!ReceiptHandle) return;\n    await this.client.send(new DeleteMessageCommand({ QueueUrl, ReceiptHandle }));\n  }\n\n  private async release(QueueUrl: string, ReceiptHandle?: string): Promise<void> {\n    if (!ReceiptHandle) return;\n    await this.client\n      .send(new ChangeMessageVisibilityCommand({ QueueUrl, ReceiptHandle, VisibilityTimeout: 0 }))\n      .catch(() => {});\n  }\n\n  /** Stop the pollers and (if we created it) destroy the SQS client so the process can exit. */\n  async close(): Promise<void> {\n    this.running = false;\n    this.abort.abort();\n    await Promise.allSettled([this.taskLoop, this.resultLoop]);\n    if (this.ownsClient) this.client.destroy();\n  }\n}\n"],"mappings":";;;;AAAA,SACEA,gCACAC,oBACAC,sBACAC,oBACAC,uBACAC,oBACAC,iBAEK;AACP,SAMEC,sBACK;AA+DA,IAAMC,eAAN,MAAMA;EAhFb,OAgFaA;;;EACMC;EACAC;EACAC;EACAC;EACAC;EACAC;EACAC;EACAC;EACAC;EACAC;EAEAC,WAAW,oBAAIC,IAAAA;EACfC,WAAW,oBAAID,IAAAA;EACxBE,UAAU;EACDC,QAAQ,IAAIC,gBAAAA;EACrBC;EACAC;EAER,YAAYC,SAA8B;AACxC,SAAKlB,SAASkB,QAAQlB,UAAU,IAAImB,UAAUD,QAAQE,gBAAgB,CAAC,CAAA;AACvE,SAAKnB,aAAa,CAACiB,QAAQlB;AAC3B,SAAKE,QAAQgB,QAAQhB;AACrB,SAAKC,SAASe,QAAQf,UAAU;AAChC,SAAKC,aAAac,QAAQd,cAAc;AACxC,SAAKC,SAASa,QAAQb;AACtB,SAAKC,kBAAkBY,QAAQZ,mBAAmB;AAClD,SAAKC,uBAAuBW,QAAQX,wBAAwB;AAC5D,SAAKC,uBAAuBU,QAAQG;AACpC,SAAKZ,wBAAwBS,QAAQI;EACvC;;EAIA,MAAcC,cAAcC,MAA+B;AACzD,UAAMC,SAAS,KAAKb,SAASc,IAAIF,IAAAA;AACjC,QAAIC,OAAQ,QAAOA;AACnB,QAAIE;AACJ,QAAI;AACF,YAAMC,MAAM,MAAM,KAAK5B,OAAO6B,KAAK,IAAIC,mBAAmB;QAAEC,WAAWP;MAAK,CAAA,CAAA;AAC5EG,YAAMC,IAAII;IACZ,SAASC,KAAK;AACZ,UAAI,KAAK7B,YAAY;AACnB,cAAM8B,UAAU,MAAM,KAAKlC,OAAO6B,KAAK,IAAIM,mBAAmB;UAAEJ,WAAWP;QAAK,CAAA,CAAA;AAChFG,cAAMO,QAAQF;MAChB,OAAO;AACL,cAAMC;MACR;IACF;AACA,SAAKrB,SAASwB,IAAIZ,MAAMG,GAAAA;AACxB,WAAOA;EACT;EAEA,MAAcU,SAASnC,OAAgC;AACrD,QAAI,KAAKM,qBAAsB,QAAO,KAAKA,qBAAqBN,KAAAA;AAChE,WAAO,KAAKqB,cAAc,GAAG,KAAKpB,MAAM,UAAUD,KAAAA,EAAO;EAC3D;EAEA,MAAcoC,aAA8B;AAC1C,UAAMC,MAAM,KAAK9B;AACjB,QAAI,OAAO8B,QAAQ,SAAU,QAAOA;AACpC,QAAI,OAAOA,QAAQ,WAAY,QAAOA,IAAAA;AACtC,WAAO,KAAKhB,cAAc,GAAG,KAAKpB,MAAM,UAAU;EACpD;;EAIA,MAAMqC,SAASC,MAAiC;AAC9C,UAAMT,WAAW,MAAM,KAAKK,SAASI,KAAKvC,KAAK;AAC/C,UAAM,KAAKF,OAAO6B,KAChB,IAAIa,mBAAmB;MACrBV;MACAW,aAAaC,KAAKC,UAAUJ,IAAAA;MAC5BK,mBAAmB,KAAKzC,SACpB;QAAE,CAAC,KAAKA,MAAM,GAAG;UAAE0C,UAAU;UAAUC,aAAa;QAAI;MAAE,IAC1DC;IACN,CAAA,CAAA;EAEJ;;;EAKAC,OAAO1B,MAAc2B,IAAuB;AAC1C,QAAI,CAAC,KAAKjD,MAAO,OAAM,IAAIkD,MAAM,mDAAA;AACjC,SAAK1C,SAAS0B,IAAIZ,MAAM2B,EAAAA;AACxB,QAAI,CAAC,KAAKnC,SAAU,MAAKA,WAAW,KAAKqC,UAAU,KAAKnD,KAAK;EAC/D;EAEAoD,SAASC,SAAsD;AAC7D,QAAI,CAAC,KAAKtC,WAAY,MAAKA,aAAa,KAAKuC,YAAYD,OAAAA;EAC3D;EAEAE,YAAYC,UAAoD;EAEhE;;EAIA,MAAcL,UAAUnD,OAA8B;AACpD,UAAM8B,WAAW,MAAM,KAAKK,SAASnC,KAAAA;AACrC,WAAO,KAAKW,SAAS;AACnB,YAAM8C,WAAW,MAAM,KAAKC,QAAQ5B,UAAU,KAAKzB,oBAAoB;AACvE,iBAAWsD,OAAOF,UAAU;AAC1B,YAAI,KAAKG,WAAWD,GAAAA,GAAM;AAExB,gBAAM,KAAKE,QAAQ/B,UAAU6B,IAAIG,aAAa;AAC9C;QACF;AACA,cAAMvB,OAAOG,KAAKqB,MAAMJ,IAAIK,QAAQ,IAAA;AACpC,cAAMC,SAAS,MAAMC,eAAe3B,MAAM,KAAK/B,SAASgB,IAAIe,KAAKjB,IAAI,CAAA;AACrE,cAAM,KAAKxB,OAAO6B,KAChB,IAAIa,mBAAmB;UACrBV,UAAU,MAAM,KAAKM,WAAU;UAC/BK,aAAaC,KAAKC,UAAUsB,MAAAA;UAC5BrB,mBAAmB,KAAKzC,SACpB;YAAE,CAAC,KAAKA,MAAM,GAAG;cAAE0C,UAAU;cAAUC,aAAa;YAAI;UAAE,IAC1DC;QACN,CAAA,CAAA;AAEF,cAAM,KAAKoB,IAAIrC,UAAU6B,IAAIG,aAAa;MAC5C;IACF;EACF;EAEA,MAAcR,YAAYD,SAA+D;AACvF,UAAMvB,WAAW,MAAM,KAAKM,WAAU;AACtC,WAAO,KAAKzB,SAAS;AACnB,YAAM8C,WAAW,MAAM,KAAKC,QAAQ5B,UAAU,EAAA;AAC9C,iBAAW6B,OAAOF,UAAU;AAC1B,YAAI,KAAKG,WAAWD,GAAAA,GAAM;AACxB,gBAAM,KAAKE,QAAQ/B,UAAU6B,IAAIG,aAAa;AAC9C;QACF;AACA,cAAMT,QAAQX,KAAKqB,MAAMJ,IAAIK,QAAQ,IAAA,CAAA;AACrC,cAAM,KAAKG,IAAIrC,UAAU6B,IAAIG,aAAa;MAC5C;IACF;EACF;;EAGA,MAAcJ,QAAQ5B,UAAkBsC,YAAoB;AAC1D,QAAI;AACF,YAAM1C,MAAM,MAAM,KAAK5B,OAAO6B,KAC5B,IAAI0C,sBAAsB;QACxBvC;QACAwC,qBAAqB;QACrBC,iBAAiB,KAAKnE;QACtBoE,mBAAmBJ;QACnBK,uBAAuB,KAAKtE,SAAS;UAAC;YAAS4C;MACjD,CAAA,GACA;QAAE2B,aAAa,KAAK9D,MAAM+D;MAAO,CAAA;AAEnC,aAAOjD,IAAIkD,YAAY,CAAA;IACzB,SAAS7C,KAAK;AACZ,UAAI,CAAC,KAAKpB,QAAS,QAAO,CAAA;AAC1BkE,cAAQC,MAAM,iCAAiC/C,GAAAA;AAC/C,YAAM,IAAIgD,QAAQ,CAACC,MAAMC,WAAWD,GAAG,GAAA,CAAA;AACvC,aAAO,CAAA;IACT;EACF;EAEQpB,WAAWD,KAA2E;AAC5F,WAAO,CAAC,CAAC,KAAKxD,UAAU,CAACwD,IAAIf,oBAAoB,KAAKzC,MAAM;EAC9D;EAEA,MAAcgE,IAAIrC,UAAkBgC,eAAuC;AACzE,QAAI,CAACA,cAAe;AACpB,UAAM,KAAKhE,OAAO6B,KAAK,IAAIuD,qBAAqB;MAAEpD;MAAUgC;IAAc,CAAA,CAAA;EAC5E;EAEA,MAAcD,QAAQ/B,UAAkBgC,eAAuC;AAC7E,QAAI,CAACA,cAAe;AACpB,UAAM,KAAKhE,OACR6B,KAAK,IAAIwD,+BAA+B;MAAErD;MAAUgC;MAAeU,mBAAmB;IAAE,CAAA,CAAA,EACxFY,MAAM,MAAA;IAAO,CAAA;EAClB;;EAGA,MAAMC,QAAuB;AAC3B,SAAK1E,UAAU;AACf,SAAKC,MAAMA,MAAK;AAChB,UAAMmE,QAAQO,WAAW;MAAC,KAAKxE;MAAU,KAAKC;KAAW;AACzD,QAAI,KAAKhB,WAAY,MAAKD,OAAOyF,QAAO;EAC1C;AACF;","names":["ChangeMessageVisibilityCommand","CreateQueueCommand","DeleteMessageCommand","GetQueueUrlCommand","ReceiveMessageCommand","SendMessageCommand","SQSClient","runStepHandler","SqsTransport","client","ownsClient","group","prefix","autoCreate","marker","waitTimeSeconds","visibilityTimeoutSec","taskQueueUrlResolver","resultsQueueUrlOption","handlers","Map","urlCache","running","abort","AbortController","taskLoop","resultLoop","options","SQSClient","clientConfig","taskQueueUrl","resultsQueueUrl","resolveByName","name","cached","get","url","out","send","GetQueueUrlCommand","QueueName","QueueUrl","err","created","CreateQueueCommand","set","tasksUrl","resultsUrl","opt","dispatch","task","SendMessageCommand","MessageBody","JSON","stringify","MessageAttributes","DataType","StringValue","undefined","handle","fn","Error","pollTasks","onResult","handler","pollResults","onHeartbeat","_handler","messages","receive","msg","shouldSkip","release","ReceiptHandle","parse","Body","result","runStepHandler","del","visibility","ReceiveMessageCommand","MaxNumberOfMessages","WaitTimeSeconds","VisibilityTimeout","MessageAttributeNames","abortSignal","signal","Messages","console","error","Promise","r","setTimeout","DeleteMessageCommand","ChangeMessageVisibilityCommand","catch","close","allSettled","destroy"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dudousxd/nestjs-durable-transport-sqs",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Amazon SQS Transport for nestjs-durable — dispatch remote steps over existing SQS queues (ElasticMQ-compatible)",
   "license": "MIT",
   "author": "Davide Carvalho",
@@ -9,9 +9,9 @@
     "url": "https://github.com/DavideCarvalho/nestjs-durable.git",
     "directory": "packages/transport-sqs"
   },
-  "type": "commonjs",
-  "main": "dist/index.js",
-  "types": "dist/index.d.ts",
+  "type": "module",
+  "main": "./dist/index.cjs",
+  "types": "./dist/index.d.ts",
   "files": [
     "dist"
   ],
@@ -21,12 +21,27 @@
   },
   "devDependencies": {
     "@aws-sdk/client-sqs": "3.1034.0",
+    "testcontainers": "10.28.0",
     "typescript": "5.9.3",
     "zod": "3.25.76",
-    "@dudousxd/nestjs-durable-core": "^0.1.0"
+    "tsup": "8.3.5",
+    "@dudousxd/nestjs-durable-core": "^0.28.0"
+  },
+  "module": "./dist/index.js",
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      },
+      "require": {
+        "types": "./dist/index.d.cts",
+        "default": "./dist/index.cjs"
+      }
+    }
   },
   "scripts": {
-    "build": "tsc -p tsconfig.json",
+    "build": "tsup",
     "typecheck": "tsc -p tsconfig.json --noEmit"
   }
 }

package/dist/index.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,iBAAiB,CAAC"}

package/dist/sqs-transport.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"sqs-transport.d.ts","sourceRoot":"","sources":["../src/sqs-transport.ts"],"names":[],"mappings":"AAAA,OAAO,EAOL,SAAS,EACT,KAAK,eAAe,EACrB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,KAAK,SAAS,EACd,KAAK,UAAU,EACf,KAAK,WAAW,EAChB,KAAK,UAAU,EACf,KAAK,SAAS,EAEf,MAAM,+BAA+B,CAAC;AAEvC,kFAAkF;AAClF,MAAM,MAAM,gBAAgB,GAAG,CAAC,KAAK,EAAE,MAAM,KAAK,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;AAE3E,MAAM,WAAW,mBAAmB;IAClC;;;OAGG;IACH,MAAM,CAAC,EAAE,SAAS,CAAC;IACnB,mGAAmG;IACnG,YAAY,CAAC,EAAE,eAAe,CAAC;IAE/B,wFAAwF;IACxF,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;;;;OAKG;IACH,YAAY,CAAC,EAAE,gBAAgB,CAAC;IAChC;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,GAAG,CAAC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;IAE5D,kGAAkG;IAClG,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;OAGG;IACH,UAAU,CAAC,EAAE,OAAO,CAAC;IAErB;;;;;OAKG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB,wDAAwD;IACxD,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,wGAAwG;IACxG,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B;AAED;;;;;;;;;;GAUG;AACH,qBAAa,YAAa,YAAW,SAAS;IAC5C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAY;IACnC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAU;IACrC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAU;IACrC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAS;IACzC,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAS;IAC9C,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAmB;IACzD,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAA4C;IAEnF,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAkC;IAC3D,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA6B;IACtD,OAAO,CAAC,OAAO,CAAQ;IACvB,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAyB;IAC/C,OAAO,CAAC,QAAQ,CAAC,CAAgB;IACjC,OAAO,CAAC,UAAU,CAAC,CAAgB;gBAEvB,OAAO,EAAE,mBAAmB;YAe1B,aAAa;YAmBb,QAAQ;YAKR,UAAU;IASlB,QAAQ,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAe/C,6FAA6F;IAC7F,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,WAAW,GAAG,IAAI;IAM3C,QAAQ,CAAC,OAAO,EAAE,CAAC,MAAM,EAAE,UAAU,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI;IAI9D,WAAW,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,SAAS,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI;YAMjD,SAAS;YA0BT,WAAW;IAezB,mGAAmG;YACrF,OAAO;IAqBrB,OAAO,CAAC,UAAU;YAIJ,GAAG;YAKH,OAAO;IAOrB,8FAA8F;IACxF,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAM7B"}

package/dist/sqs-transport.js

@@ -1,191 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.SqsTransport = void 0;
-const client_sqs_1 = require("@aws-sdk/client-sqs");
-const nestjs_durable_core_1 = require("@dudousxd/nestjs-durable-core");
-/**
- * An SQS-backed `Transport` — dispatch remote steps over Amazon SQS (or ElasticMQ). Unlike the
- * BullMQ transport, it does **not** mint its own queue names: pass `taskQueueUrl` / `resultsQueueUrl`
- * to ride the queues you already operate. The wire payload is the documented `RemoteTask` /
- * `StepResult` JSON, so non-Node workers on the same queues interoperate.
- *
- * Run one instance engine-side (calls `onResult`, dispatches tasks) and one per worker process
- * (calls `handle()` for its group). SQS is at-least-once: a task whose process crashes before the
- * result is sent reappears after the visibility timeout (durable retry); a duplicate `StepResult`
- * is harmless because the engine resolves a step's `pending` exactly once by `stepId`.
- */
-class SqsTransport {
-    client;
-    ownsClient;
-    group;
-    prefix;
-    autoCreate;
-    marker;
-    waitTimeSeconds;
-    visibilityTimeoutSec;
-    taskQueueUrlResolver;
-    resultsQueueUrlOption;
-    handlers = new Map();
-    urlCache = new Map();
-    running = true;
-    abort = new AbortController();
-    taskLoop;
-    resultLoop;
-    constructor(options) {
-        this.client = options.client ?? new client_sqs_1.SQSClient(options.clientConfig ?? {});
-        this.ownsClient = !options.client;
-        this.group = options.group;
-        this.prefix = options.prefix ?? 'durable';
-        this.autoCreate = options.autoCreate ?? false;
-        this.marker = options.marker;
-        this.waitTimeSeconds = options.waitTimeSeconds ?? 20;
-        this.visibilityTimeoutSec = options.visibilityTimeoutSec ?? 60;
-        this.taskQueueUrlResolver = options.taskQueueUrl;
-        this.resultsQueueUrlOption = options.resultsQueueUrl;
-    }
-    // ── queue resolution ───────────────────────────────────────────────────────────────────────
-    async resolveByName(name) {
-        const cached = this.urlCache.get(name);
-        if (cached)
-            return cached;
-        let url;
-        try {
-            const out = await this.client.send(new client_sqs_1.GetQueueUrlCommand({ QueueName: name }));
-            url = out.QueueUrl;
-        }
-        catch (err) {
-            if (this.autoCreate) {
-                const created = await this.client.send(new client_sqs_1.CreateQueueCommand({ QueueName: name }));
-                url = created.QueueUrl;
-            }
-            else {
-                throw err;
-            }
-        }
-        this.urlCache.set(name, url);
-        return url;
-    }
-    async tasksUrl(group) {
-        if (this.taskQueueUrlResolver)
-            return this.taskQueueUrlResolver(group);
-        return this.resolveByName(`${this.prefix}-tasks-${group}`);
-    }
-    async resultsUrl() {
-        const opt = this.resultsQueueUrlOption;
-        if (typeof opt === 'string')
-            return opt;
-        if (typeof opt === 'function')
-            return opt();
-        return this.resolveByName(`${this.prefix}-results`);
-    }
-    // ── engine → worker ────────────────────────────────────────────────────────────────────────
-    async dispatch(task) {
-        const QueueUrl = await this.tasksUrl(task.group);
-        await this.client.send(new client_sqs_1.SendMessageCommand({
-            QueueUrl,
-            MessageBody: JSON.stringify(task),
-            MessageAttributes: this.marker
-                ? { [this.marker]: { DataType: 'String', StringValue: '1' } }
-                : undefined,
-        }));
-    }
-    // ── worker → engine ────────────────────────────────────────────────────────────────────────
-    /** Register a step handler (worker side). Starts the group's task consumer on first call. */
-    handle(name, fn) {
-        if (!this.group)
-            throw new Error('SqsTransport needs a `group` to register handlers');
-        this.handlers.set(name, fn);
-        if (!this.taskLoop)
-            this.taskLoop = this.pollTasks(this.group);
-    }
-    onResult(handler) {
-        if (!this.resultLoop)
-            this.resultLoop = this.pollResults(handler);
-    }
-    onHeartbeat(_handler) {
-        // Not modelled over SQS; the queue's visibility-timeout redelivery is the liveness mechanism.
-    }
-    // ── polling loops ──────────────────────────────────────────────────────────────────────────
-    async pollTasks(group) {
-        const QueueUrl = await this.tasksUrl(group);
-        while (this.running) {
-            const messages = await this.receive(QueueUrl, this.visibilityTimeoutSec);
-            for (const msg of messages) {
-                if (this.shouldSkip(msg)) {
-                    // Not ours (shared queue): release immediately so the legacy consumer can take it.
-                    await this.release(QueueUrl, msg.ReceiptHandle);
-                    continue;
-                }
-                const task = JSON.parse(msg.Body ?? '{}');
-                const result = await (0, nestjs_durable_core_1.runStepHandler)(task, this.handlers.get(task.name));
-                await this.client.send(new client_sqs_1.SendMessageCommand({
-                    QueueUrl: await this.resultsUrl(),
-                    MessageBody: JSON.stringify(result),
-                    MessageAttributes: this.marker
-                        ? { [this.marker]: { DataType: 'String', StringValue: '1' } }
-                        : undefined,
-                }));
-                await this.del(QueueUrl, msg.ReceiptHandle);
-            }
-        }
-    }
-    async pollResults(handler) {
-        const QueueUrl = await this.resultsUrl();
-        while (this.running) {
-            const messages = await this.receive(QueueUrl, 30);
-            for (const msg of messages) {
-                if (this.shouldSkip(msg)) {
-                    await this.release(QueueUrl, msg.ReceiptHandle);
-                    continue;
-                }
-                await handler(JSON.parse(msg.Body ?? '{}'));
-                await this.del(QueueUrl, msg.ReceiptHandle);
-            }
-        }
-    }
-    /** One long-poll receive. Swallows the abort thrown on `close()`; logs and backs off otherwise. */
-    async receive(QueueUrl, visibility) {
-        try {
-            const out = await this.client.send(new client_sqs_1.ReceiveMessageCommand({
-                QueueUrl,
-                MaxNumberOfMessages: 10,
-                WaitTimeSeconds: this.waitTimeSeconds,
-                VisibilityTimeout: visibility,
-                MessageAttributeNames: this.marker ? ['All'] : undefined,
-            }), { abortSignal: this.abort.signal });
-            return out.Messages ?? [];
-        }
-        catch (err) {
-            if (!this.running)
-                return [];
-            console.error('[SqsTransport] receive failed', err);
-            await new Promise((r) => setTimeout(r, 1000));
-            return [];
-        }
-    }
-    shouldSkip(msg) {
-        return !!this.marker && !msg.MessageAttributes?.[this.marker];
-    }
-    async del(QueueUrl, ReceiptHandle) {
-        if (!ReceiptHandle)
-            return;
-        await this.client.send(new client_sqs_1.DeleteMessageCommand({ QueueUrl, ReceiptHandle }));
-    }
-    async release(QueueUrl, ReceiptHandle) {
-        if (!ReceiptHandle)
-            return;
-        await this.client
-            .send(new client_sqs_1.ChangeMessageVisibilityCommand({ QueueUrl, ReceiptHandle, VisibilityTimeout: 0 }))
-            .catch(() => { });
-    }
-    /** Stop the pollers and (if we created it) destroy the SQS client so the process can exit. */
-    async close() {
-        this.running = false;
-        this.abort.abort();
-        await Promise.allSettled([this.taskLoop, this.resultLoop]);
-        if (this.ownsClient)
-            this.client.destroy();
-    }
-}
-exports.SqsTransport = SqsTransport;
-//# sourceMappingURL=sqs-transport.js.map

package/dist/sqs-transport.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"sqs-transport.js","sourceRoot":"","sources":["../src/sqs-transport.ts"],"names":[],"mappings":";;;AAAA,oDAS6B;AAC7B,uEAOuC;AAoDvC;;;;;;;;;;GAUG;AACH,MAAa,YAAY;IACN,MAAM,CAAY;IAClB,UAAU,CAAU;IACpB,KAAK,CAAU;IACf,MAAM,CAAS;IACf,UAAU,CAAU;IACpB,MAAM,CAAU;IAChB,eAAe,CAAS;IACxB,oBAAoB,CAAS;IAC7B,oBAAoB,CAAoB;IACxC,qBAAqB,CAA6C;IAElE,QAAQ,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC1C,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC9C,OAAO,GAAG,IAAI,CAAC;IACN,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;IACvC,QAAQ,CAAiB;IACzB,UAAU,CAAiB;IAEnC,YAAY,OAA4B;QACtC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,IAAI,sBAAS,CAAC,OAAO,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;QAC1E,IAAI,CAAC,UAAU,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC;QAClC,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,SAAS,CAAC;QAC1C,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,KAAK,CAAC;QAC9C,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,IAAI,EAAE,CAAC;QACrD,IAAI,CAAC,oBAAoB,GAAG,OAAO,CAAC,oBAAoB,IAAI,EAAE,CAAC;QAC/D,IAAI,CAAC,oBAAoB,GAAG,OAAO,CAAC,YAAY,CAAC;QACjD,IAAI,CAAC,qBAAqB,GAAG,OAAO,CAAC,eAAe,CAAC;IACvD,CAAC;IAED,8FAA8F;IAEtF,KAAK,CAAC,aAAa,CAAC,IAAY;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC;QAC1B,IAAI,GAAW,CAAC;QAChB,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,+BAAkB,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAChF,GAAG,GAAG,GAAG,CAAC,QAAkB,CAAC;QAC/B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACpB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,+BAAkB,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;gBACpF,GAAG,GAAG,OAAO,CAAC,QAAkB,CAAC;YACnC,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QAC7B,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,KAAa;QAClC,IAAI,IAAI,CAAC,oBAAoB;YAAE,OAAO,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACvE,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC,MAAM,UAAU,KAAK,EAAE,CAAC,CAAC;IAC7D,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,MAAM,GAAG,GAAG,IAAI,CAAC,qBAAqB,CAAC;QACvC,IAAI,OAAO,GAAG,KAAK,QAAQ;YAAE,OAAO,GAAG,CAAC;QACxC,IAAI,OAAO,GAAG,KAAK,UAAU;YAAE,OAAO,GAAG,EAAE,CAAC;QAC5C,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC,MAAM,UAAU,CAAC,CAAC;IACtD,CAAC;IAED,8FAA8F;IAE9F,KAAK,CAAC,QAAQ,CAAC,IAAgB;QAC7B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CACpB,IAAI,+BAAkB,CAAC;YACrB,QAAQ;YACR,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YACjC,iBAAiB,EAAE,IAAI,CAAC,MAAM;gBAC5B,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,EAAE,EAAE;gBAC7D,CAAC,CAAC,SAAS;SACd,CAAC,CACH,CAAC;IACJ,CAAC;IAED,8FAA8F;IAE9F,6FAA6F;IAC7F,MAAM,CAAC,IAAY,EAAE,EAAe;QAClC,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACtF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC5B,IAAI,CAAC,IAAI,CAAC,QAAQ;YAAE,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACjE,CAAC;IAED,QAAQ,CAAC,OAA8C;QACrD,IAAI,CAAC,IAAI,CAAC,UAAU;YAAE,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IACpE,CAAC;IAED,WAAW,CAAC,QAA4C;QACtD,8FAA8F;IAChG,CAAC;IAED,8FAA8F;IAEtF,KAAK,CAAC,SAAS,CAAC,KAAa;QACnC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC5C,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACzE,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAC3B,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBACzB,mFAAmF;oBACnF,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;oBAChD,SAAS;gBACX,CAAC;gBACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,IAAI,IAAI,CAAe,CAAC;gBACxD,MAAM,MAAM,GAAG,MAAM,IAAA,oCAAc,EAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;gBACxE,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CACpB,IAAI,+BAAkB,CAAC;oBACrB,QAAQ,EAAE,MAAM,IAAI,CAAC,UAAU,EAAE;oBACjC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;oBACnC,iBAAiB,EAAE,IAAI,CAAC,MAAM;wBAC5B,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,EAAE,EAAE;wBAC7D,CAAC,CAAC,SAAS;iBACd,CAAC,CACH,CAAC;gBACF,MAAM,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,OAA8C;QACtE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACzC,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YAClD,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAC3B,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBACzB,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;oBAChD,SAAS;gBACX,CAAC;gBACD,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,IAAI,IAAI,CAAe,CAAC,CAAC;gBAC1D,MAAM,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;IACH,CAAC;IAED,mGAAmG;IAC3F,KAAK,CAAC,OAAO,CAAC,QAAgB,EAAE,UAAkB;QACxD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAChC,IAAI,kCAAqB,CAAC;gBACxB,QAAQ;gBACR,mBAAmB,EAAE,EAAE;gBACvB,eAAe,EAAE,IAAI,CAAC,eAAe;gBACrC,iBAAiB,EAAE,UAAU;gBAC7B,qBAAqB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS;aACzD,CAAC,EACF,EAAE,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CACnC,CAAC;YACF,OAAO,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC;QAC5B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,IAAI,CAAC,OAAO;gBAAE,OAAO,EAAE,CAAC;YAC7B,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,GAAG,CAAC,CAAC;YACpD,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;YAC9C,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,GAAoD;QACrE,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAChE,CAAC;IAEO,KAAK,CAAC,GAAG,CAAC,QAAgB,EAAE,aAAsB;QACxD,IAAI,CAAC,aAAa;YAAE,OAAO;QAC3B,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,iCAAoB,CAAC,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC;IAChF,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,QAAgB,EAAE,aAAsB;QAC5D,IAAI,CAAC,aAAa;YAAE,OAAO;QAC3B,MAAM,IAAI,CAAC,MAAM;aACd,IAAI,CAAC,IAAI,2CAA8B,CAAC,EAAE,QAAQ,EAAE,aAAa,EAAE,iBAAiB,EAAE,CAAC,EAAE,CAAC,CAAC;aAC3F,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACrB,CAAC;IAED,8FAA8F;IAC9F,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACrB,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;QAC3D,IAAI,IAAI,CAAC,UAAU;YAAE,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;IAC7C,CAAC;CACF;AAzLD,oCAyLC"}