@rudderjs/database 1.1.0 → 1.2.1

package/dist/native/schema/types-generator.js

@@ -0,0 +1,181 @@
+// ─── Schema → TypeScript types generator (GATE 7-types) ────
+//
+// The headline of the migrations plan: after `migrate`, a model's column types
+// are GENERATED from the live schema, not hand-maintained — so they can't drift.
+// This module is the PURE core: it turns introspected columns (+ a model's
+// declared `casts`) into a `SchemaRegistry` `.d.ts` that augments `@rudderjs/orm`,
+// which `Model<'table'>` then resolves. Mirrors `@rudderjs/vite`'s scanner
+// emitting `pages/__view/registry.d.ts` (same "generated .d.ts augments framework
+// types" pattern).
+//
+// PURE: string building only. The node-only orchestration (introspect every
+// table → write the file) lives in `schema-types.ts`; the runtime DB reads live
+// in `introspect.ts`. Keeping the mapping pure makes the column→TS contract
+// directly unit-testable.
+/**
+ * Map a SQLite declared type (`PRAGMA table_info.type`) to a base TS type. The
+ * mapping is by affinity rules — SQLite types are fuzzy, so we match on
+ * substrings the same way SQLite's own affinity algorithm does (INT→integer,
+ * CHAR/CLOB/TEXT→text, BLOB→blob, REAL/FLOA/DOUB→real, else numeric).
+ */
+export function sqliteTypeToTs(declared) {
+    const t = declared.toUpperCase();
+    if (t.includes('INT'))
+        return 'number';
+    if (t.includes('CHAR') || t.includes('CLOB') || t.includes('TEXT'))
+        return 'string';
+    if (t.includes('BLOB') || t === '')
+        return 'Uint8Array';
+    if (t.includes('REAL') || t.includes('FLOA') || t.includes('DOUB'))
+        return 'number';
+    return 'number'; // NUMERIC / DECIMAL affinity — stored as number
+}
+/**
+ * Map a Postgres `information_schema.columns.data_type` to a base TS type. The
+ * mapping reflects what the porsager driver actually returns on READ (so the
+ * generated type matches runtime), which `casts` then refine:
+ *  - `int8`/`bigint` → `number` (the pg driver parses OID 20 to a JS number);
+ *  - `numeric`/`money` → `string` (porsager keeps these as strings for
+ *    precision safety — a `float`/`decimal` cast refines them to `number`);
+ *  - `json`/`jsonb` → `unknown` (a `json` cast or a typed `casts` entry refines);
+ *  - `timestamp*`/`date` → `Date`; `bytea` → `Uint8Array`; the rest → `string`.
+ */
+export function pgTypeToTs(declared) {
+    const t = declared.toLowerCase();
+    switch (t) {
+        case 'boolean': return 'boolean';
+        case 'smallint':
+        case 'integer':
+        case 'bigint':
+        case 'real':
+        case 'double precision': return 'number';
+        case 'numeric':
+        case 'money': return 'string';
+        case 'json':
+        case 'jsonb': return 'unknown';
+        case 'bytea': return 'Uint8Array';
+        case 'date': return 'Date';
+        default:
+            // varchar/char surface as `character varying`/`character`; timestamps as
+            // `timestamp with/without time zone`; times as `time …`. Match by prefix.
+            if (t.startsWith('timestamp'))
+                return 'Date';
+            if (t.startsWith('character') || t.startsWith('time') || t === 'text' || t === 'uuid')
+                return 'string';
+            return 'unknown';
+    }
+}
+/**
+ * Map a MySQL `information_schema.columns.data_type` to a base TS type. The
+ * `data_type` is the base type WITHOUT length/precision, which `casts` then
+ * refine:
+ *  - integer family (`tinyint`/`smallint`/`mediumint`/`int`/`bigint`) → `number`.
+ *    Note `tinyint(1)` (MySQL's BOOLEAN alias) surfaces as `tinyint` here → `number`;
+ *    a declared `boolean` cast refines it to `boolean` (same treatment as pg);
+ *  - `decimal`/`numeric` → `string` (precision safety, matching pg — a
+ *    `float`/`decimal` cast refines to `number`);
+ *  - `float`/`double`/`real` → `number`;
+ *  - `json` → `unknown` (a `json` cast or typed `casts` entry refines);
+ *  - `date`/`datetime`/`timestamp` → `Date`; `blob`/`binary` → `Uint8Array`;
+ *  - char/text families → `string`; everything else → `unknown`.
+ */
+export function mysqlTypeToTs(declared) {
+    const t = declared.toLowerCase();
+    switch (t) {
+        case 'tinyint':
+        case 'smallint':
+        case 'mediumint':
+        case 'int':
+        case 'integer':
+        case 'bigint': return 'number';
+        case 'decimal':
+        case 'numeric': return 'string';
+        case 'float':
+        case 'double':
+        case 'double precision':
+        case 'real': return 'number';
+        case 'json': return 'unknown';
+        case 'date':
+        case 'datetime':
+        case 'timestamp': return 'Date';
+        default:
+            // varchar/char/text families → string; blob/binary families → Uint8Array.
+            if (t.includes('char') || t.includes('text') || t === 'enum' || t === 'set')
+                return 'string';
+            if (t.includes('blob') || t.includes('binary'))
+                return 'Uint8Array';
+            return 'unknown';
+    }
+}
+/**
+ * The TS type a declared cast produces on READ (`toJSON`/hydration). Casts
+ * OVERRIDE the storage type — a `boolean` cast turns an INTEGER column into
+ * `boolean`, a `json` cast turns TEXT into `unknown`, etc. Returns null for
+ * casts that don't change the read type (or unknown cast names), so the caller
+ * falls back to the storage mapping.
+ */
+export function castToTs(cast) {
+    switch (cast) {
+        case 'boolean': return 'boolean';
+        case 'integer': return 'number';
+        case 'float': return 'number';
+        case 'string': return 'string';
+        case 'encrypted': return 'string';
+        case 'date': return 'Date';
+        case 'datetime': return 'Date';
+        case 'json': return 'unknown';
+        case 'array': return 'unknown[]';
+        case 'encrypted:array': return 'unknown[]';
+        case 'collection': return 'unknown[]';
+        case 'encrypted:object': return 'Record<string, unknown>';
+        default:
+            // `vector(...)` and custom CastUsing classes resolve to number[] / unknown;
+            // be conservative — unknown name → let storage type win.
+            return null;
+    }
+}
+/**
+ * Resolve one column's TS type: a declared cast wins over the storage mapping,
+ * and a nullable column widens with `| null`. The primary key and NOT NULL
+ * columns stay non-null. `typeToTs` is the per-dialect storage mapper
+ * ({@link sqliteTypeToTs} by default; {@link pgTypeToTs} for Postgres).
+ */
+export function resolveColumnType(col, casts, typeToTs = sqliteTypeToTs) {
+    const declaredCast = casts[col.name];
+    const base = (declaredCast && castToTs(declaredCast)) || typeToTs(col.type);
+    // A column is nullable on read when it permits NULL and isn't the PK.
+    const nullable = !col.notNull && col.pk === 0;
+    return { name: col.name, ts: nullable ? `${base} | null` : base };
+}
+/** Build one table's {@link TableTypes} from its columns + casts, using the
+ *  given per-dialect storage mapper (defaults to the SQLite mapping). */
+export function buildTableTypes(table, columns, casts = {}, typeToTs = sqliteTypeToTs) {
+    return { table, columns: columns.map((c) => resolveColumnType(c, casts, typeToTs)) };
+}
+/**
+ * Emit the full `registry.d.ts` contents: a `declare module '@rudderjs/orm'`
+ * augmentation extending `SchemaRegistry` with one entry per table. Tables are
+ * sorted for deterministic output (stable diffs / git). An empty schema still
+ * emits a valid (empty) augmentation so a stale file is always overwritten.
+ */
+export function emitRegistryDts(tables) {
+    const sorted = [...tables].sort((a, b) => a.table.localeCompare(b.table));
+    const entries = sorted.map((t) => {
+        const cols = t.columns.map((c) => `      ${quoteKey(c.name)}: ${c.ts}`).join('\n');
+        return `    ${quoteKey(t.table)}: {\n${cols}\n    }`;
+    }).join('\n');
+    return (`// AUTO-GENERATED by @rudderjs/orm — do not edit.\n` +
+        `// Source: the migrated database schema (run \`rudder schema:types\` or \`migrate\`).\n` +
+        `import '@rudderjs/orm'\n\n` +
+        `declare module '@rudderjs/orm' {\n` +
+        `  interface SchemaRegistry {\n` +
+        `${entries}\n` +
+        `  }\n` +
+        `}\n`);
+}
+/** Quote an object key only when it isn't a plain JS identifier (keeps the
+ *  common case clean, stays correct for snake_case-with-dashes or odd names). */
+function quoteKey(key) {
+    return /^[A-Za-z_$][A-Za-z0-9_$]*$/.test(key) ? key : JSON.stringify(key);
+}
+//# sourceMappingURL=types-generator.js.map

