pqb 0.40.7 → 0.40.8

package/dist/index.mjs

@@ -3395,6 +3395,32 @@ class Transaction {
       }
     }
   }
+  /**
+   * Use the `$ensureTransaction` when you want to ensure the sequence of queries is running in a transaction, but there is no need for Postgres [savepoints](https://www.postgresql.org/docs/current/sql-savepoint.html).
+   *
+   * ```ts
+   * async function updateUserBalance(userId: string, amount: number) {
+   *   await db.$ensureTransaction(async () => {
+   *     await db.transfer.create({ userId, amount })
+   *     await db.user.find(userId).increment({ balance: amount })
+   *   })
+   * }
+   *
+   * async function saveDeposit(userId: string, deposit: { ... }) {
+   *   await db.$ensureTransaction(async () => {
+   *     await db.deposit.create(deposit)
+   *     // transaction in updateUserBalance won't be started
+   *     await updateUserBalance(userId, deposit.amount)
+   *   })
+   * }
+   * ```
+   */
+  ensureTransaction(cb) {
+    const trx = this.internal.transactionStorage.getStore();
+    if (trx)
+      return cb();
+    return Transaction.prototype.transaction.call(this, cb);
+  }
 }
 const runAfterCommit = async (afterCommit, result) => {
   if (afterCommit) {