@mikro-orm/core 7.0.4-dev.9 → 7.0.4

package/utils/TransactionManager.d.ts

@@ -4,62 +4,62 @@ import { type TransactionOptions } from '../enums.js';
  * Manages transaction lifecycle and propagation for EntityManager.
  */
 export declare class TransactionManager {
-    private readonly em;
-    constructor(em: EntityManager);
-    /**
-     * Main entry point for handling transactional operations with propagation support.
-     */
-    handle<T>(cb: (em: EntityManager) => T | Promise<T>, options?: TransactionOptions): Promise<T>;
-    /**
-     * Executes the callback with the specified propagation type.
-     */
-    private executeWithPropagation;
-    /**
-     * Suspends the current transaction and returns the suspended resources.
-     */
-    private suspendTransaction;
-    /**
-     * Resumes a previously suspended transaction.
-     */
-    private resumeTransaction;
-    /**
-     * Executes operation without transaction context.
-     */
-    private executeWithoutTransaction;
-    /**
-     * Creates new independent transaction, suspending any existing one.
-     */
-    private executeWithNewTransaction;
-    /**
-     * Creates new transaction context.
-     */
-    private createNewTransaction;
-    /**
-     * Executes nested transaction with savepoint.
-     */
-    private executeNestedTransaction;
-    /**
-     * Creates a fork of the EntityManager with the given options.
-     */
-    private createFork;
-    /**
-     * Determines if changes should be propagated to the upper context.
-     */
-    private shouldPropagateToUpperContext;
-    /**
-     * Merges entities from fork to parent EntityManager.
-     */
-    private mergeEntitiesToParent;
-    /**
-     * Registers a deletion handler to unset entity identities after flush.
-     */
-    private registerDeletionHandler;
-    /**
-     * Processes transaction execution.
-     */
-    private processTransaction;
-    /**
-     * Executes transaction workflow with entity synchronization.
-     */
-    private executeTransactionFlow;
+  private readonly em;
+  constructor(em: EntityManager);
+  /**
+   * Main entry point for handling transactional operations with propagation support.
+   */
+  handle<T>(cb: (em: EntityManager) => T | Promise<T>, options?: TransactionOptions): Promise<T>;
+  /**
+   * Executes the callback with the specified propagation type.
+   */
+  private executeWithPropagation;
+  /**
+   * Suspends the current transaction and returns the suspended resources.
+   */
+  private suspendTransaction;
+  /**
+   * Resumes a previously suspended transaction.
+   */
+  private resumeTransaction;
+  /**
+   * Executes operation without transaction context.
+   */
+  private executeWithoutTransaction;
+  /**
+   * Creates new independent transaction, suspending any existing one.
+   */
+  private executeWithNewTransaction;
+  /**
+   * Creates new transaction context.
+   */
+  private createNewTransaction;
+  /**
+   * Executes nested transaction with savepoint.
+   */
+  private executeNestedTransaction;
+  /**
+   * Creates a fork of the EntityManager with the given options.
+   */
+  private createFork;
+  /**
+   * Determines if changes should be propagated to the upper context.
+   */
+  private shouldPropagateToUpperContext;
+  /**
+   * Merges entities from fork to parent EntityManager.
+   */
+  private mergeEntitiesToParent;
+  /**
+   * Registers a deletion handler to unset entity identities after flush.
+   */
+  private registerDeletionHandler;
+  /**
+   * Processes transaction execution.
+   */
+  private processTransaction;
+  /**
+   * Executes transaction workflow with entity synchronization.
+   */
+  private executeTransactionFlow;
 }

package/utils/TransactionManager.js