package/dist/native/schema/types-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types-generator.js","sourceRoot":"","sources":["../../../src/native/schema/types-generator.ts"],"names":[],"mappings":"AAAA,8DAA8D;AAC9D,EAAE;AACF,+EAA+E;AAC/E,iFAAiF;AACjF,2EAA2E;AAC3E,mFAAmF;AACnF,2EAA2E;AAC3E,kFAAkF;AAClF,mBAAmB;AACnB,EAAE;AACF,4EAA4E;AAC5E,gFAAgF;AAChF,4EAA4E;AAC5E,0BAA0B;AAmB1B;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAAC,QAAgB;IAC7C,MAAM,CAAC,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAA;IAChC,IAAI,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,QAAQ,CAAA;IACtC,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,QAAQ,CAAA;IACnF,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE;QAAE,OAAO,YAAY,CAAA;IACvD,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,QAAQ,CAAA;IACnF,OAAO,QAAQ,CAAA,CAAC,gDAAgD;AAClE,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,UAAU,UAAU,CAAC,QAAgB;IACzC,MAAM,CAAC,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAA;IAChC,QAAQ,CAAC,EAAE,CAAC;QACV,KAAK,SAAS,CAAC,CAAqB,OAAO,SAAS,CAAA;QACpD,KAAK,UAAU,CAAC;QAChB,KAAK,SAAS,CAAC;QACf,KAAK,QAAQ,CAAC;QACd,KAAK,MAAM,CAAC;QACZ,KAAK,kBAAkB,CAAC,CAAY,OAAO,QAAQ,CAAA;QACnD,KAAK,SAAS,CAAC;QACf,KAAK,OAAO,CAAC,CAAuB,OAAO,QAAQ,CAAA;QACnD,KAAK,MAAM,CAAC;QACZ,KAAK,OAAO,CAAC,CAAuB,OAAO,SAAS,CAAA;QACpD,KAAK,OAAO,CAAC,CAAuB,OAAO,YAAY,CAAA;QACvD,KAAK,MAAM,CAAC,CAAwB,OAAO,MAAM,CAAA;QACjD;YACE,yEAAyE;YACzE,0EAA0E;YAC1E,IAAI,CAAC,CAAC,UAAU,CAAC,WAAW,CAAC;gBAAE,OAAO,MAAM,CAAA;YAC5C,IAAI,CAAC,CAAC,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,MAAM,IAAI,CAAC,KAAK,MAAM;gBAAE,OAAO,QAAQ,CAAA;YACtG,OAAO,SAAS,CAAA;IACpB,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,aAAa,CAAC,QAAgB;IAC5C,MAAM,CAAC,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAA;IAChC,QAAQ,CAAC,EAAE,CAAC;QACV,KAAK,SAAS,CAAC;QACf,KAAK,UAAU,CAAC;QAChB,KAAK,WAAW,CAAC;QACjB,KAAK,KAAK,CAAC;QACX,KAAK,SAAS,CAAC;QACf,KAAK,QAAQ,CAAC,CAAgB,OAAO,QAAQ,CAAA;QAC7C,KAAK,SAAS,CAAC;QACf,KAAK,SAAS,CAAC,CAAe,OAAO,QAAQ,CAAA;QAC7C,KAAK,OAAO,CAAC;QACb,KAAK,QAAQ,CAAC;QACd,KAAK,kBAAkB,CAAC;QACxB,KAAK,MAAM,CAAC,CAAkB,OAAO,QAAQ,CAAA;QAC7C,KAAK,MAAM,CAAC,CAAkB,OAAO,SAAS,CAAA;QAC9C,KAAK,MAAM,CAAC;QACZ,KAAK,UAAU,CAAC;QAChB,KAAK,WAAW,CAAC,CAAa,OAAO,MAAM,CAAA;QAC3C;YACE,0EAA0E;YAC1E,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,MAAM,IAAI,CAAC,KAAK,KAAK;gBAAE,OAAO,QAAQ,CAAA;YAC5F,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBAAE,OAAO,YAAY,CAAA;YACnE,OAAO,SAAS,CAAA;IACpB,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,QAAQ,CAAC,IAA0B;IACjD,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,SAAS,CAAC,CAAW,OAAO,SAAS,CAAA;QAC1C,KAAK,SAAS,CAAC,CAAW,OAAO,QAAQ,CAAA;QACzC,KAAK,OAAO,CAAC,CAAa,OAAO,QAAQ,CAAA;QACzC,KAAK,QAAQ,CAAC,CAAY,OAAO,QAAQ,CAAA;QACzC,KAAK,WAAW,CAAC,CAAS,OAAO,QAAQ,CAAA;QACzC,KAAK,MAAM,CAAC,CAAc,OAAO,MAAM,CAAA;QACvC,KAAK,UAAU,CAAC,CAAU,OAAO,MAAM,CAAA;QACvC,KAAK,MAAM,CAAC,CAAc,OAAO,SAAS,CAAA;QAC1C,KAAK,OAAO,CAAC,CAAa,OAAO,WAAW,CAAA;QAC5C,KAAK,iBAAiB,CAAC,CAAG,OAAO,WAAW,CAAA;QAC5C,KAAK,YAAY,CAAC,CAAQ,OAAO,WAAW,CAAA;QAC5C,KAAK,kBAAkB,CAAC,CAAE,OAAO,yBAAyB,CAAA;QAC1D;YACE,4EAA4E;YAC5E,yDAAyD;YACzD,OAAO,IAAI,CAAA;IACf,CAAC;AACH,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,iBAAiB,CAC/B,GAAc,EACd,KAA6B,EAC7B,WAAyC,cAAc;IAEvD,MAAM,YAAY,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;IACpC,MAAM,IAAI,GAAG,CAAC,YAAY,IAAI,QAAQ,CAAC,YAAY,CAAC,CAAC,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;IAC3E,sEAAsE;IACtE,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC,EAAE,KAAK,CAAC,CAAA;IAC7C,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,EAAE,EAAE,QAAQ,CAAC,CAAC,CAAC,GAAG,IAAI,SAAS,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;AACnE,CAAC;AAED;yEACyE;AACzE,MAAM,UAAU,eAAe,CAC7B,KAAa,EACb,OAAoB,EACpB,QAAgC,EAAE,EAClC,WAAyC,cAAc;IAEvD,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC,EAAE,CAAA;AACtF,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,eAAe,CAAC,MAAoB;IAClD,MAAM,MAAM,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAA;IACzE,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAC/B,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAClF,OAAO,OAAO,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,IAAI,SAAS,CAAA;IACtD,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAEb,OAAO,CACL,qDAAqD;QACrD,yFAAyF;QACzF,4BAA4B;QAC5B,oCAAoC;QACpC,gCAAgC;QAChC,GAAG,OAAO,IAAI;QACd,OAAO;QACP,KAAK,CACN,CAAA;AACH,CAAC;AAED;iFACiF;AACjF,SAAS,QAAQ,CAAC,GAAW;IAC3B,OAAO,4BAA4B,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAA;AAC3E,CAAC"}

