@iworkflow/dbos 0.6.0 → 0.7.0

package/dist/index.js

@@ -2,6 +2,7 @@
 import { DBOS, DBOSClient, Error as DBOSErrors } from "@dbos-inc/dbos-sdk";
 import {
   stepScope,
+  checkLimits,
   checkPriority,
   DEFAULT_PRIORITY,
   exitSignal,
@@ -127,7 +128,6 @@ class Semaphore {
 class AttemptTimeout extends Error {
 }
 var sleepMs = (ms) => new Promise((r) => setTimeout(r, ms));
-var scope = stepScope(alsScope);
 async function withTimeout(p, ms) {
   let timer;
   try {
@@ -145,7 +145,7 @@ async function attempts(fn, args, opts, pool, alive, priority, runtime) {
   let interval = opts.retryIntervalMs ?? 200;
   for (let n = 1;; n++) {
     try {
-      const attempt = () => scope.run(runtime(n), () => withTimeout(fn(...args), opts.timeoutMs ?? 60000));
+      const attempt = () => stepScope(alsScope).run(runtime(n), () => withTimeout(fn(...args), opts.timeoutMs ?? 60000));
       return await (pool ? pool.run(attempt, priority) : attempt());
     } catch (e) {
       const stop = e instanceof Failure && e.nonRetryable || (opts.nonRetryableTypes ?? []).includes(typeOf(e));
@@ -189,6 +189,7 @@ 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 ?? []);
@@ -204,6 +205,7 @@ function dbosRuntime() {
       attempt: () => n
     });
     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,
@@ -234,7 +236,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 () => {
@@ -287,15 +289,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;
       }
@@ -303,8 +310,8 @@ function dbosRuntime() {
     async stop() {
       try {
         await DBOS.shutdown();
-        await pool?.end();
       } finally {
+        await closePool();
         exit.resolve();
       }
     }
@@ -343,8 +350,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" ? { 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.0",
+  "version": "0.7.0",
   "description": "DBOS adapter for @iworkflow/core",
   "keywords": [
     "workflow",
@@ -37,7 +37,7 @@
     "prepack": "bun run build"
   },
   "dependencies": {
-    "@iworkflow/core": "^0.6.0",
+    "@iworkflow/core": "^0.7.0",
     "pg": "^8.0.0"
   },
   "peerDependencies": {

package/src/index.ts

@@ -4,6 +4,7 @@ import { DBOS, DBOSClient, Error as DBOSErrors, type WorkflowStatus } from "@dbo
 import {
   stepScope,
   type StepRuntime,
+  checkLimits,
   checkPriority,
   DEFAULT_PRIORITY,
   exitSignal,
@@ -166,7 +167,7 @@ class AttemptTimeout extends Error {}
 
 const sleepMs = (ms: number) => new Promise((r) => setTimeout(r, ms));
 
-const scope = stepScope(alsScope);
+
 
 /** Попытка, урезанная по сроку: брошенная продолжает идти в фоне, её итог отбрасывается. */
 async function withTimeout<T>(p: Promise<T>, ms: number): Promise<T> {
@@ -196,7 +197,7 @@ async function attempts(
   let interval = opts.retryIntervalMs ?? 200;
   for (let n = 1; ; n++) {
     try {
-      const attempt = () => scope.run(runtime(n), () => withTimeout(fn(...args), opts.timeoutMs ?? 60_000));
+      const attempt = () => stepScope(alsScope).run(runtime(n), () => withTimeout(fn(...args), opts.timeoutMs ?? 60_000));
       return await (pool ? pool.run(attempt, priority) : attempt());
     } catch (e) {
       const stop = (e instanceof Failure && e.nonRetryable) || (opts.nonRetryableTypes ?? []).includes(typeOf(e));
@@ -249,6 +250,7 @@ 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 ?? []);
@@ -268,6 +270,8 @@ export function dbosRuntime(): EngineRuntime {
       attempt: () => n,
     });
     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 +316,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 ещё не отменил процесс — ждём отмены: иначе процесс поймает отказ и завершится успехом.
@@ -382,16 +386,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 +408,9 @@ export function dbosRuntime(): EngineRuntime {
     async stop() {
       try {
         await DBOS.shutdown();
-        await pool?.end();
       } finally {
+        // Пул закрывается и при сбое shutdown: иначе соединения копятся от цикла к циклу.
+        await closePool();
         exit.resolve();
       }
     },
@@ -444,12 +454,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" ? { 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">, topic: string, value: unknown) {
       // Не атомарно, но без потерь: сигнал ждёт в очереди темы, пока процесс до него дойдёт.
       const h = await self.start<T>(target, args, opts);
       // id завершённого DBOS не переиспользует: сигнал ушёл бы в никуда.