@@ -8,216 +8,218 @@ import { helper } from '../entity/wrap.js';
  * Manages transaction lifecycle and propagation for EntityManager.
  */
 export class TransactionManager {
-    em;
-    constructor(em) {
-        this.em = em;
-    }
-    /**
-     * Main entry point for handling transactional operations with propagation support.
-     */
-    async handle(cb, options = {}) {
-        const em = this.em.getContext(false);
-        options.propagation ??= TransactionPropagation.NESTED;
-        options.ctx ??= em.getTransactionContext();
-        const hasExistingTransaction = !!em.getTransactionContext();
-        return this.executeWithPropagation(options.propagation, em, cb, options, hasExistingTransaction);
-    }
-    /**
-     * Executes the callback with the specified propagation type.
-     */
-    async executeWithPropagation(propagation, em, cb, options, hasExistingTransaction) {
-        switch (propagation) {
-            case TransactionPropagation.NOT_SUPPORTED:
-                return this.executeWithoutTransaction(em, cb, options);
-            case TransactionPropagation.REQUIRES_NEW:
-                return this.executeWithNewTransaction(em, cb, options, hasExistingTransaction);
-            case TransactionPropagation.REQUIRED:
-                if (hasExistingTransaction) {
-                    return cb(em);
-                }
-                return this.createNewTransaction(em, cb, options);
-            case TransactionPropagation.NESTED:
-                if (hasExistingTransaction) {
-                    return this.executeNestedTransaction(em, cb, options);
-                }
-                return this.createNewTransaction(em, cb, options);
-            case TransactionPropagation.SUPPORTS:
-                if (hasExistingTransaction) {
-                    return cb(em);
-                }
-                return this.executeWithoutTransaction(em, cb, options);
-            case TransactionPropagation.MANDATORY:
-                if (!hasExistingTransaction) {
-                    throw TransactionStateError.requiredTransactionNotFound(propagation);
-                }
-                return cb(em);
-            case TransactionPropagation.NEVER:
-                if (hasExistingTransaction) {
-                    throw TransactionStateError.transactionNotAllowed(propagation);
-                }
-                return this.executeWithoutTransaction(em, cb, options);
-            default:
-                throw TransactionStateError.invalidPropagation(propagation);
-        }
-    }
-    /**
-     * Suspends the current transaction and returns the suspended resources.
-     */
-    suspendTransaction(em) {
-        const suspended = em.getTransactionContext();
-        em.resetTransactionContext();
-        return suspended;
-    }
-    /**
-     * Resumes a previously suspended transaction.
-     */
-    resumeTransaction(em, suspended) {
-        if (suspended != null) {
-            em.setTransactionContext(suspended);
-        }
-    }
-    /**
-     * Executes operation without transaction context.
-     */
-    async executeWithoutTransaction(em, cb, options) {
-        const suspended = this.suspendTransaction(em);
-        const fork = this.createFork(em, { ...options, disableTransactions: true });
-        const propagateToUpperContext = this.shouldPropagateToUpperContext(em);
-        try {
-            return await this.executeTransactionFlow(fork, cb, propagateToUpperContext, em);
+  em;
+  constructor(em) {
+    this.em = em;
+  }
+  /**
+   * Main entry point for handling transactional operations with propagation support.
+   */
+  async handle(cb, options = {}) {
+    const em = this.em.getContext(false);
+    options.propagation ??= TransactionPropagation.NESTED;
+    options.ctx ??= em.getTransactionContext();
+    const hasExistingTransaction = !!em.getTransactionContext();
+    return this.executeWithPropagation(options.propagation, em, cb, options, hasExistingTransaction);
+  }
+  /**
+   * Executes the callback with the specified propagation type.
+   */
+  async executeWithPropagation(propagation, em, cb, options, hasExistingTransaction) {
+    switch (propagation) {
+      case TransactionPropagation.NOT_SUPPORTED:
+        return this.executeWithoutTransaction(em, cb, options);
+      case TransactionPropagation.REQUIRES_NEW:
+        return this.executeWithNewTransaction(em, cb, options, hasExistingTransaction);
+      case TransactionPropagation.REQUIRED:
+        if (hasExistingTransaction) {
+          return cb(em);
         }
-        finally {
-            this.resumeTransaction(em, suspended);
+        return this.createNewTransaction(em, cb, options);
+      case TransactionPropagation.NESTED:
+        if (hasExistingTransaction) {
+          return this.executeNestedTransaction(em, cb, options);
         }
-    }
-    /**
-     * Creates new independent transaction, suspending any existing one.
-     */
-    async executeWithNewTransaction(em, cb, options, hasExistingTransaction) {
-        const fork = this.createFork(em, options);
-        let suspended = null;
-        // Suspend existing transaction if present
+        return this.createNewTransaction(em, cb, options);
+      case TransactionPropagation.SUPPORTS:
         if (hasExistingTransaction) {
-            suspended = this.suspendTransaction(em);
+          return cb(em);
         }
-        const newOptions = { ...options, ctx: undefined };
-        try {
-            return await this.processTransaction(em, fork, cb, newOptions);
+        return this.executeWithoutTransaction(em, cb, options);
+      case TransactionPropagation.MANDATORY:
+        if (!hasExistingTransaction) {
+          throw TransactionStateError.requiredTransactionNotFound(propagation);
         }
-        finally {
-            if (suspended != null) {
-                this.resumeTransaction(em, suspended);
-            }
+        return cb(em);
+      case TransactionPropagation.NEVER:
+        if (hasExistingTransaction) {
+          throw TransactionStateError.transactionNotAllowed(propagation);
         }
+        return this.executeWithoutTransaction(em, cb, options);
+      default:
+        throw TransactionStateError.invalidPropagation(propagation);
     }
-    /**
-     * Creates new transaction context.
-     */
-    async createNewTransaction(em, cb, options) {
-        const fork = this.createFork(em, options);
-        return this.processTransaction(em, fork, cb, options);
+  }
+  /**
+   * Suspends the current transaction and returns the suspended resources.
+   */
+  suspendTransaction(em) {
+    const suspended = em.getTransactionContext();
+    em.resetTransactionContext();
+    return suspended;
+  }
+  /**
+   * Resumes a previously suspended transaction.
+   */
+  resumeTransaction(em, suspended) {
+    if (suspended != null) {
+      em.setTransactionContext(suspended);
     }
-    /**
-     * Executes nested transaction with savepoint.
-     */
-    async executeNestedTransaction(em, cb, options) {
-        const fork = this.createFork(em, options);
-        // Pass existing context to create savepoint
-        const nestedOptions = { ...options, ctx: em.getTransactionContext() };
-        return this.processTransaction(em, fork, cb, nestedOptions);
+  }
+  /**
+   * Executes operation without transaction context.
+   */
+  async executeWithoutTransaction(em, cb, options) {
+    const suspended = this.suspendTransaction(em);
+    const fork = this.createFork(em, { ...options, disableTransactions: true });
+    const propagateToUpperContext = this.shouldPropagateToUpperContext(em);
+    try {
+      return await this.executeTransactionFlow(fork, cb, propagateToUpperContext, em);
+    } finally {
+      this.resumeTransaction(em, suspended);
     }
-    /**
-     * Creates a fork of the EntityManager with the given options.
-     */
-    createFork(em, options) {
-        return em.fork({
-            clear: options.clear ?? false,
-            flushMode: options.flushMode,
-            cloneEventManager: true,
-            disableTransactions: options.ignoreNestedTransactions,
-            loggerContext: options.loggerContext,
-        });
+  }
+  /**
+   * Creates new independent transaction, suspending any existing one.
+   */
+  async executeWithNewTransaction(em, cb, options, hasExistingTransaction) {
+    const fork = this.createFork(em, options);
+    let suspended = null;
+    // Suspend existing transaction if present
+    if (hasExistingTransaction) {
+      suspended = this.suspendTransaction(em);
     }
-    /**
-     * Determines if changes should be propagated to the upper context.
-     */
-    shouldPropagateToUpperContext(em) {
-        return !em.global || this.em.config.get('allowGlobalContext');
+    const newOptions = { ...options, ctx: undefined };
+    try {
+      return await this.processTransaction(em, fork, cb, newOptions);
+    } finally {
+      if (suspended != null) {
+        this.resumeTransaction(em, suspended);
+      }
     }
-    /**
-     * Merges entities from fork to parent EntityManager.
-     */
-    mergeEntitiesToParent(fork, parent) {
-        const parentUoW = parent.getUnitOfWork(false);
-        // perf: if parent is empty, we can just move all entities from the fork to skip the `em.merge` overhead
-        if (parentUoW.getIdentityMap().keys().length === 0) {
-            for (const entity of fork.getUnitOfWork(false).getIdentityMap()) {
-                parentUoW.getIdentityMap().store(entity);
-                helper(entity).__em = parent;
-            }
-            return;
-        }
-        for (const entity of fork.getUnitOfWork(false).getIdentityMap()) {
-            const wrapped = helper(entity);
-            const meta = wrapped.__meta;
-            const parentEntity = parentUoW.getById(meta.class, wrapped.getPrimaryKey(), parent.schema, true);
-            if (parentEntity && parentEntity !== entity) {
-                const parentWrapped = helper(parentEntity);
-                parentWrapped.__data = wrapped.__data;
-                parentWrapped.__originalEntityData = wrapped.__originalEntityData;
-                for (const prop of meta.hydrateProps) {
-                    if (prop.kind === ReferenceKind.SCALAR) {
-                        parentEntity[prop.name] = entity[prop.name];
-                    }
-                }
-            }
-            else {
-                parentUoW.merge(entity, new Set([entity]));
-            }
-        }
-    }
-    /**
-     * Registers a deletion handler to unset entity identities after flush.
-     */
-    registerDeletionHandler(fork, parent) {
-        fork.getEventManager().registerSubscriber({
-            afterFlush: (args) => {
-                const deletionChangeSets = args.uow
-                    .getChangeSets()
-                    .filter(cs => cs.type === ChangeSetType.DELETE || cs.type === ChangeSetType.DELETE_EARLY);
-                for (const cs of deletionChangeSets) {
-                    parent.getUnitOfWork(false).unsetIdentity(cs.entity);
-                }
-            },
-        });
+  }
+  /**
+   * Creates new transaction context.
+   */
+  async createNewTransaction(em, cb, options) {
+    const fork = this.createFork(em, options);
+    return this.processTransaction(em, fork, cb, options);
+  }
+  /**
+   * Executes nested transaction with savepoint.
+   */
+  async executeNestedTransaction(em, cb, options) {
+    const fork = this.createFork(em, options);
+    // Pass existing context to create savepoint
+    const nestedOptions = { ...options, ctx: em.getTransactionContext() };
+    return this.processTransaction(em, fork, cb, nestedOptions);
+  }
+  /**
+   * Creates a fork of the EntityManager with the given options.
+   */
+  createFork(em, options) {
+    return em.fork({
+      clear: options.clear ?? false,
+      flushMode: options.flushMode,
+      cloneEventManager: true,
+      disableTransactions: options.ignoreNestedTransactions,
+      loggerContext: options.loggerContext,
+    });
+  }
+  /**
+   * Determines if changes should be propagated to the upper context.
+   */
+  shouldPropagateToUpperContext(em) {
+    return !em.global || this.em.config.get('allowGlobalContext');
+  }
+  /**
+   * Merges entities from fork to parent EntityManager.
+   */
+  mergeEntitiesToParent(fork, parent) {
+    const parentUoW = parent.getUnitOfWork(false);
+    // perf: if parent is empty, we can just move all entities from the fork to skip the `em.merge` overhead
+    if (parentUoW.getIdentityMap().keys().length === 0) {
+      for (const entity of fork.getUnitOfWork(false).getIdentityMap()) {
+        parentUoW.getIdentityMap().store(entity);
+        helper(entity).__em = parent;
+      }
+      return;
     }
-    /**
-     * Processes transaction execution.
-     */
-    async processTransaction(em, fork, cb, options) {
-        const propagateToUpperContext = this.shouldPropagateToUpperContext(em);
-        const eventBroadcaster = new TransactionEventBroadcaster(fork, undefined);
-        return TransactionContext.create(fork, () => fork.getConnection().transactional(async (trx) => {
-            fork.setTransactionContext(trx);
-            return this.executeTransactionFlow(fork, cb, propagateToUpperContext, em);
-        }, { ...options, eventBroadcaster }));
+    for (const entity of fork.getUnitOfWork(false).getIdentityMap()) {
+      const wrapped = helper(entity);
+      const meta = wrapped.__meta;
+      const parentEntity = parentUoW.getById(meta.class, wrapped.getPrimaryKey(), parent.schema, true);
+      if (parentEntity && parentEntity !== entity) {
+        const parentWrapped = helper(parentEntity);
+        parentWrapped.__data = wrapped.__data;
+        parentWrapped.__originalEntityData = wrapped.__originalEntityData;
+        for (const prop of meta.hydrateProps) {
+          if (prop.kind === ReferenceKind.SCALAR) {
+            parentEntity[prop.name] = entity[prop.name];
+          }
+        }
+      } else {
+        parentUoW.merge(entity, new Set([entity]));
+      }
     }
-    /**
-     * Executes transaction workflow with entity synchronization.
-     */
-    async executeTransactionFlow(fork, cb, propagateToUpperContext, parentEm) {
-        if (!propagateToUpperContext) {
-            const ret = await cb(fork);
-            await fork.flush();
-            return ret;
+  }
+  /**
+   * Registers a deletion handler to unset entity identities after flush.
+   */
+  registerDeletionHandler(fork, parent) {
+    fork.getEventManager().registerSubscriber({
+      afterFlush: args => {
+        const deletionChangeSets = args.uow
+          .getChangeSets()
+          .filter(cs => cs.type === ChangeSetType.DELETE || cs.type === ChangeSetType.DELETE_EARLY);
+        for (const cs of deletionChangeSets) {
+          parent.getUnitOfWork(false).unsetIdentity(cs.entity);
         }
-        // Setup: Register deletion handler before execution
-        this.registerDeletionHandler(fork, parentEm);
-        // Execute callback and flush
-        const ret = await cb(fork);
-        await fork.flush();
-        // Synchronization: Merge entities back to the parent
-        this.mergeEntitiesToParent(fork, parentEm);
-        return ret;
+      },
+    });
+  }
+  /**
+   * Processes transaction execution.
+   */
+  async processTransaction(em, fork, cb, options) {
+    const propagateToUpperContext = this.shouldPropagateToUpperContext(em);
+    const eventBroadcaster = new TransactionEventBroadcaster(fork, undefined);
+    return TransactionContext.create(fork, () =>
+      fork.getConnection().transactional(
+        async trx => {
+          fork.setTransactionContext(trx);
+          return this.executeTransactionFlow(fork, cb, propagateToUpperContext, em);
+        },
+        { ...options, eventBroadcaster },
+      ),
+    );
+  }
+  /**
+   * Executes transaction workflow with entity synchronization.
+   */
+  async executeTransactionFlow(fork, cb, propagateToUpperContext, parentEm) {
+    if (!propagateToUpperContext) {
+      const ret = await cb(fork);
+      await fork.flush();
+      return ret;
     }
+    // Setup: Register deletion handler before execution
+    this.registerDeletionHandler(fork, parentEm);
+    // Execute callback and flush
+    const ret = await cb(fork);
+    await fork.flush();
+    // Synchronization: Merge entities back to the parent
+    this.mergeEntitiesToParent(fork, parentEm);
+    return ret;
+  }
 }