package/dist/registry-bridge.d.ts

@@ -1,13 +1,14 @@
-import type { OrmAdapter } from '@rudderjs/contracts';
+import type { OrmAdapter, TransactionOptions } from '@rudderjs/contracts';
 /**
  * Runs `fn` inside a database transaction, returning its result. Pushed in by
  * `@rudderjs/orm` (its `transaction()` free function) so `DB.transaction()` reuses
  * the ORM's `AsyncLocalStorage` transaction scoping — every `Model.*` AND `DB.*`
  * call inside `fn` joins the *same* open transaction (one connection, not two).
  * `@rudderjs/database` can't import `@rudderjs/orm`, so the runner is injected the
- * same way the adapter resolver is.
+ * same way the adapter resolver is. `opts` (e.g. `isolationLevel`) flows through
+ * to the adapter untouched.
  */
-export type TransactionRunner = <T>(fn: () => Promise<T>) => Promise<T>;
+export type TransactionRunner = <T>(fn: () => Promise<T>, opts?: TransactionOptions) => Promise<T>;
 /**
  * Resolves the adapter for a NAMED connection (`DB.connection('reporting')`),
  * opening it lazily on first use. Pushed in by `@rudderjs/orm`'s `db-bridge`
@@ -19,7 +20,17 @@ export type TransactionRunner = <T>(fn: () => Promise<T>) => Promise<T>;
 export type ConnectionResolver = (name: string) => Promise<OrmAdapter>;
 /** Runs `fn` inside a transaction on a NAMED connection — the named
  *  counterpart of {@link TransactionRunner}, same injection rationale. */
