@dudousxd/nestjs-durable-core 0.85.0 → 0.85.2

package/dist/index.cjs

@@ -2630,7 +2630,31 @@ __name(resolveCronParse, "resolveCronParse");
 
 // src/scheduler.ts
 var nodeRequire = (0, import_node_module.createRequire)(__importMetaUrl);
-var defaultSleep = /* @__PURE__ */ __name((ms) => new Promise((r) => setTimeout(r, ms)), "defaultSleep");
+var jitterFireAt = /* @__PURE__ */ new WeakMap();
+function deferredWindows(engine, key) {
+  return [
+    ...jitterFireAt.get(engine)?.get(key)?.keys() ?? []
+  ];
+}
+__name(deferredWindows, "deferredWindows");
+function jitterDue(engine, s, candidates, nowMs, random) {
+  let byKey = jitterFireAt.get(engine);
+  if (byKey === void 0) {
+    byKey = /* @__PURE__ */ new Map();
+    jitterFireAt.set(engine, byKey);
+  }
+  const previous = byKey.get(s.key);
+  const current = /* @__PURE__ */ new Map();
+  const due = [];
+  for (const runId of candidates) {
+    const at = previous?.get(runId) ?? nowMs + Math.floor(random() * s.jitter);
+    if (nowMs >= at) due.push(runId);
+    else current.set(runId, at);
+  }
+  byKey.set(s.key, current);
+  return due;
+}
+__name(jitterDue, "jitterDue");
 function scheduledRunId(key, everyMs, nowMs) {
   return `sched:${key}:${Math.floor(nowMs / everyMs)}`;
 }
@@ -2700,13 +2724,19 @@ function scheduleRunIdsAt(s, nowMs) {
 __name(scheduleRunIdsAt, "scheduleRunIdsAt");
 async function runSchedules(engine, schedules, nowMs, opts) {
   const random = opts?.random ?? Math.random;
-  const sleep = opts?.sleep ?? defaultSleep;
   const settled = settledIdsFor(engine);
   const ids = [];
   for (const s of schedules) {
     if (engine.isDraining?.()) break;
     if (s.paused) continue;
-    const candidates = scheduleRunIdsAt(s, nowMs).filter((id) => !settled.get(s.key)?.has(id));
+    const due = scheduleRunIdsAt(s, nowMs);
+    const windows = s.jitter && s.jitter > 0 ? [
+      .../* @__PURE__ */ new Set([
+        ...deferredWindows(engine, s.key),
+        ...due
+      ])
+    ] : due;
+    const candidates = windows.filter((id) => !settled.get(s.key)?.has(id));
     if (candidates.length === 0) continue;
     if (s.overlap === "skip" && s.everyMs) {
       const prevBucket = Math.floor(nowMs / s.everyMs) - 1;
@@ -2715,8 +2745,10 @@ async function runSchedules(engine, schedules, nowMs, opts) {
         if (prev && (prev.status === "running" || prev.status === "suspended")) continue;
       }
     }
-    if (s.jitter && s.jitter > 0) await sleep(Math.floor(random() * s.jitter));
-    for (const runId of candidates) {
+    if (!(s.jitter && s.jitter > 0)) jitterFireAt.get(engine)?.delete(s.key);
+    const firing = s.jitter && s.jitter > 0 ? jitterDue(engine, s, candidates, nowMs, random) : candidates;
+    if (firing.length === 0) continue;
+    for (const runId of firing) {
       if (engine.isDraining?.()) break;
       const existing = await engine.getRun(runId);
       remember(settled, s.key, runId);