@dbos-inc/dbos-enterprise 5.3.1-preview → 5.4.1-preview

package/README.md

@@ -1,7 +1,9 @@
 # DBOS Enterprise for TypeScript
 
-Closed-source extension of [DBOS Transact](https://github.com/dbos-inc/dbos-transact-ts).
-Requires a DBOS Conductor API key.
+Closed-source extension of [DBOS Transact](https://github.com/dbos-inc/dbos-transact-ts)
+that connects applications to DBOS Conductor. Requires a DBOS Conductor API key.
+DBOS loads it automatically when you pass `conductorKey` to `DBOS.launch`, or
+when running in DBOS Cloud.
 Licensed under the [DBOS Conductor Software License](https://www.dbos.dev/conductor-license).
 
 ```

package/dist/src/auto_upgrade.d.ts

@@ -0,0 +1,52 @@
+import type { EnrolledWorkflow } from '@dbos-inc/dbos-sdk/internal/auto_upgrade';
+import type { DBOSExecutor } from '@dbos-inc/dbos-sdk/internal/dbos-executor';
+import type { AutoUpgradeOptions, AutoUpgradePass as AutoUpgradePassContract } from '@dbos-inc/dbos-sdk/internal/enterprise';
+import { type SystemDatabase, type VersionInfo } from '@dbos-inc/dbos-sdk/internal/system_database';
+export declare const DEFAULT_CHUNK_SIZE = 1000;
+/**
+ * The pg_try_advisory_lock argument for one application's auto-upgrade: the leading 8 bytes of
+ * SHA-256 over a fixed string, read as a signed big-endian 64-bit integer. Every DBOS SDK derives
+ * the key this way, so passes in different languages against one system database contend for the
+ * same lock; changing it here changes it everywhere.
+ */
+export declare function autoUpgradeLockKey(schemaName: string, appName: string): bigint;
+/**
+ * A session-scoped advisory lock on one dedicated connection, held across passes. The connection
+ * is registered with the system database so its destroy() cuts it, lock included. CockroachDB
+ * has no advisory locks and always takes it, so it upgrades unprotected rather than not at all.
+ */
+export declare class AutoUpgradeLease {
+    #private;
+    readonly sysdb: SystemDatabase;
+    readonly key: string;
+    held: boolean;
+    constructor(sysdb: SystemDatabase, appName: string);
+    acquire(): Promise<boolean>;
+    release(): Promise<void>;
+}
+/** The hook tests use to stop a pass while a chunk is in flight. */
+export declare const chunkHooks: {
+    afterChunk: (() => void) | undefined;
+};
+export interface UpgradePassOptions {
+    chunkSize?: number;
+    graceMs?: number;
+}
+/**
+ * Upgrade the enrolled workflows' work from previous application versions to `latestVersion`.
+ *
+ * Three phases, each collecting the rows to move and then moving them in chunks: the
+ * ENQUEUED rows, the DELAYED rows, then the PENDING rows. The PENDING phase is skipped until
+ * the latest version has been the latest for `graceMs`. Returns how many rows moved.
+ */
+export declare function upgradePass(sysdb: SystemDatabase, workflows: readonly EnrolledWorkflow[], latestVersion: VersionInfo, signal: AbortSignal, options?: UpgradePassOptions): Promise<number>;
+/** Satisfies core's AutoUpgradePass contract: core's loop calls run() on a timer and close() when it stops. */
+export declare class AutoUpgradePass implements AutoUpgradePassContract {
+    readonly executor: DBOSExecutor;
+    readonly lease: AutoUpgradeLease;
+    readonly graceMs: number;
+    constructor(executor: DBOSExecutor, options: AutoUpgradeOptions);
+    close(): Promise<void>;
+    run(signal: AbortSignal): Promise<number>;
+}
+//# sourceMappingURL=auto_upgrade.d.ts.map

package/dist/src/auto_upgrade.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto_upgrade.d.ts","sourceRoot":"","sources":["../../src/auto_upgrade.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAC;AACjF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2CAA2C,CAAC;AAC9E,OAAO,KAAK,EACV,kBAAkB,EAClB,eAAe,IAAI,uBAAuB,EAC3C,MAAM,wCAAwC,CAAC;AAGhD,OAAO,EAGL,KAAK,cAAc,EACnB,KAAK,WAAW,EACjB,MAAM,6CAA6C,CAAC;AAKrD,eAAO,MAAM,kBAAkB,OAAO,CAAC;AAEvC;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,CAE9E;AAED;;;;GAIG;AACH,qBAAa,gBAAgB;;IAMzB,QAAQ,CAAC,KAAK,EAAE,cAAc;IALhC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IAErB,IAAI,UAAS;gBAGF,KAAK,EAAE,cAAc,EAC9B,OAAO,EAAE,MAAM;IAKX,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC;IA8B3B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CA0B/B;AAwED,oEAAoE;AACpE,eAAO,MAAM,UAAU;uBACW,IAAI;CACrC,CAAC;AAyDF,MAAM,WAAW,kBAAkB;IACjC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;;;;;GAMG;AACH,wBAAsB,WAAW,CAC/B,KAAK,EAAE,cAAc,EACrB,SAAS,EAAE,SAAS,gBAAgB,EAAE,EACtC,aAAa,EAAE,WAAW,EAC1B,MAAM,EAAE,WAAW,EACnB,OAAO,GAAE,kBAAuB,GAC/B,OAAO,CAAC,MAAM,CAAC,CA4BjB;AAED,+GAA+G;AAC/G,qBAAa,eAAgB,YAAW,uBAAuB;IAK3D,QAAQ,CAAC,QAAQ,EAAE,YAAY;IAJjC,QAAQ,CAAC,KAAK,EAAE,gBAAgB,CAAC;IACjC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;gBAGd,QAAQ,EAAE,YAAY,EAC/B,OAAO,EAAE,kBAAkB;IAUvB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAItB,GAAG,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;CAkChD"}

package/dist/src/auto_upgrade.js

@@ -0,0 +1,273 @@
+"use strict";
+// Auto-upgrade: hand work left on an older application version to the latest one.
+//
+// This file declares the logic for one auto-upgrade pass. Core owns the loop that calls it.
+//
+// Each pass:
+//   - skips unless this executor runs the latest application version;
+//   - takes a session advisory lock, so only one pass runs at a time per application;
+//   - moves the rows.
+//
+// PENDING workflows get a grace period of `graceMs` before the pass moves them.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AutoUpgradePass = exports.upgradePass = exports.chunkHooks = exports.AutoUpgradeLease = exports.autoUpgradeLockKey = exports.DEFAULT_CHUNK_SIZE = void 0;
+const cockroach_1 = require("./cockroach");
+const error_1 = require("@dbos-inc/dbos-sdk/internal/error");
+const context_1 = require("@dbos-inc/dbos-sdk/internal/context");
+const system_database_1 = require("@dbos-inc/dbos-sdk/internal/system_database");
+const utils_1 = require("@dbos-inc/dbos-sdk/internal/utils");
+const workflow_1 = require("@dbos-inc/dbos-sdk/internal/workflow");
+exports.DEFAULT_CHUNK_SIZE = 1000;
+/**
+ * The pg_try_advisory_lock argument for one application's auto-upgrade: the leading 8 bytes of
+ * SHA-256 over a fixed string, read as a signed big-endian 64-bit integer. Every DBOS SDK derives
+ * the key this way, so passes in different languages against one system database contend for the
+ * same lock; changing it here changes it everywhere.
+ */
+function autoUpgradeLockKey(schemaName, appName) {
+    return (0, system_database_1.advisoryLockKey)(`dbos.auto_upgrade.${schemaName}.${appName}`);
+}
+exports.autoUpgradeLockKey = autoUpgradeLockKey;
+/**
+ * A session-scoped advisory lock on one dedicated connection, held across passes. The connection
+ * is registered with the system database so its destroy() cuts it, lock included. CockroachDB
+ * has no advisory locks and always takes it, so it upgrades unprotected rather than not at all.
+ */
+class AutoUpgradeLease {
+    sysdb;
+    key;
+    #client = undefined;
+    held = false;
+    constructor(sysdb, appName) {
+        this.sysdb = sysdb;
+        this.key = autoUpgradeLockKey(sysdb.schemaName, appName).toString();
+    }
+    async acquire() {
+        if (await (0, cockroach_1.isCockroach)(this.sysdb)) {
+            this.held = true;
+            return true;
+        }
+        // Held on a valid connection.
+        if (this.held && this.#client && this.sysdb.retentionClients.has(this.#client)) {
+            return true;
+        }
+        // Otherwise, let go of whatever is left and take it afresh.
+        await this.release();
+        if (this.sysdb.destroyed) {
+            return false;
+        }
+        const client = await this.sysdb.connect();
+        this.sysdb.retentionClients.add(client);
+        this.#client = client;
+        try {
+            const { rows } = await client.query('SELECT pg_try_advisory_lock($1) AS locked', [this.key]);
+            this.held = rows[0]?.locked === true;
+        }
+        catch (e) {
+            await this.release();
+            throw e;
+        }
+        if (!this.held) {
+            await this.release();
+        }
+        return this.held;
+    }
+    async release() {
+        const client = this.#client;
+        const held = this.held;
+        this.#client = undefined;
+        this.held = false;
+        if (!client) {
+            return;
+        }
+        if (!this.sysdb.retentionClients.delete(client)) {
+            return;
+        }
+        try {
+            if (held) {
+                const { rows } = await client.query('SELECT pg_advisory_unlock($1) AS released', [
+                    this.key,
+                ]);
+                if (rows[0]?.released !== true) {
+                    this.sysdb.logger.warn('Could not release the auto-upgrade lock: this session no longer holds it.');
+                }
+            }
+        }
+        catch (e) {
+            this.sysdb.logger.warn(`Could not release the auto-upgrade lock: ${e.message}`);
+        }
+        finally {
+            client.release();
+        }
+    }
+}
+exports.AutoUpgradeLease = AutoUpgradeLease;
+/** Rows of this application, under an enrolled workflow's name. */
+function scopePredicate(sysdb, workflows, params) {
+    const app = sysdb.appNameFilter('application_name', sysdb.appName, params);
+    // A free function's class name is stored as NULL, which '' names in the registry.
+    params.push(workflows.map((w) => (w.className === '' ? null : w.className)));
+    const classNames = `$${params.length}::text[]`;
+    params.push(workflows.map((w) => w.name));
+    const names = `$${params.length}::text[]`;
+    return `${app} AND EXISTS (
+    SELECT 1 FROM unnest(${classNames}, ${names}) AS enrolled(class_name, name)
+    WHERE enrolled.name = ws.name AND enrolled.class_name IS NOT DISTINCT FROM ws.class_name)`;
+}
+/** Rows of one queued status, ENQUEUED or DELAYED, pinned to a version other than the latest. */
+function queuedPredicate(sysdb, workflows, latest, status) {
+    const params = [latest];
+    const sql = `ws.status = '${status}'
+    AND ${scopePredicate(sysdb, workflows, params)}
+    AND ws.application_version IS NOT NULL AND ws.application_version <> $1`;
+    return { sql, params };
+}
+/** PENDING rows on any version but the latest, none included. */
+function pendingPredicate(sysdb, workflows, latest) {
+    const params = [latest];
+    const sql = `ws.status = '${workflow_1.StatusString.PENDING}'
+    AND ${scopePredicate(sysdb, workflows, params)}
+    AND ws.application_version IS DISTINCT FROM $1`;
+    return { sql, params };
+}
+async function collect(sysdb, predicate) {
+    const { rows } = await (0, system_database_1.withDbRetry)(() => sysdb.pool.query(`SELECT ws.workflow_uuid FROM "${sysdb.schemaName}".workflow_status ws WHERE ${predicate.sql}`, predicate.params));
+    return rows.map((r) => r.workflow_uuid);
+}
+// Dropping the application-version: the row then runs on whatever version dequeues it, which only the latest one will.
+const UNPIN = { set: 'application_version = NULL', params: [] };
+// Upgrading a PENDING workflow is a bit like recovery: clear the ownership and re-enqueue with no version.
+const RECOVER = {
+    set: `status = '${workflow_1.StatusString.ENQUEUED}', application_version = NULL, owner_xid = NULL, started_at_epoch_ms = NULL,
+        updated_at = (EXTRACT(EPOCH FROM now()) * 1000)::bigint, queue_name = COALESCE(queue_name, $1)`,
+    params: [utils_1.INTERNAL_QUEUE_NAME],
+};
+/** The hook tests use to stop a pass while a chunk is in flight. */
+exports.chunkHooks = {
+    afterChunk: undefined,
+};
+async function moveChunk(sysdb, ids, predicate, move) {
+    // Move params first, so the predicate's placeholders can be shifted by a fixed amount.
+    const params = [...move.params];
+    const shift = params.length;
+    const where = predicate.sql.replace(/\$(\d+)/g, (_, n) => `$${Number(n) + shift}`);
+    params.push(...predicate.params);
+    params.push(ids);
+    // SKIP LOCKED: upgrading ENQUEUED/DELAYED rows competes with queue workers flipping them to PENDING,
+    // which the PENDING phase then resolves; upgrading PENDING rows competes with their executors, and
+    // either they win and make progress, or we win and clear ownership.
+    const result = await (0, system_database_1.withDbRetry)(() => sysdb.pool.query(`UPDATE "${sysdb.schemaName}".workflow_status
+       SET ${move.set}
+       WHERE workflow_uuid IN (
+         SELECT ws.workflow_uuid FROM "${sysdb.schemaName}".workflow_status ws
+         WHERE ws.workflow_uuid = ANY($${params.length}) AND ${where}
+         FOR UPDATE SKIP LOCKED)`, params));
+    exports.chunkHooks.afterChunk?.();
+    return result.rowCount ?? 0;
+}
+async function moveAll(sysdb, what, predicate, move, signal, chunkSize) {
+    // The collect and the chunks retry transient database failures, until the pass is stopped.
+    return (0, context_1.runWithAbortableDbRetries)(signal, async () => {
+        // Collect the IDs first: the application version is not indexed, so every chunk would
+        // otherwise walk the rows it already upgraded.
+        const ids = await collect(sysdb, predicate);
+        let moved = 0;
+        for (let start = 0; start < ids.length; start += chunkSize) {
+            if (signal.aborted)
+                break;
+            // Each chunk is its own transaction.
+            const rows = await moveChunk(sysdb, ids.slice(start, start + chunkSize), predicate, move);
+            moved += rows;
+            sysdb.logger.debug(`Auto-upgrade moved ${rows} ${what} workflows (chunk)`);
+        }
+        return moved;
+    });
+}
+/**
+ * Upgrade the enrolled workflows' work from previous application versions to `latestVersion`.
+ *
+ * Three phases, each collecting the rows to move and then moving them in chunks: the
+ * ENQUEUED rows, the DELAYED rows, then the PENDING rows. The PENDING phase is skipped until
+ * the latest version has been the latest for `graceMs`. Returns how many rows moved.
+ */
+async function upgradePass(sysdb, workflows, latestVersion, signal, options = {}) {
+    const chunkSize = options.chunkSize ?? exports.DEFAULT_CHUNK_SIZE;
+    const graceMs = options.graceMs ?? 0;
+    const latest = latestVersion.versionName;
+    // One collect per queued status so the query can use their partial index.
+    let moved = 0;
+    for (const [what, status] of [
+        ['enqueued', workflow_1.StatusString.ENQUEUED],
+        ['delayed', workflow_1.StatusString.DELAYED],
+    ]) {
+        if (signal.aborted)
+            break;
+        moved += await moveAll(sysdb, what, queuedPredicate(sysdb, workflows, latest, status), UNPIN, signal, chunkSize);
+    }
+    // PENDING rows wait out the grace period, measured from when the version became the latest.
+    const settledForMs = Date.now() - latestVersion.versionTimestamp;
+    if (settledForMs < graceMs) {
+        sysdb.logger.debug(`Auto-upgrade: version ${latest} became the latest ${Math.round(settledForMs / 1000)}s ago, ` +
+            `PENDING workflows wait out the ${graceMs}ms grace period`);
+        return moved;
+    }
+    if (!signal.aborted) {
+        moved += await moveAll(sysdb, 'pending', pendingPredicate(sysdb, workflows, latest), RECOVER, signal, chunkSize);
+    }
+    return moved;
+}
+exports.upgradePass = upgradePass;
+/** Satisfies core's AutoUpgradePass contract: core's loop calls run() on a timer and close() when it stops. */
+class AutoUpgradePass {
+    executor;
+    lease;
+    graceMs;
+    constructor(executor, options) {
+        this.executor = executor;
+        const appName = executor.systemDatabase.appName;
+        if (appName === undefined) {
+            throw new error_1.DBOSInitializationError('Auto-upgrading workflows requires an application name');
+        }
+        this.lease = new AutoUpgradeLease(executor.systemDatabase, appName);
+        this.graceMs = options.graceMs;
+    }
+    async close() {
+        await this.lease.release();
+    }
+    async run(signal) {
+        const sysdb = this.executor.systemDatabase;
+        sysdb.logger.debug('Starting auto-upgrade pass');
+        const workflows = this.executor.autoUpgradableWorkflows();
+        if (workflows.length === 0) {
+            return 0;
+        }
+        let latestVersion;
+        try {
+            latestVersion = await sysdb.getLatestApplicationVersion();
+        }
+        catch (e) {
+            // Nothing registered yet: there is no latest version to move work to.
+            if (e instanceof error_1.DBOSInitializationError)
+                return 0;
+            throw e;
+        }
+        if (latestVersion.versionName !== utils_1.globalParams.appVersion) {
+            // Only the latest version moves work. Let go of the lease if we hold it, so an executor
+            // on the latest version can take it.
+            sysdb.logger.debug(`Auto-upgrade: skipping the pass, this executor runs version ${utils_1.globalParams.appVersion} ` +
+                `and the latest is ${latestVersion.versionName}`);
+            await this.lease.release();
+            return 0;
+        }
+        if (!(await this.lease.acquire())) {
+            return 0;
+        }
+        const moved = await upgradePass(sysdb, workflows, latestVersion, signal, { graceMs: this.graceMs });
+        if (moved > 0) {
+            sysdb.logger.info(`Auto-upgrade moved ${moved} workflows to the latest application version`);
+        }
+        return moved;
+    }
+}
+exports.AutoUpgradePass = AutoUpgradePass;
+//# sourceMappingURL=auto_upgrade.js.map

package/dist/src/auto_upgrade.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto_upgrade.js","sourceRoot":"","sources":["../../src/auto_upgrade.ts"],"names":[],"mappings":";AAAA,kFAAkF;AAClF,EAAE;AACF,4FAA4F;AAC5F,EAAE;AACF,aAAa;AACb,sEAAsE;AACtE,sFAAsF;AACtF,sBAAsB;AACtB,EAAE;AACF,gFAAgF;;;AAEhF,2CAA0C;AAO1C,6DAA4E;AAC5E,iEAAgF;AAChF,iFAKqD;AACrD,6DAAsF;AACtF,mEAAoE;AAGvD,QAAA,kBAAkB,GAAG,IAAI,CAAC;AAEvC;;;;;GAKG;AACH,SAAgB,kBAAkB,CAAC,UAAkB,EAAE,OAAe;IACpE,OAAO,IAAA,iCAAe,EAAC,qBAAqB,UAAU,IAAI,OAAO,EAAE,CAAC,CAAC;AACvE,CAAC;AAFD,gDAEC;AAED;;;;GAIG;AACH,MAAa,gBAAgB;IAMhB;IALF,GAAG,CAAS;IACrB,OAAO,GAA2B,SAAS,CAAC;IAC5C,IAAI,GAAG,KAAK,CAAC;IAEb,YACW,KAAqB,EAC9B,OAAe;QADN,UAAK,GAAL,KAAK,CAAgB;QAG9B,IAAI,CAAC,GAAG,GAAG,kBAAkB,CAAC,KAAK,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,MAAM,IAAA,uBAAW,EAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YAClC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;YACjB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,8BAA8B;QAC9B,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/E,OAAO,IAAI,CAAC;QACd,CAAC;QACD,4DAA4D;QAC5D,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;YACzB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QAC1C,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC;YACH,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,KAAK,CAAsB,2CAA2C,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAClH,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;QACvC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YACrB,MAAM,CAAC,CAAC;QACV,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QACD,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC;QACzB,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC;QAClB,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAChD,OAAO;QACT,CAAC;QACD,IAAI,CAAC;YACH,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,KAAK,CAAwB,2CAA2C,EAAE;oBACtG,IAAI,CAAC,GAAG;iBACT,CAAC,CAAC;gBACH,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,QAAQ,KAAK,IAAI,EAAE,CAAC;oBAC/B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,2EAA2E,CAAC,CAAC;gBACtG,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,4CAA6C,CAAW,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7F,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;CACF;AApED,4CAoEC;AAQD,mEAAmE;AACnE,SAAS,cAAc,CAAC,KAAqB,EAAE,SAAsC,EAAE,MAAiB;IACtG,MAAM,GAAG,GAAG,KAAK,CAAC,aAAa,CAAC,kBAAkB,EAAE,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC3E,kFAAkF;IAClF,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAC7E,MAAM,UAAU,GAAG,IAAI,MAAM,CAAC,MAAM,UAAU,CAAC;IAC/C,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1C,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,MAAM,UAAU,CAAC;IAC1C,OAAO,GAAG,GAAG;2BACY,UAAU,KAAK,KAAK;8FAC+C,CAAC;AAC/F,CAAC;AAID,iGAAiG;AACjG,SAAS,eAAe,CACtB,KAAqB,EACrB,SAAsC,EACtC,MAAc,EACd,MAAoB;IAEpB,MAAM,MAAM,GAAc,CAAC,MAAM,CAAC,CAAC;IACnC,MAAM,GAAG,GAAG,gBAAgB,MAAM;UAC1B,cAAc,CAAC,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC;4EAC0B,CAAC;IAC3E,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC;AACzB,CAAC;AAED,iEAAiE;AACjE,SAAS,gBAAgB,CAAC,KAAqB,EAAE,SAAsC,EAAE,MAAc;IACrG,MAAM,MAAM,GAAc,CAAC,MAAM,CAAC,CAAC;IACnC,MAAM,GAAG,GAAG,gBAAgB,uBAAY,CAAC,OAAO;UACxC,cAAc,CAAC,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC;mDACC,CAAC;IAClD,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC;AACzB,CAAC;AAED,KAAK,UAAU,OAAO,CAAC,KAAqB,EAAE,SAAoB;IAChE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAA,6BAAW,EAAC,GAAG,EAAE,CACtC,KAAK,CAAC,IAAI,CAAC,KAAK,CACd,iCAAiC,KAAK,CAAC,UAAU,8BAA8B,SAAS,CAAC,GAAG,EAAE,EAC9F,SAAS,CAAC,MAAM,CACjB,CACF,CAAC;IACF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC;AAC1C,CAAC;AAQD,uHAAuH;AACvH,MAAM,KAAK,GAAS,EAAE,GAAG,EAAE,4BAA4B,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;AAEtE,2GAA2G;AAC3G,MAAM,OAAO,GAAS;IACpB,GAAG,EAAE,aAAa,uBAAY,CAAC,QAAQ;uGAC8D;IACrG,MAAM,EAAE,CAAC,2BAAmB,CAAC;CAC9B,CAAC;AAEF,oEAAoE;AACvD,QAAA,UAAU,GAAG;IACxB,UAAU,EAAE,SAAqC;CAClD,CAAC;AAEF,KAAK,UAAU,SAAS,CACtB,KAAqB,EACrB,GAAsB,EACtB,SAAoB,EACpB,IAAU;IAEV,uFAAuF;IACvF,MAAM,MAAM,GAAc,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;IAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC;IAC5B,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,CAAS,EAAE,EAAE,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,GAAG,KAAK,EAAE,CAAC,CAAC;IAC3F,MAAM,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;IACjC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACjB,qGAAqG;IACrG,mGAAmG;IACnG,oEAAoE;IACpE,MAAM,MAAM,GAAG,MAAM,IAAA,6BAAW,EAAC,GAAG,EAAE,CACpC,KAAK,CAAC,IAAI,CAAC,KAAK,CACd,WAAW,KAAK,CAAC,UAAU;aACpB,IAAI,CAAC,GAAG;;yCAEoB,KAAK,CAAC,UAAU;yCAChB,MAAM,CAAC,MAAM,SAAS,KAAK;iCACnC,EAC3B,MAAM,CACP,CACF,CAAC;IACF,kBAAU,CAAC,UAAU,EAAE,EAAE,CAAC;IAC1B,OAAO,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC;AAC9B,CAAC;AAED,KAAK,UAAU,OAAO,CACpB,KAAqB,EACrB,IAAY,EACZ,SAAoB,EACpB,IAAU,EACV,MAAmB,EACnB,SAAiB;IAEjB,2FAA2F;IAC3F,OAAO,IAAA,mCAAyB,EAAC,MAAM,EAAE,KAAK,IAAI,EAAE;QAClD,sFAAsF;QACtF,+CAA+C;QAC/C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC5C,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,GAAG,CAAC,MAAM,EAAE,KAAK,IAAI,SAAS,EAAE,CAAC;YAC3D,IAAI,MAAM,CAAC,OAAO;gBAAE,MAAM;YAC1B,qCAAqC;YACrC,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,SAAS,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;YAC1F,KAAK,IAAI,IAAI,CAAC;YACd,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,IAAI,IAAI,IAAI,oBAAoB,CAAC,CAAC;QAC7E,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC,CAAC,CAAC;AACL,CAAC;AAOD;;;;;;GAMG;AACI,KAAK,UAAU,WAAW,CAC/B,KAAqB,EACrB,SAAsC,EACtC,aAA0B,EAC1B,MAAmB,EACnB,UAA8B,EAAE;IAEhC,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,0BAAkB,CAAC;IAC1D,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,aAAa,CAAC,WAAW,CAAC;IAEzC,0EAA0E;IAC1E,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI;QAC3B,CAAC,UAAU,EAAE,uBAAY,CAAC,QAAQ,CAAC;QACnC,CAAC,SAAS,EAAE,uBAAY,CAAC,OAAO,CAAC;KACzB,EAAE,CAAC;QACX,IAAI,MAAM,CAAC,OAAO;YAAE,MAAM;QAC1B,KAAK,IAAI,MAAM,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,eAAe,CAAC,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IACnH,CAAC;IAED,4FAA4F;IAC5F,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,aAAa,CAAC,gBAAgB,CAAC;IACjE,IAAI,YAAY,GAAG,OAAO,EAAE,CAAC;QAC3B,KAAK,CAAC,MAAM,CAAC,KAAK,CAChB,yBAAyB,MAAM,sBAAsB,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS;YAC3F,kCAAkC,OAAO,iBAAiB,CAC7D,CAAC;QACF,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,KAAK,IAAI,MAAM,OAAO,CAAC,KAAK,EAAE,SAAS,EAAE,gBAAgB,CAAC,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IACnH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAlCD,kCAkCC;AAED,+GAA+G;AAC/G,MAAa,eAAe;IAKf;IAJF,KAAK,CAAmB;IACxB,OAAO,CAAS;IAEzB,YACW,QAAsB,EAC/B,OAA2B;QADlB,aAAQ,GAAR,QAAQ,CAAc;QAG/B,MAAM,OAAO,GAAG,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC;QAChD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YAC1B,MAAM,IAAI,+BAAuB,CAAC,uDAAuD,CAAC,CAAC;QAC7F,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,IAAI,gBAAgB,CAAC,QAAQ,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QACpE,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAAmB;QAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC;QAC3C,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;QACjD,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,uBAAuB,EAAE,CAAC;QAC1D,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,OAAO,CAAC,CAAC;QACX,CAAC;QACD,IAAI,aAA0B,CAAC;QAC/B,IAAI,CAAC;YACH,aAAa,GAAG,MAAM,KAAK,CAAC,2BAA2B,EAAE,CAAC;QAC5D,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,sEAAsE;YACtE,IAAI,CAAC,YAAY,+BAAuB;gBAAE,OAAO,CAAC,CAAC;YACnD,MAAM,CAAC,CAAC;QACV,CAAC;QACD,IAAI,aAAa,CAAC,WAAW,KAAK,oBAAY,CAAC,UAAU,EAAE,CAAC;YAC1D,wFAAwF;YACxF,qCAAqC;YACrC,KAAK,CAAC,MAAM,CAAC,KAAK,CAChB,+DAA+D,oBAAY,CAAC,UAAU,GAAG;gBACvF,qBAAqB,aAAa,CAAC,WAAW,EAAE,CACnD,CAAC;YACF,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YAC3B,OAAO,CAAC,CAAC;QACX,CAAC;QACD,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YAClC,OAAO,CAAC,CAAC;QACX,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,WAAW,CAAC,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QACpG,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACd,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,KAAK,8CAA8C,CAAC,CAAC;QAC/F,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAtDD,0CAsDC"}

package/dist/src/cockroach.d.ts

@@ -0,0 +1,4 @@
+import { SystemDatabase } from '@dbos-inc/dbos-sdk/internal/system_database';
+/** Whether the system database is CockroachDB, asked once per system database. */
+export declare function isCockroach(sysdb: SystemDatabase): Promise<boolean>;
+//# sourceMappingURL=cockroach.d.ts.map

package/dist/src/cockroach.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cockroach.d.ts","sourceRoot":"","sources":["../../src/cockroach.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,cAAc,EAAE,MAAM,6CAA6C,CAAC;AAI5F,kFAAkF;AAClF,wBAAsB,WAAW,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,OAAO,CAAC,CAYzE"}

package/dist/src/cockroach.js

@@ -0,0 +1,22 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isCockroach = void 0;
+const system_database_1 = require("@dbos-inc/dbos-sdk/internal/system_database");
+const cockroachBySysDb = new WeakMap();
+/** Whether the system database is CockroachDB, asked once per system database. */
+async function isCockroach(sysdb) {
+    let cockroach = cockroachBySysDb.get(sysdb);
+    if (cockroach === undefined) {
+        const client = await sysdb.connect();
+        try {
+            cockroach = await (0, system_database_1.isCockroachDB)(client);
+        }
+        finally {
+            client.release();
+        }
+        cockroachBySysDb.set(sysdb, cockroach);
+    }
+    return cockroach;
+}
+exports.isCockroach = isCockroach;
+//# sourceMappingURL=cockroach.js.map

package/dist/src/cockroach.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cockroach.js","sourceRoot":"","sources":["../../src/cockroach.ts"],"names":[],"mappings":";;;AAAA,iFAA4F;AAE5F,MAAM,gBAAgB,GAAG,IAAI,OAAO,EAA2B,CAAC;AAEhE,kFAAkF;AAC3E,KAAK,UAAU,WAAW,CAAC,KAAqB;IACrD,IAAI,SAAS,GAAG,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC5C,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;QACrC,IAAI,CAAC;YACH,SAAS,GAAG,MAAM,IAAA,+BAAa,EAAC,MAAM,CAAC,CAAC;QAC1C,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;QACD,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IACzC,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAZD,kCAYC"}

package/dist/src/conductor/conductor.d.ts

@@ -0,0 +1,39 @@
+/// <reference types="node" />
+/// <reference types="node" />
+import { DBOSExecutor } from '@dbos-inc/dbos-sdk/internal/dbos-executor';
+import WebSocket from 'ws';
+import type { ConductorConnection, ConductorOptions } from '@dbos-inc/dbos-sdk/internal/enterprise';
+interface IntervalTimeout {
+    interval: NodeJS.Timeout | undefined;
+    timeout: NodeJS.Timeout | undefined;
+}
+export declare class ConductorWebsocket implements ConductorConnection {
+    readonly dbosExec: DBOSExecutor;
+    url: string;
+    websocket: WebSocket | undefined;
+    isShuttingDown: boolean;
+    pingPeriodMs: number;
+    pingTimeoutMs: number;
+    pingIntervalTimeout: IntervalTimeout | undefined;
+    reconnectDelayMs: number;
+    reconnectTimeout: NodeJS.Timeout | undefined;
+    retentionRound: Promise<void> | undefined;
+    readonly appName: string;
+    readonly executorMetadata?: Record<string, unknown>;
+    readonly metadataOnlyMode: boolean;
+    constructor(dbosExec: DBOSExecutor, options: ConductorOptions);
+    start(): void;
+    /** Logs a command's exception in full and returns the error message Conductor may see. */
+    reportException(context: string, e: unknown): string;
+    resetWebsocket(currWebsocket?: WebSocket, currPing?: IntervalTimeout): void;
+    setPingInterval(currWebsocket: WebSocket, currPing: IntervalTimeout): void;
+    dispatchLoop(): void;
+    /**
+     * Give a round in flight a bounded chance to finish before shutdown takes the system
+     * database away from it.
+     */
+    awaitRetention(timeoutMs?: number): Promise<void>;
+    stop(): Promise<void>;
+}
+export {};
+//# sourceMappingURL=conductor.d.ts.map

package/dist/src/conductor/conductor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"conductor.d.ts","sourceRoot":"","sources":["../../../src/conductor/conductor.ts"],"names":[],"mappings":";;AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,2CAA2C,CAAC;AAGzE,OAAO,SAAS,MAAM,IAAI,CAAC;AAO3B,OAAO,KAAK,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AA2BpG,UAAU,eAAe;IACvB,QAAQ,EAAE,MAAM,CAAC,OAAO,GAAG,SAAS,CAAC;IACrC,OAAO,EAAE,MAAM,CAAC,OAAO,GAAG,SAAS,CAAC;CACrC;AAED,qBAAa,kBAAmB,YAAW,mBAAmB;IAiB1D,QAAQ,CAAC,QAAQ,EAAE,YAAY;IAhBjC,GAAG,EAAE,MAAM,CAAC;IACZ,SAAS,EAAE,SAAS,GAAG,SAAS,CAAa;IAC7C,cAAc,UAAS;IACvB,YAAY,SAAS;IACrB,aAAa,SAAS;IACtB,mBAAmB,EAAE,eAAe,GAAG,SAAS,CAAa;IAC7D,gBAAgB,SAAQ;IACxB,gBAAgB,EAAE,MAAM,CAAC,OAAO,GAAG,SAAS,CAAa;IAEzD,cAAc,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,SAAS,CAAa;IAEtD,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACpD,QAAQ,CAAC,gBAAgB,EAAE,OAAO,CAAC;gBAGxB,QAAQ,EAAE,YAAY,EAC/B,OAAO,EAAE,gBAAgB;IAU3B,KAAK;IAIL,0FAA0F;IAC1F,eAAe,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,OAAO,GAAG,MAAM;IAUpD,cAAc,CAAC,aAAa,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,eAAe;IAoBpE,eAAe,CAAC,aAAa,EAAE,SAAS,EAAE,QAAQ,EAAE,eAAe;IAuBnE,YAAY;IAk5BZ;;;OAGG;IACG,cAAc,CAAC,SAAS,GAAE,MAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAaxD,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;CAW5B"}