-export type NamedTransactionRunner = <T>(name: string, fn: () => Promise<T>) => Promise<T>;
+export type NamedTransactionRunner = <T>(name: string, fn: () => Promise<T>, opts?: TransactionOptions) => Promise<T>;
+/**
+ * Queues `fn` to run after the current transaction commits (drops it on
+ * rollback; runs immediately when no transaction is open). Pushed in by
+ * `@rudderjs/orm` (its `afterCommit()` free function, which owns the
+ * transaction-tree bookkeeping) so `DB.afterCommit()` shares the queue with
+ * `Model.*` / `transaction()` call sites.
+ */
+export type AfterCommitRunner = (fn: () => void | Promise<void>, opts?: {
+    connection?: string;
+}) => Promise<void>;
 /**
  * Register the function that resolves the active ORM adapter. Called once by
  * `@rudderjs/orm`'s `db-bridge` module (imported for side effect from each
@@ -66,4 +77,13 @@ export declare function registerNamedTransactionRunner(fn: NamedTransactionRunne
 /** Resolve the named transaction runner. Throws a clear error when none has
  *  been registered. */
 export declare function resolveNamedTransactionRunner(): NamedTransactionRunner;
+/**
+ * Register the function that queues after-commit callbacks. Called by
+ * `@rudderjs/orm`'s `db-bridge` module. Idempotent — last registration wins.
+ */
+export declare function registerAfterCommitRunner(fn: AfterCommitRunner): void;
+/** Resolve the after-commit runner. Throws a clear error when none has been
+ *  registered — i.e. `@rudderjs/orm` (and a database provider) wasn't loaded,
+ *  or the installed `@rudderjs/orm` predates after-commit support. */
+export declare function resolveAfterCommitRunner(): AfterCommitRunner;
 //# sourceMappingURL=registry-bridge.d.ts.map

