@nest-omni/core 4.1.3-10 → 4.1.3-12

package/transaction/transaction-manager.d.ts

@@ -0,0 +1,230 @@
+import { EntityManager, DataSource, QueryRunner } from 'typeorm';
+import { TransactionContextService } from './transaction-context.service';
+/**
+ * 事务传播行为
+ */
+export declare enum Propagation {
+    /**
+     * 支持当前事务，如果不存在就创建一个新事务
+     */
+    REQUIRED = 0,
+    /**
+     * 支持当前事务，如果不存在就以非事务方式执行
+     */
+    SUPPORTS = 1,
+    /**
+     * 支持当前事务，如果不存在就抛出异常
+     */
+    MANDATORY = 2,
+    /**
+     * 创建新事务，如果当前存在事务就挂起它
+     */
+    REQUIRES_NEW = 3,
+    /**
+     * 以非事务方式执行，如果当前存在事务就挂起它
+     */
+    NOT_SUPPORTED = 4,
+    /**
+     * 以非事务方式执行，如果当前存在事务就抛出异常
+     */
+    NEVER = 5,
+    /**
+     * 如果当前存在事务，就在嵌套事务中执行
+     */
+    NESTED = 6
+}
+/**
+ * 事务隔离级别
+ */
+export declare enum IsolationLevel {
+    DEFAULT = -1,
+    READ_UNCOMMITTED = 1,
+    READ_COMMITTED = 2,
+    REPEATABLE_READ = 3,
+    SERIALIZABLE = 4
+}
+/**
+ * 事务超时处理回调
+ */
+export interface TransactionTimeoutCallback {
+    (status: TransactionStatus): Promise<void> | void;
+}
+/**
+ * 事务定义
+ */
+export interface TransactionDefinition {
+    propagation: Propagation;
+    isolationLevel?: IsolationLevel;
+    readOnly?: boolean;
+    timeout?: number;
+    name?: string;
+    dataSourceName?: string;
+    onTimeout?: TransactionTimeoutCallback;
+}
+/**
+ * 事务状态
+ */
+export interface TransactionStatus {
+    isCompleted: boolean;
+    isRollbackOnly: boolean;
+    isNewTransaction: boolean;
+    dataSourceName: string;
+    startTime: Date;
+    definition: TransactionDefinition;
+    nestingLevel: number;
+    timeoutTimer?: NodeJS.Timeout;
+}
+/**
+ * 事务资源
+ */
+interface TransactionResource {
+    dataSourceName: string;
+    dataSource: DataSource;
+    queryRunner: QueryRunner;
+    transactionStarted: boolean;
+    nestingLevel: number;
+    savepoints: string[];
+}
+/**
+ * 挂起的事务资源
+ */
+interface SuspendedTransaction {
+    dataSourceName: string;
+    resource: TransactionResource;
+    status: TransactionStatus;
+}
+/**
+ * 统一的事务上下文管理类
+ * 减少状态管理复杂度，避免不一致
+ */
+export declare class TransactionContextManager {
+    private readonly resources;
+    private readonly activeTransactions;
+    private readonly suspendedTransactions;
+    getResource(name: string): TransactionResource | undefined;
+    getActiveTransaction(name: string): TransactionStatus | null;
+    getAllActiveTransactions(): Map<string, TransactionStatus>;
+    getAllResources(): Map<string, TransactionResource>;
+    setResource(name: string, resource: TransactionResource): void;
+    setActiveTransaction(name: string, status: TransactionStatus): void;
+    deleteResource(name: string): boolean;
+    deleteActiveTransaction(name: string): boolean;
+    suspendTransaction(name: string): SuspendedTransaction | null;
+    resumeSuspendedTransaction(name: string): SuspendedTransaction | null;
+    clear(): void;
+    getStats(): {
+        activeTransactions: {
+            dataSource: string;
+            nestingLevel: number;
+            startTime: Date;
+            propagation: Propagation;
+            isRollbackOnly: boolean;
+        }[];
+        resources: {
+            dataSource: string;
+            hasTransaction: boolean;
+            nestingLevel: number;
+        }[];
+        suspendedTransactions: number;
+    };
+}
+/**
+ * 事务管理器
+ * 统一管理多数据源事务，支持自动嵌套
+ */
+export declare class TransactionManager {
+    private readonly transactionContext;
+    private readonly contextManager;
+    private readonly config?;
+    private readonly logger;
+    constructor(transactionContext: TransactionContextService, contextManager: TransactionContextManager, config?: {
+        defaultTimeout?: number;
+        defaultReadOnly?: boolean;
+        logTransactionDetails?: boolean;
+    });
+    /**
+     * 获取或创建事务
+     * 处理各种传播行为
+     */
+    getTransaction(definition: TransactionDefinition): Promise<TransactionStatus>;
+    /**
+     * 提交事务
+     */
+    commit(status: TransactionStatus): Promise<void>;
+    /**
+     * 回滚事务
+     */
+    rollback(status: TransactionStatus): Promise<void>;
+    /**
+     * 获取当前事务的 EntityManager
+     */
+    getCurrentEntityManager(dataSourceName?: string): EntityManager | null;
+    /**
+     * 获取当前事务状态
+     */
+    getCurrentTransaction(dataSourceName?: string): TransactionStatus | null;
+    /**
+     * 获取事务统计
+     */
+    getTransactionStats(): {
+        activeTransactions: {
+            dataSource: string;
+            nestingLevel: number;
+            startTime: Date;
+            propagation: Propagation;
+            isRollbackOnly: boolean;
+        }[];
+        resources: {
+            dataSource: string;
+            hasTransaction: boolean;
+            nestingLevel: number;
+        }[];
+        suspendedTransactions: number;
+    };
+    /**
+     * 根据 Entity 获取数据源名称
+     * 支持装饰器、静态方法和动态方法
+     */
+    getDataSourceForEntity(entityClassOrInstance: Function | Object): string | null;
+    /**
+     * 为 Entity 获取或创建 EntityManager
+     * 自动检测 Entity 的数据源
+     */
+    getEntityManagerForEntity(entityClassOrInstance: any, fallbackDataSourceName?: string): EntityManager | null;
+    private handleRequired;
+    private handleRequiresNew;
+    private handleNested;
+    private handleSupports;
+    private handleNotSupported;
+    private handleMandatory;
+    private handleNever;
+    private createNewTransaction;
+    private createParticipatingTransaction;
+    private getActiveTransaction;
+    private getDataSource;
+    private convertIsolationLevel;
+    private markCompleted;
+    private removeActiveTransaction;
+    private markParentForRollback;
+    private rollbackOnCommitException;
+    private cleanupResource;
+    /**
+     * 恢复挂起的事务 - 已移到 TransactionContextManager 中
+     * 此方法保留以兼容
+     */
+    private resumeSuspendedTransaction;
+    /**
+     * 获取事务超时时间（毫秒）
+     */
+    private getTimeoutMs;
+    /**
+     * 处理事务超时
+     */
+    private handleTransactionTimeout;
+    /**
+     * 清理事务定时器
+     * 确保定时器被正确清理,避免内存泄漏
+     */
+    private clearTimeout;
+}
+export {};