@mikro-orm/core 7.0.4 → 7.0.5-dev.0

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,218 +8,216 @@ 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);
+    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);
         }
-        return this.createNewTransaction(em, cb, options);
-      case TransactionPropagation.NESTED:
-        if (hasExistingTransaction) {
-          return this.executeNestedTransaction(em, 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);
         }
-        return this.createNewTransaction(em, cb, options);
-      case TransactionPropagation.SUPPORTS:
-        if (hasExistingTransaction) {
-          return cb(em);
+    }
+    /**
+     * 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);
         }
-        return this.executeWithoutTransaction(em, cb, options);
-      case TransactionPropagation.MANDATORY:
-        if (!hasExistingTransaction) {
-          throw TransactionStateError.requiredTransactionNotFound(propagation);
+        finally {
+            this.resumeTransaction(em, suspended);
         }
-        return cb(em);
-      case TransactionPropagation.NEVER:
+    }
+    /**
+     * 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) {
-          throw TransactionStateError.transactionNotAllowed(propagation);
+            suspended = this.suspendTransaction(em);
+        }
+        const newOptions = { ...options, ctx: undefined };
+        try {
+            return await this.processTransaction(em, fork, cb, newOptions);
+        }
+        finally {
+            if (suspended != null) {
+                this.resumeTransaction(em, suspended);
+            }
         }
-        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);
-    } finally {
-      this.resumeTransaction(em, suspended);
+    /**
+     * Creates new transaction context.
+     */
+    async createNewTransaction(em, cb, options) {
+        const fork = this.createFork(em, options);
+        return this.processTransaction(em, fork, 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
-    if (hasExistingTransaction) {
-      suspended = this.suspendTransaction(em);
+    /**
+     * 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);
     }
-    const newOptions = { ...options, ctx: undefined };
-    try {
-      return await this.processTransaction(em, fork, cb, newOptions);
-    } finally {
-      if (suspended != null) {
-        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 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;
+    /**
+     * Determines if changes should be propagated to the upper context.
+     */
+    shouldPropagateToUpperContext(em) {
+        return !em.global || this.em.config.get('allowGlobalContext');
     }
-    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];
-          }
+    /**
+     * 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;
         }
-      } else {
-        parentUoW.merge(entity, new Set([entity]));
-      }
+        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);
+                }
+            },
+        });
+    }
+    /**
+     * 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 }));
     }
-  }
-  /**
-   * 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);
+    /**
+     * Executes transaction workflow with entity synchronization.
+     */
+    async executeTransactionFlow(fork, cb, propagateToUpperContext, parentEm) {
+        if (!propagateToUpperContext) {
+            const ret = await cb(fork);
+            await fork.flush();
+            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;
     }
-    // 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;
-  }
 }