package/dist/registry-bridge.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"registry-bridge.d.ts","sourceRoot":"","sources":["../src/registry-bridge.ts"],"names":[],"mappings":"AAaA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAA;AAErD;;;;;;;GAOG;AACH,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,CAAA;AAEvE;;;;;;;GAOG;AACH,MAAM,MAAM,kBAAkB,GAAG,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,UAAU,CAAC,CAAA;AAEtE;0EAC0E;AAC1E,MAAM,MAAM,sBAAsB,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,CAAA;AAO1F;;;;;GAKG;AACH,wBAAgB,uBAAuB,CAAC,EAAE,EAAE,MAAM,UAAU,GAAG,IAAI,CAElE;AAED;;;;GAIG;AACH,wBAAgB,cAAc,IAAI,UAAU,CAS3C;AAED;;;;;GAKG;AACH,wBAAgB,yBAAyB,CAAC,EAAE,EAAE,iBAAiB,GAAG,IAAI,CAErE;AAED;;;GAGG;AACH,wBAAgB,wBAAwB,IAAI,iBAAiB,CAS5D;AAED;;;;GAIG;AACH,wBAAgB,0BAA0B,CAAC,EAAE,EAAE,kBAAkB,GAAG,IAAI,CAEvE;AAED;;;;GAIG;AACH,wBAAgB,yBAAyB,IAAI,kBAAkB,CAU9D;AAED;;;;GAIG;AACH,wBAAgB,8BAA8B,CAAC,EAAE,EAAE,sBAAsB,GAAG,IAAI,CAE/E;AAED;uBACuB;AACvB,wBAAgB,6BAA6B,IAAI,sBAAsB,CAStE"}
+{"version":3,"file":"registry-bridge.d.ts","sourceRoot":"","sources":["../src/registry-bridge.ts"],"names":[],"mappings":"AAaA,OAAO,KAAK,EAAE,UAAU,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAA;AAEzE;;;;;;;;GAQG;AACH,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,kBAAkB,KAAK,OAAO,CAAC,CAAC,CAAC,CAAA;AAElG;;;;;;;GAOG;AACH,MAAM,MAAM,kBAAkB,GAAG,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,UAAU,CAAC,CAAA;AAEtE;0EAC0E;AAC1E,MAAM,MAAM,sBAAsB,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,kBAAkB,KAAK,OAAO,CAAC,CAAC,CAAC,CAAA;AAErH;;;;;;GAMG;AACH,MAAM,MAAM,iBAAiB,GAAG,CAAC,EAAE,EAAE,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,EAAE;IAAE,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAA;AAQjH;;;;;GAKG;AACH,wBAAgB,uBAAuB,CAAC,EAAE,EAAE,MAAM,UAAU,GAAG,IAAI,CAElE;AAED;;;;GAIG;AACH,wBAAgB,cAAc,IAAI,UAAU,CAS3C;AAED;;;;;GAKG;AACH,wBAAgB,yBAAyB,CAAC,EAAE,EAAE,iBAAiB,GAAG,IAAI,CAErE;AAED;;;GAGG;AACH,wBAAgB,wBAAwB,IAAI,iBAAiB,CAS5D;AAED;;;;GAIG;AACH,wBAAgB,0BAA0B,CAAC,EAAE,EAAE,kBAAkB,GAAG,IAAI,CAEvE;AAED;;;;GAIG;AACH,wBAAgB,yBAAyB,IAAI,kBAAkB,CAU9D;AAED;;;;GAIG;AACH,wBAAgB,8BAA8B,CAAC,EAAE,EAAE,sBAAsB,GAAG,IAAI,CAE/E;AAED;uBACuB;AACvB,wBAAgB,6BAA6B,IAAI,sBAAsB,CAStE;AAED;;;GAGG;AACH,wBAAgB,yBAAyB,CAAC,EAAE,EAAE,iBAAiB,GAAG,IAAI,CAErE;AAED;;sEAEsE;AACtE,wBAAgB,wBAAwB,IAAI,iBAAiB,CAS5D"}

package/dist/registry-bridge.js

@@ -14,6 +14,7 @@ let resolver = null;
 let txRunner = null;
 let connectionResolver = null;
 let namedTxRunner = null;
+let afterCommitRunner = null;
 /**
  * Register the function that resolves the active ORM adapter. Called once by
  * `@rudderjs/orm`'s `db-bridge` module (imported for side effect from each
@@ -97,11 +98,30 @@ export function resolveNamedTransactionRunner() {
     }
     return namedTxRunner;
 }
+/**
+ * Register the function that queues after-commit callbacks. Called by
+ * `@rudderjs/orm`'s `db-bridge` module. Idempotent — last registration wins.
+ */
+export function registerAfterCommitRunner(fn) {
+    afterCommitRunner = fn;
+}
+/** Resolve the after-commit runner. Throws a clear error when none has been
+ *  registered — i.e. `@rudderjs/orm` (and a database provider) wasn't loaded,
+ *  or the installed `@rudderjs/orm` predates after-commit support. */
+export function resolveAfterCommitRunner() {
+    if (!afterCommitRunner) {
+        throw new Error('[RudderJS DB] No after-commit runner is available. DB.afterCommit() runs ' +
+            'through @rudderjs/orm — make sure a database provider is registered (and ' +
+            '@rudderjs/orm installed) before calling DB.afterCommit().');
+    }
+    return afterCommitRunner;
+}
 /** @internal — test-only reset of the registered resolvers + transaction runners. */
 export function __resetAdapterResolver() {
     resolver = null;
     txRunner = null;
     connectionResolver = null;
     namedTxRunner = null;
+    afterCommitRunner = null;
 }
 //# sourceMappingURL=registry-bridge.js.map

package/dist/registry-bridge.js.map

