@dudousxd/nestjs-durable 0.46.0 → 0.46.2

package/dist/index.js

@@ -1173,6 +1173,31 @@ ProxyRunGateway = _ts_decorate6([
 // src/retention-poller.ts
 import { DURABLE_OPTIONS_CANONICAL as DURABLE_OPTIONS_CANONICAL4, STATE_STORE_CANONICAL, TERMINAL_RUN_STATUSES, parseDuration } from "@dudousxd/nestjs-durable-core";
 import { Inject as Inject4, Injectable as Injectable7 } from "@nestjs/common";
+
+// src/settle-within.ts
+async function settleWithin(p, timeoutMs) {
+  let timer;
+  let timedOut = false;
+  const timeout = new Promise((resolve) => {
+    timer = setTimeout(() => {
+      timedOut = true;
+      resolve();
+    }, timeoutMs);
+    timer.unref?.();
+  });
+  try {
+    await Promise.race([
+      p.then(void 0, () => void 0),
+      timeout
+    ]);
+  } finally {
+    if (timer) clearTimeout(timer);
+  }
+  return timedOut;
+}
+__name(settleWithin, "settleWithin");
+
+// src/retention-poller.ts
 function _ts_decorate7(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function") {
@@ -1243,6 +1268,9 @@ var RetentionPoller = class {
   options;
   timer;
   sweeping = false;
+  stopping = false;
+  /** The sweep in flight, so shutdown can wait for it before the store closes. */
+  current;
   constructor(store, options) {
     this.store = store;
     this.options = options;
@@ -1270,8 +1298,18 @@ var RetentionPoller = class {
       this.timer.unref?.();
     }
   }
-  onModuleDestroy() {
+  /** Stop sweeping and wait for the sweep in flight (bounded by `shutdownTimeoutMs`, default 10s),
+  *  so no prune is still pending when the host closes its connection pool. This wait runs concurrently
+  *  with the timer poller's and is NOT deducted from the engine's drain budget (the retention poller
+  *  holds no engine), so a slow prune can add up to one more `shutdownTimeoutMs` before the drain. */
+  async onModuleDestroy() {
     if (this.timer) clearInterval(this.timer);
+    this.stopping = true;
+    if (!this.current) return;
+    const budget = this.options.shutdownTimeoutMs ?? 1e4;
+    if (await settleWithin(this.current, budget)) {
+      console.warn(`[nestjs-durable] retention sweep still running after ${budget}ms at shutdown; continuing`);
+    }
   }
   /**
   * Prune one round.
@@ -1282,8 +1320,16 @@ var RetentionPoller = class {
   * Pruning history is the least urgent thing this package does — a `pruneTerminalRuns` that fails
   * on a locked table must cost this sweep and nothing more.
   */
-  async sweep() {
-    if (this.sweeping) return;
+  sweep() {
+    if (this.sweeping) return this.current ?? Promise.resolve();
+    const run = this.sweepOnce().finally(() => {
+      if (this.current === run) this.current = void 0;
+    });
+    this.current = run;
+    return run;
+  }
+  async sweepOnce() {
+    if (this.stopping) return;
     const retention = this.options.retention;
     const store = this.requireStore();
     const prune = store.pruneTerminalRuns;
@@ -1294,6 +1340,7 @@ var RetentionPoller = class {
       const now = Date.now();
       for (const policy of retention.policies) {
         for (let batch = 0; batch < MAX_BATCHES_PER_POLICY; batch++) {
+          if (this.stopping) return;
           const deleted = await prune.call(store, policy, now, batchSize);
           if (deleted < batchSize) break;
         }
@@ -1705,6 +1752,8 @@ var TimerPoller = class {
   options;
   timer;
   polling = false;
+  /** The sweep in flight, so shutdown can wait for it instead of racing the store's close. */
+  current;
   unsubscribeEnqueued;
   constructor(engine, options) {
     this.engine = engine;
@@ -1724,9 +1773,27 @@ var TimerPoller = class {
       this.timer.unref?.();
     }
   }
-  onModuleDestroy() {
+  /**
+  * Stop polling and wait for the sweep in flight (bounded by `shutdownTimeoutMs`, default 10s).
+  * Nest runs every `onModuleDestroy` before any `onApplicationShutdown`, so returning here only once
+  * the sweep has settled means no store call is still pending when the registrar drains and the
+  * host closes its connection pool. The engine is told to stop first so a sweep working through a
+  * backlog stops locking runs instead of finishing the backlog. The time spent waiting is deducted
+  * from the later `engine.drain()`, so poll wait plus drain stay within one `shutdownTimeoutMs`.
+  */
+  async onModuleDestroy() {
     if (this.timer) clearInterval(this.timer);
     this.unsubscribeEnqueued?.();
+    if (!isDrivingOperator(this.options)) return;
+    this.engine.beginDrain?.();
+    if (!this.current) return;
+    const budget = this.options.shutdownTimeoutMs ?? 1e4;
+    const started = Date.now();
+    const timedOut = await settleWithin(this.current, budget);
+    this.engine.beginDrain?.(Date.now() - started);
+    if (timedOut) {
+      console.warn(`[nestjs-durable] timer poll still running after ${budget}ms at shutdown; continuing (it may report store errors if the host ends its pool now)`);
+    }
   }
   /**
   * One sweep of the engine's recovery work.
@@ -1743,10 +1810,18 @@ var TimerPoller = class {
   * `recoverIncomplete` says nothing about whether durable timers should still fire, and skipping
   * the rest of the tick would silently widen one subsystem's outage into all of them.
   */
-  async poll() {
-    if (this.polling) return;
+  poll() {
+    if (this.polling) return this.current ?? Promise.resolve();
     this.polling = true;
-    try {
+    const run = this.pollOnce().finally(() => {
+      this.polling = false;
+      if (this.current === run) this.current = void 0;
+    });
+    this.current = run;
+    return run;
+  }
+  async pollOnce() {
+    {
       await this.sweep("runPending", () => this.engine.runPending());
       await this.sweep("recoverIncomplete", () => this.engine.recoverIncomplete());
       await this.sweep("resumeDueTimers", () => this.engine.resumeDueTimers());
@@ -1758,8 +1833,6 @@ var TimerPoller = class {
       if (this.options.persistedSchedules) {
         await this.sweep("persistedSchedules", () => this.engine.schedules.tick());
       }
-    } finally {
-      this.polling = false;
     }
   }
   /** Run one sweep, absorbing and reporting its failure — see {@link poll}. */