@kronos-ts/drizzle 0.2.3 → 0.3.0

package/dist/drizzle-transaction-manager.d.ts

@@ -10,6 +10,26 @@ export interface DrizzleDatabaseLike {
  * The Drizzle transaction object — has the same query API as the database.
  */
 export type DrizzleTransaction = any;
+export interface DrizzleTransactionManagerOptions {
+    /**
+     * Runs once on every transaction, right after it opens and before the UoW
+     * gets the handle. This is where a Postgres-backed deployment arms session
+     * GUCs — chiefly `idle_in_transaction_session_timeout` — so a stalled UoW
+     * (e.g. a hung dead-letter drain whose replay never returns) is aborted by
+     * the database instead of pinning a connection indefinitely. The Postgres
+     * adapter does this for its own transactions; this hook is the equivalent
+     * seam for the (DB-agnostic) Drizzle/Knex path. No-op by default.
+     *
+     * Example (postgres-js driver):
+     * ```ts
+     * drizzleTransactionManager(db, {
+     *   onBeginTransaction: (tx) => tx.execute(sql.raw(
+     *     "SET LOCAL idle_in_transaction_session_timeout = 30000")),
+     * })
+     * ```
+     */
+    readonly onBeginTransaction?: (tx: DrizzleTransaction) => Promise<void>;
+}
 /**
  * Creates a TransactionManager for Drizzle ORM.
  *
@@ -32,5 +52,5 @@ export type DrizzleTransaction = any;
  * const txManager = drizzleTransactionManager(db)
  * ```
  */
-export declare function drizzleTransactionManager(db: DrizzleDatabaseLike): TransactionManager<DrizzleTransaction>;
+export declare function drizzleTransactionManager(db: DrizzleDatabaseLike, options?: DrizzleTransactionManagerOptions): TransactionManager<DrizzleTransaction>;
 //# sourceMappingURL=drizzle-transaction-manager.d.ts.map

package/dist/drizzle-transaction-manager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"drizzle-transaction-manager.d.ts","sourceRoot":"","sources":["../src/drizzle-transaction-manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAA;AAE9D;;;GAGG;AACH,MAAM,WAAW,mBAAmB;IAClC,WAAW,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAA;CACxD;AAED;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG,GAAG,CAAA;AAEpC;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAgB,yBAAyB,CACvC,EAAE,EAAE,mBAAmB,GACtB,kBAAkB,CAAC,kBAAkB,CAAC,CA2CxC"}
+{"version":3,"file":"drizzle-transaction-manager.d.ts","sourceRoot":"","sources":["../src/drizzle-transaction-manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAA;AAE9D;;;GAGG;AACH,MAAM,WAAW,mBAAmB;IAClC,WAAW,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAA;CACxD;AAED;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG,GAAG,CAAA;AAEpC,MAAM,WAAW,gCAAgC;IAC/C;;;;;;;;;;;;;;;;OAgBG;IACH,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC,EAAE,EAAE,kBAAkB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAA;CACxE;AAED;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAgB,yBAAyB,CACvC,EAAE,EAAE,mBAAmB,EACvB,OAAO,GAAE,gCAAqC,GAC7C,kBAAkB,CAAC,kBAAkB,CAAC,CA6DxC"}

package/dist/drizzle-transaction-manager.js

@@ -20,23 +20,42 @@
  * const txManager = drizzleTransactionManager(db)
  * ```
  */
-export function drizzleTransactionManager(db) {
+export function drizzleTransactionManager(db, options = {}) {
+    const { onBeginTransaction } = options;
     return {
         async begin() {
             let resolveTx;
+            let rejectTx;
             let resolveCompletion;
             let rejectCompletion;
-            const txReady = new Promise((resolve) => {
+            const txReady = new Promise((resolve, reject) => {
                 resolveTx = resolve;
+                rejectTx = reject;
             });
             const completionSignal = new Promise((resolve, reject) => {
                 resolveCompletion = resolve;
                 rejectCompletion = reject;
             });
             const txPromise = db.transaction(async (tx) => {
+                // Arm session settings (e.g. idle-in-transaction timeout) before the
+                // UoW gets the handle, so begin() only resolves once the tx is bounded.
+                if (onBeginTransaction) {
+                    try {
+                        await onBeginTransaction(tx);
+                    }
+                    catch (err) {
+                        // Surface as a begin() rejection rather than hanging on txReady.
+                        rejectTx(err);
+                        throw err;
+                    }
+                }
                 resolveTx(tx);
                 await completionSignal;
             });
+            // If db.transaction() rejects before onBeginTransaction runs (e.g. the
+            // pool can't hand out a connection), make begin() reject instead of
+            // hanging on txReady forever.
+            txPromise.catch(rejectTx);
             const tx = await txReady;
             tx.__kronos_commit = resolveCompletion;
             tx.__kronos_rollback = rejectCompletion;

package/dist/drizzle-transaction-manager.js.map

@@ -1 +1 @@
-{"version":3,"file":"drizzle-transaction-manager.js","sourceRoot":"","sources":["../src/drizzle-transaction-manager.ts"],"names":[],"mappings":"AAeA;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,UAAU,yBAAyB,CACvC,EAAuB;IAEvB,OAAO;QACL,KAAK,CAAC,KAAK;YACT,IAAI,SAA4C,CAAA;YAChD,IAAI,iBAA8B,CAAA;YAClC,IAAI,gBAA2C,CAAA;YAE/C,MAAM,OAAO,GAAG,IAAI,OAAO,CAAqB,CAAC,OAAO,EAAE,EAAE;gBAC1D,SAAS,GAAG,OAAO,CAAA;YACrB,CAAC,CAAC,CAAA;YAEF,MAAM,gBAAgB,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC7D,iBAAiB,GAAG,OAAO,CAAA;gBAC3B,gBAAgB,GAAG,MAAM,CAAA;YAC3B,CAAC,CAAC,CAAA;YAEF,MAAM,SAAS,GAAG,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE;gBAC5C,SAAS,CAAC,EAAE,CAAC,CAAA;gBACb,MAAM,gBAAgB,CAAA;YACxB,CAAC,CAAC,CAAA;YAEF,MAAM,EAAE,GAAG,MAAM,OAAO,CACvB;YAAC,EAAU,CAAC,eAAe,GAAG,iBAAiB,CAC/C;YAAC,EAAU,CAAC,iBAAiB,GAAG,gBAAgB,CAChD;YAAC,EAAU,CAAC,kBAAkB,GAAG,SAAS,CAAA;YAE3C,OAAO,EAAE,CAAA;QACX,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,EAAsB;YACjC,MAAM,MAAM,GAAI,EAAU,CAAC,eAA6B,CAAA;YACxD,MAAM,SAAS,GAAI,EAAU,CAAC,kBAAmC,CAAA;YACjE,MAAM,EAAE,CAAA;YACR,MAAM,SAAS,CAAA;QACjB,CAAC;QAED,KAAK,CAAC,QAAQ,CAAC,EAAsB;YACnC,MAAM,QAAQ,GAAI,EAAU,CAAC,iBAA6C,CAAA;YAC1E,MAAM,SAAS,GAAI,EAAU,CAAC,kBAAmC,CAAA;YACjE,QAAQ,CAAC,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC,CAAA;YAC9C,IAAI,CAAC;gBAAC,MAAM,SAAS,CAAA;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,cAAc,CAAC,CAAC;QAClD,CAAC;KACF,CAAA;AACH,CAAC"}
+{"version":3,"file":"drizzle-transaction-manager.js","sourceRoot":"","sources":["../src/drizzle-transaction-manager.ts"],"names":[],"mappings":"AAoCA;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,UAAU,yBAAyB,CACvC,EAAuB,EACvB,UAA4C,EAAE;IAE9C,MAAM,EAAE,kBAAkB,EAAE,GAAG,OAAO,CAAA;IACtC,OAAO;QACL,KAAK,CAAC,KAAK;YACT,IAAI,SAA4C,CAAA;YAChD,IAAI,QAAmC,CAAA;YACvC,IAAI,iBAA8B,CAAA;YAClC,IAAI,gBAA2C,CAAA;YAE/C,MAAM,OAAO,GAAG,IAAI,OAAO,CAAqB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAClE,SAAS,GAAG,OAAO,CAAA;gBACnB,QAAQ,GAAG,MAAM,CAAA;YACnB,CAAC,CAAC,CAAA;YAEF,MAAM,gBAAgB,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC7D,iBAAiB,GAAG,OAAO,CAAA;gBAC3B,gBAAgB,GAAG,MAAM,CAAA;YAC3B,CAAC,CAAC,CAAA;YAEF,MAAM,SAAS,GAAG,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE;gBAC5C,qEAAqE;gBACrE,wEAAwE;gBACxE,IAAI,kBAAkB,EAAE,CAAC;oBACvB,IAAI,CAAC;wBACH,MAAM,kBAAkB,CAAC,EAAE,CAAC,CAAA;oBAC9B,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,iEAAiE;wBACjE,QAAQ,CAAC,GAAG,CAAC,CAAA;wBACb,MAAM,GAAG,CAAA;oBACX,CAAC;gBACH,CAAC;gBACD,SAAS,CAAC,EAAE,CAAC,CAAA;gBACb,MAAM,gBAAgB,CAAA;YACxB,CAAC,CAAC,CAAA;YACF,uEAAuE;YACvE,oEAAoE;YACpE,8BAA8B;YAC9B,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;YAEzB,MAAM,EAAE,GAAG,MAAM,OAAO,CACvB;YAAC,EAAU,CAAC,eAAe,GAAG,iBAAiB,CAC/C;YAAC,EAAU,CAAC,iBAAiB,GAAG,gBAAgB,CAChD;YAAC,EAAU,CAAC,kBAAkB,GAAG,SAAS,CAAA;YAE3C,OAAO,EAAE,CAAA;QACX,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,EAAsB;YACjC,MAAM,MAAM,GAAI,EAAU,CAAC,eAA6B,CAAA;YACxD,MAAM,SAAS,GAAI,EAAU,CAAC,kBAAmC,CAAA;YACjE,MAAM,EAAE,CAAA;YACR,MAAM,SAAS,CAAA;QACjB,CAAC;QAED,KAAK,CAAC,QAAQ,CAAC,EAAsB;YACnC,MAAM,QAAQ,GAAI,EAAU,CAAC,iBAA6C,CAAA;YAC1E,MAAM,SAAS,GAAI,EAAU,CAAC,kBAAmC,CAAA;YACjE,QAAQ,CAAC,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC,CAAA;YAC9C,IAAI,CAAC;gBAAC,MAAM,SAAS,CAAA;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,cAAc,CAAC,CAAC;QAClD,CAAC;KACF,CAAA;AACH,CAAC"}

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-export { drizzleTransactionManager, type DrizzleDatabaseLike, type DrizzleTransaction, } from "./drizzle-transaction-manager.js";
+export { drizzleTransactionManager, type DrizzleDatabaseLike, type DrizzleTransaction, type DrizzleTransactionManagerOptions, } from "./drizzle-transaction-manager.js";
 export { drizzleTokenStore, type DrizzleTokenStoreConfig, } from "./drizzle-token-store.js";
 export { drizzleDeadLetterQueue, type DrizzleDeadLetterQueueConfig, } from "./drizzle-dead-letter-queue.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,EACzB,KAAK,mBAAmB,EACxB,KAAK,kBAAkB,GACxB,MAAM,kCAAkC,CAAA;AAEzC,OAAO,EACL,iBAAiB,EACjB,KAAK,uBAAuB,GAC7B,MAAM,0BAA0B,CAAA;AAEjC,OAAO,EACL,sBAAsB,EACtB,KAAK,4BAA4B,GAClC,MAAM,gCAAgC,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,EACzB,KAAK,mBAAmB,EACxB,KAAK,kBAAkB,EACvB,KAAK,gCAAgC,GACtC,MAAM,kCAAkC,CAAA;AAEzC,OAAO,EACL,iBAAiB,EACjB,KAAK,uBAAuB,GAC7B,MAAM,0BAA0B,CAAA;AAEjC,OAAO,EACL,sBAAsB,EACtB,KAAK,4BAA4B,GAClC,MAAM,gCAAgC,CAAA"}

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,GAG1B,MAAM,kCAAkC,CAAA;AAEzC,OAAO,EACL,iBAAiB,GAElB,MAAM,0BAA0B,CAAA;AAEjC,OAAO,EACL,sBAAsB,GAEvB,MAAM,gCAAgC,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,GAI1B,MAAM,kCAAkC,CAAA;AAEzC,OAAO,EACL,iBAAiB,GAElB,MAAM,0BAA0B,CAAA;AAEjC,OAAO,EACL,sBAAsB,GAEvB,MAAM,gCAAgC,CAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kronos-ts/drizzle",
-  "version": "0.2.3",
+  "version": "0.3.0",
   "description": "Drizzle ORM extension for Kronos.",
   "license": "Apache-2.0",
   "type": "module",
@@ -50,7 +50,7 @@
   },
   "dependencies": {
     "@kronos-ts/common": "0.1.1",
-    "@kronos-ts/messaging": "0.6.0"
+    "@kronos-ts/messaging": "0.8.0"
   },
   "peerDependencies": {
     "drizzle-orm": ">=0.30.0"

package/src/drizzle-transaction-manager.ts

@@ -13,6 +13,27 @@ export interface DrizzleDatabaseLike {
  */
 export type DrizzleTransaction = any
 
+export interface DrizzleTransactionManagerOptions {
+  /**
+   * Runs once on every transaction, right after it opens and before the UoW
+   * gets the handle. This is where a Postgres-backed deployment arms session
+   * GUCs — chiefly `idle_in_transaction_session_timeout` — so a stalled UoW
+   * (e.g. a hung dead-letter drain whose replay never returns) is aborted by
+   * the database instead of pinning a connection indefinitely. The Postgres
+   * adapter does this for its own transactions; this hook is the equivalent
+   * seam for the (DB-agnostic) Drizzle/Knex path. No-op by default.
+   *
+   * Example (postgres-js driver):
+   * ```ts
+   * drizzleTransactionManager(db, {
+   *   onBeginTransaction: (tx) => tx.execute(sql.raw(
+   *     "SET LOCAL idle_in_transaction_session_timeout = 30000")),
+   * })
+   * ```
+   */
+  readonly onBeginTransaction?: (tx: DrizzleTransaction) => Promise<void>
+}
+
 /**
  * Creates a TransactionManager for Drizzle ORM.
  *
@@ -37,15 +58,19 @@ export type DrizzleTransaction = any
  */
 export function drizzleTransactionManager(
   db: DrizzleDatabaseLike,
+  options: DrizzleTransactionManagerOptions = {},
 ): TransactionManager<DrizzleTransaction> {
+  const { onBeginTransaction } = options
   return {
     async begin(): Promise<DrizzleTransaction> {
       let resolveTx!: (tx: DrizzleTransaction) => void
+      let rejectTx!: (error: unknown) => void
       let resolveCompletion!: () => void
       let rejectCompletion!: (error: unknown) => void
 
-      const txReady = new Promise<DrizzleTransaction>((resolve) => {
+      const txReady = new Promise<DrizzleTransaction>((resolve, reject) => {
         resolveTx = resolve
+        rejectTx = reject
       })
 
       const completionSignal = new Promise<void>((resolve, reject) => {
@@ -54,9 +79,24 @@ export function drizzleTransactionManager(
       })
 
       const txPromise = db.transaction(async (tx) => {
+        // Arm session settings (e.g. idle-in-transaction timeout) before the
+        // UoW gets the handle, so begin() only resolves once the tx is bounded.
+        if (onBeginTransaction) {
+          try {
+            await onBeginTransaction(tx)
+          } catch (err) {
+            // Surface as a begin() rejection rather than hanging on txReady.
+            rejectTx(err)
+            throw err
+          }
+        }
         resolveTx(tx)
         await completionSignal
       })
+      // If db.transaction() rejects before onBeginTransaction runs (e.g. the
+      // pool can't hand out a connection), make begin() reject instead of
+      // hanging on txReady forever.
+      txPromise.catch(rejectTx)
 
       const tx = await txReady
       ;(tx as any).__kronos_commit = resolveCompletion

package/src/index.ts

@@ -2,6 +2,7 @@ export {
   drizzleTransactionManager,
   type DrizzleDatabaseLike,
   type DrizzleTransaction,
+  type DrizzleTransactionManagerOptions,
 } from "./drizzle-transaction-manager.js"
 
 export {