@@ -1 +1 @@
-{"version":3,"file":"registry-bridge.js","sourceRoot":"","sources":["../src/registry-bridge.ts"],"names":[],"mappings":"AAAA,8DAA8D;AAC9D,EAAE;AACF,iFAAiF;AACjF,4EAA4E;AAC5E,4EAA4E;AAC5E,+EAA+E;AAC/E,2EAA2E;AAC3E,0BAA0B;AAC1B,EAAE;AACF,8EAA8E;AAC9E,4EAA4E;AAC5E,+EAA+E;AA4B/E,IAAI,QAAQ,GAA8B,IAAI,CAAA;AAC9C,IAAI,QAAQ,GAA6B,IAAI,CAAA;AAC7C,IAAI,kBAAkB,GAA8B,IAAI,CAAA;AACxD,IAAI,aAAa,GAAkC,IAAI,CAAA;AAEvD;;;;;GAKG;AACH,MAAM,UAAU,uBAAuB,CAAC,EAAoB;IAC1D,QAAQ,GAAG,EAAE,CAAA;AACf,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,cAAc;IAC5B,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CACb,wEAAwE;YACtE,uEAAuE;YACvE,8DAA8D,CACjE,CAAA;IACH,CAAC;IACD,OAAO,QAAQ,EAAE,CAAA;AACnB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,yBAAyB,CAAC,EAAqB;IAC7D,QAAQ,GAAG,EAAE,CAAA;AACf,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,wBAAwB;IACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CACb,0EAA0E;YACxE,2EAA2E;YAC3E,2DAA2D,CAC9D,CAAA;IACH,CAAC;IACD,OAAO,QAAQ,CAAA;AACjB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,0BAA0B,CAAC,EAAsB;IAC/D,kBAAkB,GAAG,EAAE,CAAA;AACzB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,yBAAyB;IACvC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CACb,yEAAyE;YACvE,0EAA0E;YAC1E,sEAAsE;YACtE,kBAAkB,CACrB,CAAA;IACH,CAAC;IACD,OAAO,kBAAkB,CAAA;AAC3B,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,8BAA8B,CAAC,EAA0B;IACvE,aAAa,GAAG,EAAE,CAAA;AACpB,CAAC;AAED;uBACuB;AACvB,MAAM,UAAU,6BAA6B;IAC3C,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CACb,0DAA0D;YACxD,2EAA2E;YAC3E,wEAAwE,CAC3E,CAAA;IACH,CAAC;IACD,OAAO,aAAa,CAAA;AACtB,CAAC;AAED,qFAAqF;AACrF,MAAM,UAAU,sBAAsB;IACpC,QAAQ,GAAG,IAAI,CAAA;IACf,QAAQ,GAAG,IAAI,CAAA;IACf,kBAAkB,GAAG,IAAI,CAAA;IACzB,aAAa,GAAG,IAAI,CAAA;AACtB,CAAC"}
+{"version":3,"file":"registry-bridge.js","sourceRoot":"","sources":["../src/registry-bridge.ts"],"names":[],"mappings":"AAAA,8DAA8D;AAC9D,EAAE;AACF,iFAAiF;AACjF,4EAA4E;AAC5E,4EAA4E;AAC5E,+EAA+E;AAC/E,2EAA2E;AAC3E,0BAA0B;AAC1B,EAAE;AACF,8EAA8E;AAC9E,4EAA4E;AAC5E,+EAA+E;AAsC/E,IAAI,QAAQ,GAA8B,IAAI,CAAA;AAC9C,IAAI,QAAQ,GAA6B,IAAI,CAAA;AAC7C,IAAI,kBAAkB,GAA8B,IAAI,CAAA;AACxD,IAAI,aAAa,GAAkC,IAAI,CAAA;AACvD,IAAI,iBAAiB,GAA6B,IAAI,CAAA;AAEtD;;;;;GAKG;AACH,MAAM,UAAU,uBAAuB,CAAC,EAAoB;IAC1D,QAAQ,GAAG,EAAE,CAAA;AACf,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,cAAc;IAC5B,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CACb,wEAAwE;YACtE,uEAAuE;YACvE,8DAA8D,CACjE,CAAA;IACH,CAAC;IACD,OAAO,QAAQ,EAAE,CAAA;AACnB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,yBAAyB,CAAC,EAAqB;IAC7D,QAAQ,GAAG,EAAE,CAAA;AACf,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,wBAAwB;IACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CACb,0EAA0E;YACxE,2EAA2E;YAC3E,2DAA2D,CAC9D,CAAA;IACH,CAAC;IACD,OAAO,QAAQ,CAAA;AACjB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,0BAA0B,CAAC,EAAsB;IAC/D,kBAAkB,GAAG,EAAE,CAAA;AACzB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,yBAAyB;IACvC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CACb,yEAAyE;YACvE,0EAA0E;YAC1E,sEAAsE;YACtE,kBAAkB,CACrB,CAAA;IACH,CAAC;IACD,OAAO,kBAAkB,CAAA;AAC3B,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,8BAA8B,CAAC,EAA0B;IACvE,aAAa,GAAG,EAAE,CAAA;AACpB,CAAC;AAED;uBACuB;AACvB,MAAM,UAAU,6BAA6B;IAC3C,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CACb,0DAA0D;YACxD,2EAA2E;YAC3E,wEAAwE,CAC3E,CAAA;IACH,CAAC;IACD,OAAO,aAAa,CAAA;AACtB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,yBAAyB,CAAC,EAAqB;IAC7D,iBAAiB,GAAG,EAAE,CAAA;AACxB,CAAC;AAED;;sEAEsE;AACtE,MAAM,UAAU,wBAAwB;IACtC,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CACb,2EAA2E;YACzE,2EAA2E;YAC3E,2DAA2D,CAC9D,CAAA;IACH,CAAC;IACD,OAAO,iBAAiB,CAAA;AAC1B,CAAC;AAED,qFAAqF;AACrF,MAAM,UAAU,sBAAsB;IACpC,QAAQ,GAAG,IAAI,CAAA;IACf,QAAQ,GAAG,IAAI,CAAA;IACf,kBAAkB,GAAG,IAAI,CAAA;IACzB,aAAa,GAAG,IAAI,CAAA;IACpB,iBAAiB,GAAG,IAAI,CAAA;AAC1B,CAAC"}

