@iworkflow/dbos 0.6.1 → 0.8.0

package/dist/index.js

@@ -2,7 +2,9 @@
 import { DBOS, DBOSClient, Error as DBOSErrors } from "@dbos-inc/dbos-sdk";
 import {
   stepScope,
+  checkLimits,
   checkPriority,
+  plainAttrs,
   DEFAULT_PRIORITY,
   exitSignal,
   Failure,
@@ -69,7 +71,7 @@ function stepKey() {
   const wf = DBOS.workflowID;
   const step = DBOS.stepID;
   if (!wf || step === undefined)
-    throw new Error("heartbeat вне шага DBOS");
+    throw new Error("вызов вне шага DBOS");
   return [wf, step];
 }
 var typeOf = (e) => e instanceof Failure ? e.type : e instanceof Error ? e.name : "Error";
@@ -127,12 +129,16 @@ class Semaphore {
 class AttemptTimeout extends Error {
 }
 var sleepMs = (ms) => new Promise((r) => setTimeout(r, ms));
-async function withTimeout(p, ms) {
+async function withTimeout(p, ms, abort) {
   let timer;
   try {
     return await Promise.race([
       p,
-      new Promise((_, reject) => timer = setTimeout(() => reject(new AttemptTimeout(`попытка дольше ${ms} мс`)), ms))
+      new Promise((_, reject) => timer = setTimeout(() => {
+        const e = new AttemptTimeout(`попытка дольше ${ms} мс`);
+        abort.abort(e);
+        reject(e);
+      }, ms))
     ]);
   } finally {
     clearTimeout(timer);
@@ -144,7 +150,10 @@ async function attempts(fn, args, opts, pool, alive, priority, runtime) {
   let interval = opts.retryIntervalMs ?? 200;
   for (let n = 1;; n++) {
     try {
-      const attempt = () => stepScope(alsScope).run(runtime(n), () => withTimeout(fn(...args), opts.timeoutMs ?? 60000));
+      const attempt = () => {
+        const abort = new AbortController;
+        return stepScope(alsScope).run(runtime(n, abort.signal), () => withTimeout(fn(...args), opts.timeoutMs ?? 60000, abort));
+      };
       return await (pool ? pool.run(attempt, priority) : attempt());
     } catch (e) {
       const stop = e instanceof Failure && e.nonRetryable || (opts.nonRetryableTypes ?? []).includes(typeOf(e));
@@ -188,10 +197,11 @@ function dbosRuntime() {
   let pool;
   const exit = exitSignal();
   async function launch(def) {
+    checkLimits(def);
     pool = new Pool({ connectionString: sysDb(), max: 2 });
     const db = pool;
     const patches = patchBook(db, def.app, def.patches ?? []);
-    const stepRuntime = (n) => ({
+    const stepRuntime = (n, signal) => ({
       async heartbeat(details) {
         await db.query(`INSERT INTO iwf.heartbeat (workflow_id, step_id, details) VALUES ($1, $2, $3)
            ON CONFLICT (workflow_id, step_id) DO UPDATE SET details = EXCLUDED.details, at = now()`, [...stepKey(), JSON.stringify(details ?? null)]);
@@ -200,9 +210,15 @@ function dbosRuntime() {
         const r = await db.query("SELECT details FROM iwf.heartbeat WHERE workflow_id = $1 AND step_id = $2", stepKey());
         return r.rows[0]?.details ?? undefined;
       },
-      attempt: () => n
+      attempt: () => n,
+      info: () => {
+        const [wf, step] = stepKey();
+        return { workflowId: wf, runId: wf, stepId: String(step) };
+      },
+      signal: () => signal
     });
     const pools = Object.fromEntries(Object.entries(def.pools ?? {}).map(([p, n]) => [p, new Semaphore(n)]));
+    const main = def.maxConcurrentSteps ? new Semaphore(def.maxConcurrentSteps) : undefined;
     const enqueue = (target, args, opts, reject, priority) => DBOS.enqueueWorkflowWithOptions({
       queueName: target.app,
       workflowName: target.flow,
@@ -233,7 +249,7 @@ function dbosRuntime() {
           priority: row?.priority || DEFAULT_PRIORITY,
           now: () => DBOS.now(),
           step: (step, stepArgs, opts = {}) => {
-            const pool = opts.pool ? pools[opts.pool] : undefined;
+            const pool = opts.pool ? pools[opts.pool] : main;
             if (opts.pool && !pool)
               throw new Error(`пул "${opts.pool}" не объявлен в AppDefinition.pools модуля ${def.app}`);
             const alive = async () => {
@@ -268,6 +284,10 @@ function dbosRuntime() {
               throw new FlowError(`процесса ${target} нет или он завершён`, { reason: "not-found" });
             await DBOS.send(target, value, topic);
           },
+          log: async (level, message, attrs) => {
+            const plain = plainAttrs(attrs);
+            console[level](message, ...plain ? [plain] : []);
+          },
           setState: (key, value) => DBOS.setEvent(key, value),
           patched: (id) => patches.patched(workflowId, id),
           deprecatePatch: async () => {}
@@ -286,15 +306,20 @@ function dbosRuntime() {
     const poll = Number(process.env.DBOS_QUEUE_POLL_MS) || undefined;
     await DBOS.registerQueue(def.app, { minPollingIntervalMs: poll, onConflict: "always_update" });
   }
+  const closePool = async () => {
+    await pool?.end().catch(() => {});
+    pool = undefined;
+  };
   return {
     done: () => exit.promise(),
     async start(def) {
+      if (pool)
+        throw new Error("рантайм уже запущен");
       exit.arm();
       try {
         await launch(def);
       } catch (e) {
-        await pool?.end().catch(() => {});
-        pool = undefined;
+        await closePool();
         exit.reject(e);
         throw e;
       }
@@ -302,8 +327,8 @@ function dbosRuntime() {
     async stop() {
       try {
         await DBOS.shutdown();
-        await pool?.end();
       } finally {
+        await closePool();
         exit.resolve();
       }
     }
@@ -342,8 +367,9 @@ async function dbosClient() {
         workflowID: opts.id,
         applicationName: target.app,
         ...opts.timeoutMs ? { workflowTimeoutMS: opts.timeoutMs } : {},
-        priority: checkPriority(opts.priority) ?? DEFAULT_PRIORITY
-      }, ...args);
+        priority: checkPriority(opts.priority) ?? DEFAULT_PRIORITY,
+        ...opts.onExisting === "reject" || opts.reuseClosedId ? { workflowIDReusePolicy: "reject" } : {}
+      }, ...args).catch(alreadyStarted(opts.id));
       return wrap(h.workflowID);
     },
     async signalWithStart(target, args, opts, topic, value) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@iworkflow/dbos",
-  "version": "0.6.1",
+  "version": "0.8.0",
   "description": "DBOS adapter for @iworkflow/core",
   "keywords": [
     "workflow",
@@ -37,7 +37,7 @@
     "prepack": "bun run build"
   },
   "dependencies": {
-    "@iworkflow/core": "^0.6.1",
+    "@iworkflow/core": "^0.8.0",
     "pg": "^8.0.0"
   },
   "peerDependencies": {

package/src/index.ts

@@ -4,7 +4,9 @@ import { DBOS, DBOSClient, Error as DBOSErrors, type WorkflowStatus } from "@dbo
 import {
   stepScope,
   type StepRuntime,
+  checkLimits,
   checkPriority,
+  plainAttrs,
   DEFAULT_PRIORITY,
   exitSignal,
   Failure,
@@ -97,7 +99,7 @@ function patchBook(db: Pool, app: string, declared: string[]) {
 function stepKey(): [string, number] {
   const wf = DBOS.workflowID;
   const step = DBOS.stepID;
-  if (!wf || step === undefined) throw new Error("heartbeat вне шага DBOS");
+  if (!wf || step === undefined) throw new Error("вызов вне шага DBOS");
   return [wf, step];
 }
 
@@ -169,12 +171,17 @@ const sleepMs = (ms: number) => new Promise((r) => setTimeout(r, ms));
 
 
 /** Попытка, урезанная по сроку: брошенная продолжает идти в фоне, её итог отбрасывается. */
-async function withTimeout<T>(p: Promise<T>, ms: number): Promise<T> {
+async function withTimeout<T>(p: Promise<T>, ms: number, abort: AbortController): Promise<T> {
   let timer: ReturnType<typeof setTimeout> | undefined;
   try {
     return await Promise.race([
       p,
-      new Promise<never>((_, reject) => (timer = setTimeout(() => reject(new AttemptTimeout(`попытка дольше ${ms} мс`)), ms))),
+      new Promise<never>((_, reject) => (timer = setTimeout(() => {
+        const e = new AttemptTimeout(`попытка дольше ${ms} мс`);
+        // Брошенная попытка видит отмену через stepSignal() и может остановиться сама.
+        abort.abort(e);
+        reject(e);
+      }, ms))),
     ]);
   } finally {
     clearTimeout(timer);
@@ -189,14 +196,17 @@ async function attempts(
   pool: Semaphore | undefined,
   alive: () => Promise<boolean>,
   priority: number | undefined,
-  runtime: (n: number) => StepRuntime,
+  runtime: (n: number, signal: AbortSignal) => StepRuntime,
 ) {
   const max = opts.attempts ?? 3;
   const started = Date.now();
   let interval = opts.retryIntervalMs ?? 200;
   for (let n = 1; ; n++) {
     try {
-      const attempt = () => stepScope(alsScope).run(runtime(n), () => withTimeout(fn(...args), opts.timeoutMs ?? 60_000));
+      const attempt = () => {
+        const abort = new AbortController();
+        return stepScope(alsScope).run(runtime(n, abort.signal), () => withTimeout(fn(...args), opts.timeoutMs ?? 60_000, abort));
+      };
       return await (pool ? pool.run(attempt, priority) : attempt());
     } catch (e) {
       const stop = (e instanceof Failure && e.nonRetryable) || (opts.nonRetryableTypes ?? []).includes(typeOf(e));
@@ -249,11 +259,12 @@ export function dbosRuntime(): EngineRuntime {
   // Воркеров у DBOS нет: done() — до stop или сбоя start; падения исполнителя DBOS не видно.
   const exit = exitSignal();
   async function launch(def: AppDefinition) {
+    checkLimits(def);
     pool = new Pool({ connectionString: sysDb(), max: 2 });
     const db = pool;
     const patches = patchBook(db, def.app, def.patches ?? []);
     // Рантайм попытки шага: прогресс — в iwf.heartbeat по (процесс, шаг), номер — из цикла повторов.
-    const stepRuntime = (n: number): StepRuntime => ({
+    const stepRuntime = (n: number, signal: AbortSignal): StepRuntime => ({
       async heartbeat(details) {
         await db.query(
           `INSERT INTO iwf.heartbeat (workflow_id, step_id, details) VALUES ($1, $2, $3)
@@ -266,8 +277,16 @@ export function dbosRuntime(): EngineRuntime {
         return r.rows[0]?.details ?? undefined;
       },
       attempt: () => n,
+      // id DBOS не переиспользуется: запуск у процесса один.
+      info: () => {
+        const [wf, step] = stepKey();
+        return { workflowId: wf, runId: wf, stepId: String(step) };
+      },
+      signal: () => signal,
     });
     const pools = Object.fromEntries(Object.entries(def.pools ?? {}).map(([p, n]) => [p, new Semaphore(n)]));
+    // Шаги вне пулов — под общим пределом модуля, как maxConcurrentActivityTaskExecutions у Temporal.
+    const main = def.maxConcurrentSteps ? new Semaphore(def.maxConcurrentSteps) : undefined;
     const enqueue = (
       target: { app: string; flow: string },
       args: unknown[],
@@ -312,7 +331,7 @@ export function dbosRuntime(): EngineRuntime {
           priority: row?.priority || DEFAULT_PRIORITY,
           now: () => DBOS.now(),
           step: (step, stepArgs, opts = {}) => {
-            const pool = opts.pool ? pools[opts.pool] : undefined;
+            const pool = opts.pool ? pools[opts.pool] : main;
             if (opts.pool && !pool) throw new Error(`пул "${opts.pool}" не объявлен в AppDefinition.pools модуля ${def.app}`);
             // heartbeatTimeoutMs у DBOS не действует: умерший процесс ловится рестартом, шаг повторяется.
             // Срок вышел, а DBOS ещё не отменил процесс — ждём отмены: иначе процесс поймает отказ и завершится успехом.
@@ -359,6 +378,12 @@ export function dbosRuntime(): EngineRuntime {
               throw new FlowError(`процесса ${target} нет или он завершён`, { reason: "not-found" });
             await DBOS.send(target, value, topic);
           },
+          // Не шагом: строка журнала не сдвигает нумерацию шагов. Цена — повтор строки при восстановлении после рестарта.
+          log: async (level, message, attrs) => {
+            const plain = plainAttrs(attrs);
+            // console, а не DBOS.logger: у того logLevel "warn" — info молча пропал бы.
+            console[level](message, ...(plain ? [plain] : []));
+          },
           setState: (key, value) => DBOS.setEvent(key, value),
           patched: (id) => patches.patched(workflowId, id),
           // Старых процессов нет — ветка всегда новая; вызов можно удалять.
@@ -382,16 +407,21 @@ export function dbosRuntime(): EngineRuntime {
     const poll = Number(process.env.DBOS_QUEUE_POLL_MS) || undefined;
     await DBOS.registerQueue(def.app, { minPollingIntervalMs: poll, onConflict: "always_update" });
   }
+  const closePool = async () => {
+    await pool?.end().catch(() => {});
+    pool = undefined;
+  };
   return {
     done: () => exit.promise(),
     async start(def: AppDefinition) {
+      // Второй start не трогает пул и done() идущего: его сбой закрыл бы живое соединение.
+      if (pool) throw new Error("рантайм уже запущен");
       exit.arm();
       // Не запустился — done() отклоняется: для наблюдателя это сбой, а не штатный выход.
       try {
         await launch(def);
       } catch (e) {
-        await pool?.end().catch(() => {});
-        pool = undefined;
+        await closePool();
         exit.reject(e);
         throw e;
       }
@@ -399,8 +429,9 @@ export function dbosRuntime(): EngineRuntime {
     async stop() {
       try {
         await DBOS.shutdown();
-        await pool?.end();
       } finally {
+        // Пул закрывается и при сбое shutdown: иначе соединения копятся от цикла к циклу.
+        await closePool();
         exit.resolve();
       }
     },
@@ -444,12 +475,13 @@ export async function dbosClient(): Promise<EngineClient> {
           ...(opts.timeoutMs ? { workflowTimeoutMS: opts.timeoutMs } : {}),
           // Без приоритета DBOS ставит 0 — впереди всех; порт по умолчанию 3, как Temporal.
           priority: checkPriority(opts.priority) ?? DEFAULT_PRIORITY,
+          ...(opts.onExisting === "reject" || opts.reuseClosedId ? { workflowIDReusePolicy: "reject" as const } : {}),
         },
         ...args,
-      );
+      ).catch(alreadyStarted(opts.id));
       return wrap(h.workflowID);
     },
-    async signalWithStart<T>(target: FlowRef, args: unknown[], opts: StartOptions, topic: string, value: unknown) {
+    async signalWithStart<T>(target: FlowRef, args: unknown[], opts: Omit<StartOptions, "onExisting" | "reuseClosedId">, topic: string, value: unknown) {
       // Не атомарно, но без потерь: сигнал ждёт в очереди темы, пока процесс до него дойдёт.
       const h = await self.start<T>(target, args, opts);
       // id завершённого DBOS не переиспользует: сигнал ушёл бы в никуда.