package/dist/sticky.d.ts

@@ -0,0 +1,22 @@
+import type { MiddlewareHandler } from '@rudderjs/contracts';
+/** Run `fn` inside a fresh database context — the unit sticky reads scope to.
+ *  The middleware wraps each request in one; wrap queue-job/command bodies
+ *  manually if they need read-your-writes on a sticky connection. */
+export declare function runWithDatabaseContext<T>(fn: () => T): T;
+/** Whether a database context (request scope) is active. */
+export declare function hasDatabaseContext(): boolean;
+/** Record that `connection` performed a write in the current context.
+ *  No-op outside a context. Called by split adapters' write paths. */
+export declare function markWrote(connection: string): void;
+/** Whether `connection` wrote in the current context — when true (and the
+ *  connection is `sticky`), reads route to the write connection. Always
+ *  `false` outside a context. */
+export declare function stickyWrote(connection: string): boolean;
+/**
+ * Request middleware entering a database context, so sticky reads scope to
+ * the request. Auto-installed on the `web` + `api` groups by the native
+ * provider when any configured connection sets `sticky: true` with a `read`
+ * pool; harmless to install twice (an inner scope simply shadows the outer).
+ */
+export declare function databaseContextMiddleware(): MiddlewareHandler;
+//# sourceMappingURL=sticky.d.ts.map

package/dist/sticky.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sticky.d.ts","sourceRoot":"","sources":["../src/sticky.ts"],"names":[],"mappings":"AAqBA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAA;AAa5D;;qEAEqE;AACrE,wBAAgB,sBAAsB,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,CAExD;AAED,4DAA4D;AAC5D,wBAAgB,kBAAkB,IAAI,OAAO,CAE5C;AAED;sEACsE;AACtE,wBAAgB,SAAS,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI,CAElD;AAED;;iCAEiC;AACjC,wBAAgB,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAEvD;AAED;;;;;GAKG;AACH,wBAAgB,yBAAyB,IAAI,iBAAiB,CAE7D"}

package/dist/sticky.js

@@ -0,0 +1,61 @@
+// ─── Sticky-read scope for read/write-split connections ────
+//
+// Laravel parity: with `sticky: true` on a split connection, a read issued
+// AFTER a write *within the same request cycle* routes to the write
+// connection — the just-written row may not have replicated yet. The "request
+// cycle" is an AsyncLocalStorage scope entered per request by
+// {@link databaseContextMiddleware} (the native provider appends it to both
+// the `web` and `api` groups when a sticky split connection is configured).
+//
+// Outside a scope (queue jobs, rudder commands, scripts) the flag is a no-op
+// and reads go to the replicas. That divergence is deliberate: Laravel's
+// per-connection `$recordsModified` flag resets per request because PHP's
+// process dies with the request — a long-lived Node process would otherwise
+// go sticky-forever after its first write.
+//
+// Node-only module (top-level `node:async_hooks` import) — exported via the
+// `@rudderjs/database/sticky` subpath (canonical home since the Phase-2 engine
+// relocation; `@rudderjs/orm/sticky` stays as a re-export shim). NEVER
+// re-export it from a client-bundle-reachable entry.
+import { AsyncLocalStorage } from 'node:async_hooks';
+// Shared via globalThis for the same dual-load reason as the ORM registry:
+// adapter packages, the `@rudderjs/orm/sticky` shim, and a bundled copy of
+// this module must all see ONE scope. The key keeps its historical `orm` name
+// on purpose — renaming it would split the ALS across an upgrade re-boot.
+const STICKY_KEY = '__rudderjs_orm_sticky__';
+const _g = globalThis;
+if (!_g[STICKY_KEY]) {
+    _g[STICKY_KEY] = new AsyncLocalStorage();
+}
+const _als = _g[STICKY_KEY];
+/** Run `fn` inside a fresh database context — the unit sticky reads scope to.
+ *  The middleware wraps each request in one; wrap queue-job/command bodies
+ *  manually if they need read-your-writes on a sticky connection. */
+export function runWithDatabaseContext(fn) {
+    return _als.run(new Set(), fn);
+}
+/** Whether a database context (request scope) is active. */
+export function hasDatabaseContext() {
+    return _als.getStore() !== undefined;
+}
+/** Record that `connection` performed a write in the current context.
+ *  No-op outside a context. Called by split adapters' write paths. */
+export function markWrote(connection) {
+    _als.getStore()?.add(connection);
+}
+/** Whether `connection` wrote in the current context — when true (and the
+ *  connection is `sticky`), reads route to the write connection. Always
+ *  `false` outside a context. */
+export function stickyWrote(connection) {
+    return _als.getStore()?.has(connection) ?? false;
+}
+/**
+ * Request middleware entering a database context, so sticky reads scope to
+ * the request. Auto-installed on the `web` + `api` groups by the native
+ * provider when any configured connection sets `sticky: true` with a `read`
+ * pool; harmless to install twice (an inner scope simply shadows the outer).
+ */
+export function databaseContextMiddleware() {
+    return (_req, _res, next) => runWithDatabaseContext(() => next());
+}
+//# sourceMappingURL=sticky.js.map

package/dist/sticky.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sticky.js","sourceRoot":"","sources":["../src/sticky.ts"],"names":[],"mappings":"AAAA,8DAA8D;AAC9D,EAAE;AACF,2EAA2E;AAC3E,oEAAoE;AACpE,8EAA8E;AAC9E,8DAA8D;AAC9D,4EAA4E;AAC5E,4EAA4E;AAC5E,EAAE;AACF,6EAA6E;AAC7E,yEAAyE;AACzE,0EAA0E;AAC1E,4EAA4E;AAC5E,2CAA2C;AAC3C,EAAE;AACF,4EAA4E;AAC5E,+EAA+E;AAC/E,uEAAuE;AACvE,qDAAqD;AAErD,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAA;AAGpD,2EAA2E;AAC3E,2EAA2E;AAC3E,8EAA8E;AAC9E,0EAA0E;AAC1E,MAAM,UAAU,GAAG,yBAAyB,CAAA;AAC5C,MAAM,EAAE,GAAG,UAAqC,CAAA;AAChD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC;IACpB,EAAE,CAAC,UAAU,CAAC,GAAG,IAAI,iBAAiB,EAAe,CAAA;AACvD,CAAC;AACD,MAAM,IAAI,GAAG,EAAE,CAAC,UAAU,CAAmC,CAAA;AAE7D;;qEAEqE;AACrE,MAAM,UAAU,sBAAsB,CAAI,EAAW;IACnD,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,EAAE,EAAE,CAAC,CAAA;AAChC,CAAC;AAED,4DAA4D;AAC5D,MAAM,UAAU,kBAAkB;IAChC,OAAO,IAAI,CAAC,QAAQ,EAAE,KAAK,SAAS,CAAA;AACtC,CAAC;AAED;sEACsE;AACtE,MAAM,UAAU,SAAS,CAAC,UAAkB;IAC1C,IAAI,CAAC,QAAQ,EAAE,EAAE,GAAG,CAAC,UAAU,CAAC,CAAA;AAClC,CAAC;AAED;;iCAEiC;AACjC,MAAM,UAAU,WAAW,CAAC,UAAkB;IAC5C,OAAO,IAAI,CAAC,QAAQ,EAAE,EAAE,GAAG,CAAC,UAAU,CAAC,IAAI,KAAK,CAAA;AAClD,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,yBAAyB;IACvC,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,sBAAsB,CAAC,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,CAAA;AACnE,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rudderjs/database",
-  "version": "1.1.0",
+  "version": "1.2.1",
   "license": "MIT",
   "repository": {
     "type": "git",
@@ -21,13 +21,43 @@
     ".": {
       "import": "./dist/index.js",
       "types": "./dist/index.d.ts"
+    },
+    "./sticky": {
+      "node": "./dist/sticky.js",
+      "import": "./dist/sticky.js",
+      "default": "./dist/sticky.js",
+      "types": "./dist/sticky.d.ts"
+    },
+    "./native": {
+      "node": "./dist/native/index.js",
+      "import": "./dist/native/index.js",
+      "default": "./dist/native/index.js",
+      "types": "./dist/native/index.d.ts"
     }
   },
   "dependencies": {
-    "@rudderjs/contracts": "^1.10.0"
+    "@rudderjs/contracts": "^1.12.0"
+  },
+  "peerDependencies": {
+    "better-sqlite3": "^12.0.0",
+    "postgres": "^3.0.0",
+    "mysql2": "^3.0.0"
+  },
+  "peerDependenciesMeta": {
+    "better-sqlite3": {
+      "optional": true
+    },
+    "postgres": {
+      "optional": true
+    },
+    "mysql2": {
+      "optional": true
+    }
   },
   "devDependencies": {
+    "@types/better-sqlite3": "^7.0.0",
     "@types/node": "^20.0.0",
+    "better-sqlite3": "^12.0.0",
     "typescript": "^5.4.0",
     "tsx": "^4.0.0"
   },