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

package/transaction/data-source.util.d.ts

@@ -0,0 +1,142 @@
+import { DataSource } from 'typeorm';
+/**
+ * 数据源工具类
+ * 提供静态方法在任意位置获取数据源
+ */
+export declare class DataSourceUtil {
+    /**
+     * 获取数据源
+     * 按优先级从以下位置查找：
+     * 1. 当前请求上下文 (TransactionContextService)
+     * 2. 全局数据源存储 (GlobalDataSourceStorage)
+     *
+     * @param name 数据源名称，默认为 'default'
+     * @returns DataSource 实例
+     * @throws Error 如果数据源不存在
+     */
+    static getDataSource(name?: string): DataSource;
+    /**
+     * 安全获取数据源，不存在时返回 null
+     *
+     * @param name 数据源名称
+     * @returns DataSource 实例或 null
+     */
+    static getDataSourceSafe(name?: string): DataSource | null;
+    /**
+     * 获取默认数据源
+     */
+    static getDefaultDataSource(): DataSource;
+    /**
+     * 安全获取默认数据源
+     */
+    static getDefaultDataSourceSafe(): DataSource | null;
+    /**
+     * 检查数据源是否可用
+     *
+     * @param name 数据源名称
+     * @returns 是否可用
+     */
+    static isAvailable(name?: string): boolean;
+    /**
+     * 获取所有可用数据源名称
+     */
+    static getAvailableDataSourceNames(): string[];
+    /**
+     * 批量获取数据源
+     *
+     * @param names 数据源名称数组
+     * @returns 数据源 Map
+     */
+    static getDataSources(names: string[]): Map<string, DataSource>;
+    /**
+     * 获取所有已注册的数据源
+     */
+    static getAllDataSources(): Map<string, DataSource>;
+    /**
+     * 手动注册数据源（用于非模块场景）
+     *
+     * @param name 数据源名称
+     * @param dataSource DataSource 实例
+     * @param isDefault 是否设为默认数据源
+     */
+    static registerDataSource(name: string, dataSource: DataSource, isDefault?: boolean): void;
+    /**
+     * 批量注册数据源
+     *
+     * @param dataSources 数据源 Map
+     * @param defaultName 默认数据源名称
+     */
+    static registerDataSources(dataSources: Map<string, DataSource>, defaultName?: string): void;
+    /**
+     * 使用数据源执行操作
+     *
+     * @param name 数据源名称
+     * @param operation 要执行的操作
+     * @returns 操作结果
+     */
+    static withDataSource<T>(name: string, operation: (dataSource: DataSource) => Promise<T> | T): Promise<T>;
+    /**
+     * 使用默认数据源执行操作
+     *
+     * @param operation 要执行的操作
+     * @returns 操作结果
+     */
+    static withDefaultDataSource<T>(operation: (dataSource: DataSource) => Promise<T> | T): Promise<T>;
+    /**
+     * 在事务中执行操作
+     *
+     * @param name 数据源名称
+     * @param operation 要执行的操作
+     * @returns 操作结果
+     */
+    static withTransaction<T>(name: string, operation: (dataSource: DataSource) => Promise<T>): Promise<T>;
+    /**
+     * 在默认数据源的事务中执行操作
+     */
+    static withDefaultTransaction<T>(operation: (dataSource: DataSource) => Promise<T>): Promise<T>;
+    /**
+     * 测试数据源连接
+     *
+     * @param name 数据源名称
+     * @returns 连接是否正常
+     */
+    static testConnection(name?: string): Promise<boolean>;
+}
+/**
+ * 便捷的别名
+ */
+export declare const getDataSource: any;
+export declare const getDataSourceSafe: any;
+export declare const getDefaultDataSource: any;
+export declare const isDataSourceAvailable: any;
+export declare const withDataSource: any;
+export declare const withTransaction: any;
+/**
+ * 添加数据源到全局注册表
+ * 用于在 TypeOrmModule.forRootAsync 的 useFactory 中手动注册数据源
+ *
+ * @param name 数据源名称
+ * @param dataSource DataSource 实例
+ * @param isDefault 是否设为默认数据源
+ *
+ * @example
+ * ```typescript
+ * TypeOrmModule.forRootAsync({
+ *   inject: [ApiConfigService],
+ *   useFactory: (config: ApiConfigService) => {
+ *     const dsConfig = config.typeormConfig;
+ *     // 在这里注册，确保数据源初始化后立即可用
+ *     addDataSource('default', dataSource, true);
+ *     return dsConfig;
+ *   },
+ * }),
+ * ```
+ */
+export declare const addDataSource: any;
+/**
+ * 批量添加数据源
+ *
+ * @param dataSources 数据源 Map
+ * @param defaultName 默认数据源名称
+ */
+export declare const addDataSources: any;

package/transaction/data-source.util.js

@@ -0,0 +1,330 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.addDataSources = exports.addDataSource = exports.withTransaction = exports.withDataSource = exports.isDataSourceAvailable = exports.getDefaultDataSource = exports.getDataSourceSafe = exports.getDataSource = exports.DataSourceUtil = void 0;
+const transaction_context_service_1 = require("./transaction-context.service");
+/**
+ * 全局数据源存储
+ * 用于在非请求上下文中存储和获取数据源
+ */
+class GlobalDataSourceStorage {
+    /**
+     * 注册数据源
+     */
+    static register(name, dataSource) {
+        this.dataSources.set(name, dataSource);
+    }
+    /**
+     * 批量注册数据源
+     */
+    static registerAll(dataSources) {
+        for (const [name, dataSource] of dataSources.entries()) {
+            this.register(name, dataSource);
+        }
+    }
+    /**
+     * 获取数据源
+     */
+    static get(name) {
+        return this.dataSources.get(name);
+    }
+    /**
+     * 获取默认数据源
+     */
+    static getDefault() {
+        return this.dataSources.get(this.defaultDataSourceName);
+    }
+    /**
+     * 设置默认数据源名称
+     */
+    static setDefaultName(name) {
+        this.defaultDataSourceName = name;
+    }
+    /**
+     * 检查数据源是否存在
+     */
+    static has(name) {
+        return this.dataSources.has(name);
+    }
+    /**
+     * 获取所有已注册的数据源名称
+     */
+    static getAllNames() {
+        return Array.from(this.dataSources.keys());
+    }
+    /**
+     * 移除数据源
+     */
+    static remove(name) {
+        this.dataSources.delete(name);
+    }
+    /**
+     * 清空所有数据源
+     */
+    static clear() {
+        this.dataSources.clear();
+    }
+}
+GlobalDataSourceStorage.dataSources = new Map();
+GlobalDataSourceStorage.defaultDataSourceName = 'default';
+/**
+ * 数据源工具类
+ * 提供静态方法在任意位置获取数据源
+ */
+class DataSourceUtil {
+    /**
+     * 获取数据源
+     * 按优先级从以下位置查找：
+     * 1. 当前请求上下文 (TransactionContextService)
+     * 2. 全局数据源存储 (GlobalDataSourceStorage)
+     *
+     * @param name 数据源名称，默认为 'default'
+     * @returns DataSource 实例
+     * @throws Error 如果数据源不存在
+     */
+    static getDataSource(name = 'default') {
+        // 1. 尝试从请求上下文获取
+        try {
+            return transaction_context_service_1.TransactionContextService.getDataSource(name);
+        }
+        catch (_a) {
+            // 忽略错误，继续尝试其他方式
+        }
+        // 2. 从全局存储获取
+        const dataSource = GlobalDataSourceStorage.get(name);
+        if (dataSource) {
+            return dataSource;
+        }
+        throw new Error(`DataSource '${name}' not found. ` +
+            `Please ensure it is registered via TransactionModule.registerDataSource() or TypeOrmModule.`);
+    }
+    /**
+     * 安全获取数据源，不存在时返回 null
+     *
+     * @param name 数据源名称
+     * @returns DataSource 实例或 null
+     */
+    static getDataSourceSafe(name = 'default') {
+        try {
+            return this.getDataSource(name);
+        }
+        catch (_a) {
+            return null;
+        }
+    }
+    /**
+     * 获取默认数据源
+     */
+    static getDefaultDataSource() {
+        return this.getDataSource('default');
+    }
+    /**
+     * 安全获取默认数据源
+     */
+    static getDefaultDataSourceSafe() {
+        return this.getDataSourceSafe('default');
+    }
+    /**
+     * 检查数据源是否可用
+     *
+     * @param name 数据源名称
+     * @returns 是否可用
+     */
+    static isAvailable(name = 'default') {
+        var _a;
+        try {
+            const ds = this.getDataSource(name);
+            return (_a = ds === null || ds === void 0 ? void 0 : ds.isInitialized) !== null && _a !== void 0 ? _a : false;
+        }
+        catch (_b) {
+            return false;
+        }
+    }
+    /**
+     * 获取所有可用数据源名称
+     */
+    static getAvailableDataSourceNames() {
+        const names = new Set();
+        // 从请求上下文获取
+        try {
+            const contextNames = transaction_context_service_1.TransactionContextService.getRegisteredDataSources();
+            contextNames.forEach((name) => names.add(name));
+        }
+        catch (_a) {
+            // 忽略错误
+        }
+        // 从全局存储获取
+        GlobalDataSourceStorage.getAllNames().forEach((name) => names.add(name));
+        return Array.from(names);
+    }
+    /**
+     * 批量获取数据源
+     *
+     * @param names 数据源名称数组
+     * @returns 数据源 Map
+     */
+    static getDataSources(names) {
+        const result = new Map();
+        for (const name of names) {
+            try {
+                const ds = this.getDataSource(name);
+                if (ds) {
+                    result.set(name, ds);
+                }
+            }
+            catch (_a) {
+                // 跳过不存在的数据源
+            }
+        }
+        return result;
+    }
+    /**
+     * 获取所有已注册的数据源
+     */
+    static getAllDataSources() {
+        const names = this.getAvailableDataSourceNames();
+        return this.getDataSources(names);
+    }
+    /**
+     * 手动注册数据源（用于非模块场景）
+     *
+     * @param name 数据源名称
+     * @param dataSource DataSource 实例
+     * @param isDefault 是否设为默认数据源
+     */
+    static registerDataSource(name, dataSource, isDefault = false) {
+        GlobalDataSourceStorage.register(name, dataSource);
+        if (isDefault) {
+            GlobalDataSourceStorage.setDefaultName(name);
+        }
+    }
+    /**
+     * 批量注册数据源
+     *
+     * @param dataSources 数据源 Map
+     * @param defaultName 默认数据源名称
+     */
+    static registerDataSources(dataSources, defaultName) {
+        GlobalDataSourceStorage.registerAll(dataSources);
+        if (defaultName) {
+            GlobalDataSourceStorage.setDefaultName(defaultName);
+        }
+    }
+    /**
+     * 使用数据源执行操作
+     *
+     * @param name 数据源名称
+     * @param operation 要执行的操作
+     * @returns 操作结果
+     */
+    static withDataSource(name, operation) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const dataSource = this.getDataSource(name);
+            return operation(dataSource);
+        });
+    }
+    /**
+     * 使用默认数据源执行操作
+     *
+     * @param operation 要执行的操作
+     * @returns 操作结果
+     */
+    static withDefaultDataSource(operation) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this.withDataSource('default', operation);
+        });
+    }
+    /**
+     * 在事务中执行操作
+     *
+     * @param name 数据源名称
+     * @param operation 要执行的操作
+     * @returns 操作结果
+     */
+    static withTransaction(name, operation) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const dataSource = this.getDataSource(name);
+            return dataSource.transaction((entityManager) => __awaiter(this, void 0, void 0, function* () {
+                // 通过 EntityManager 获取 DataSource
+                const ds = entityManager.connection;
+                return operation(ds);
+            }));
+        });
+    }
+    /**
+     * 在默认数据源的事务中执行操作
+     */
+    static withDefaultTransaction(operation) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this.withTransaction('default', operation);
+        });
+    }
+    /**
+     * 测试数据源连接
+     *
+     * @param name 数据源名称
+     * @returns 连接是否正常
+     */
+    static testConnection() {
+        return __awaiter(this, arguments, void 0, function* (name = 'default') {
+            try {
+                const dataSource = this.getDataSource(name);
+                if (dataSource.isInitialized) {
+                    yield dataSource.query('SELECT 1');
+                    return true;
+                }
+                return false;
+            }
+            catch (_a) {
+                return false;
+            }
+        });
+    }
+}
+exports.DataSourceUtil = DataSourceUtil;
+/**
+ * 便捷的别名
+ */
+exports.getDataSource = DataSourceUtil.getDataSource.bind(DataSourceUtil);
+exports.getDataSourceSafe = DataSourceUtil.getDataSourceSafe.bind(DataSourceUtil);
+exports.getDefaultDataSource = DataSourceUtil.getDefaultDataSource.bind(DataSourceUtil);
+exports.isDataSourceAvailable = DataSourceUtil.isAvailable.bind(DataSourceUtil);
+exports.withDataSource = DataSourceUtil.withDataSource.bind(DataSourceUtil);
+exports.withTransaction = DataSourceUtil.withTransaction.bind(DataSourceUtil);
+/**
+ * 添加数据源到全局注册表
+ * 用于在 TypeOrmModule.forRootAsync 的 useFactory 中手动注册数据源
+ *
+ * @param name 数据源名称
+ * @param dataSource DataSource 实例
+ * @param isDefault 是否设为默认数据源
+ *
+ * @example
+ * ```typescript
+ * TypeOrmModule.forRootAsync({
+ *   inject: [ApiConfigService],
+ *   useFactory: (config: ApiConfigService) => {
+ *     const dsConfig = config.typeormConfig;
+ *     // 在这里注册，确保数据源初始化后立即可用
+ *     addDataSource('default', dataSource, true);
+ *     return dsConfig;
+ *   },
+ * }),
+ * ```
+ */
+exports.addDataSource = DataSourceUtil.registerDataSource.bind(DataSourceUtil);
+/**
+ * 批量添加数据源
+ *
+ * @param dataSources 数据源 Map
+ * @param defaultName 默认数据源名称
+ */
+exports.addDataSources = DataSourceUtil.registerDataSources.bind(DataSourceUtil);

package/transaction/index.d.ts

@@ -11,4 +11,5 @@ export { LoggingTransactionalInterceptor } from './logging-transactional.interce
 export { LoggingTransactionalInterceptor as TRANSACTION_LOGGING_INTERCEPTOR } from './logging-transactional.interceptor';
 export { TransactionError, TransactionErrorHandler } from './transaction.errors';
 export { DatabaseAdapter, DatabaseType } from './database-adapter';
+export { DataSourceUtil, getDataSource, getDataSourceSafe, getDefaultDataSource, isDataSourceAvailable, withDataSource, withTransaction, addDataSource, addDataSources, } from './data-source.util';
 export { TransactionModule } from './transaction.module';

package/transaction/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TransactionModule = exports.DatabaseAdapter = exports.TransactionErrorHandler = exports.TransactionError = exports.TRANSACTION_LOGGING_INTERCEPTOR = exports.LoggingTransactionalInterceptor = exports.isDynamicDataSourceEntity = exports.hasEntityDataSource = exports.getEntityDataSource = exports.BaseEntityWithDataSource = exports.DynamicDataSource = exports.DataSource = exports.ClsServiceManager = exports.ClsCompatibilityService = exports.TransactionContextService = exports.DataSourceRegistryService = exports.GlobalMultiTypeOrmModule = exports.TypeOrmModuleWrapper = exports.BaseService = exports.UseTransactional = exports.Tx = exports.TransactionalInterceptor = exports.ClassTransactional = exports.Transactional = exports.TransactionSynchronizationStatus = exports.TransactionSynchronizationManager = exports.TransactionContextManager = exports.IsolationLevel = exports.Propagation = exports.TransactionManager = void 0;
+exports.TransactionModule = exports.addDataSources = exports.addDataSource = exports.withTransaction = exports.withDataSource = exports.isDataSourceAvailable = exports.getDefaultDataSource = exports.getDataSourceSafe = exports.getDataSource = exports.DataSourceUtil = exports.DatabaseAdapter = exports.TransactionErrorHandler = exports.TransactionError = exports.TRANSACTION_LOGGING_INTERCEPTOR = exports.LoggingTransactionalInterceptor = exports.isDynamicDataSourceEntity = exports.hasEntityDataSource = exports.getEntityDataSource = exports.BaseEntityWithDataSource = exports.DynamicDataSource = exports.DataSource = exports.ClsServiceManager = exports.ClsCompatibilityService = exports.TransactionContextService = exports.DataSourceRegistryService = exports.GlobalMultiTypeOrmModule = exports.TypeOrmModuleWrapper = exports.BaseService = exports.UseTransactional = exports.Tx = exports.TransactionalInterceptor = exports.ClassTransactional = exports.Transactional = exports.TransactionSynchronizationStatus = exports.TransactionSynchronizationManager = exports.TransactionContextManager = exports.IsolationLevel = exports.Propagation = exports.TransactionManager = void 0;
 // 核心事务管理
 var transaction_manager_1 = require("./transaction-manager");
 Object.defineProperty(exports, "TransactionManager", { enumerable: true, get: function () { return transaction_manager_1.TransactionManager; } });
@@ -52,6 +52,17 @@ Object.defineProperty(exports, "TransactionErrorHandler", { enumerable: true, ge
 // 数据库适配器
 var database_adapter_1 = require("./database-adapter");
 Object.defineProperty(exports, "DatabaseAdapter", { enumerable: true, get: function () { return database_adapter_1.DatabaseAdapter; } });
+// 数据源工具（可在任意位置获取数据源）
+var data_source_util_1 = require("./data-source.util");
+Object.defineProperty(exports, "DataSourceUtil", { enumerable: true, get: function () { return data_source_util_1.DataSourceUtil; } });
+Object.defineProperty(exports, "getDataSource", { enumerable: true, get: function () { return data_source_util_1.getDataSource; } });
+Object.defineProperty(exports, "getDataSourceSafe", { enumerable: true, get: function () { return data_source_util_1.getDataSourceSafe; } });
+Object.defineProperty(exports, "getDefaultDataSource", { enumerable: true, get: function () { return data_source_util_1.getDefaultDataSource; } });
+Object.defineProperty(exports, "isDataSourceAvailable", { enumerable: true, get: function () { return data_source_util_1.isDataSourceAvailable; } });
+Object.defineProperty(exports, "withDataSource", { enumerable: true, get: function () { return data_source_util_1.withDataSource; } });
+Object.defineProperty(exports, "withTransaction", { enumerable: true, get: function () { return data_source_util_1.withTransaction; } });
+Object.defineProperty(exports, "addDataSource", { enumerable: true, get: function () { return data_source_util_1.addDataSource; } });
+Object.defineProperty(exports, "addDataSources", { enumerable: true, get: function () { return data_source_util_1.addDataSources; } });
 // 模块导出
 var transaction_module_1 = require("./transaction.module");
 Object.defineProperty(exports, "TransactionModule", { enumerable: true, get: function () { return transaction_module_1.TransactionModule; } });

package/validators/is-exists.validator.d.ts

@@ -1,5 +1,5 @@
 import type { ValidationArguments, ValidationOptions, ValidatorConstraintInterface } from 'class-validator';
-import type { EntitySchema, FindOptionsWhere, ObjectType } from 'typeorm';
+import type { DataSource, EntitySchema, FindOptionsWhere, ObjectType } from 'typeorm';
 export declare class IsExistsValidator implements ValidatorConstraintInterface {
     constructor();
     /**
@@ -15,12 +15,29 @@ export declare class IsExistsValidator implements ValidatorConstraintInterface {
      * 例如：appId -> App, userId -> User
      */
     private inferEntityNameFromProperty;
+    /**
+     * 获取数据源
+     * 支持多数据源配置，保持向后兼容
+     */
+    private getDataSource;
     validate<E>(value: string, args: IExistsValidationArguments<E>): Promise<boolean>;
     defaultMessage(args: ValidationArguments): string;
 }
+/**
+ * IsExists 配置选项
+ */
+export interface IsExistsOptions {
+    /**
+     * 数据源名称或 DataSource 实例
+     * 用于多数据源场景，指定使用哪个数据源进行验证
+     * 例如：'readonly', 'report', 或直接传入 DataSource 实例
+     */
+    dataSource?: string | DataSource;
+}
 type ExistsValidationConstraints<E> = [
     (ObjectType<E> | EntitySchema<E> | string | (() => ObjectType<E> | EntitySchema<E> | string)),
-    (validationArguments: ValidationArguments) => FindOptionsWhere<E>
+    (validationArguments: ValidationArguments) => FindOptionsWhere<E>,
+    IsExistsOptions?
 ];
 interface IExistsValidationArguments<E> extends ValidationArguments {
     constraints: ExistsValidationConstraints<E>;

package/validators/is-exists.validator.js

@@ -24,6 +24,7 @@ const class_validator_1 = require("class-validator");
 const nestjs_i18n_1 = require("nestjs-i18n");
 const common_1 = require("@nestjs/common");
 const transaction_1 = require("../transaction");
+const transaction_2 = require("../transaction");
 let IsExistsValidator = class IsExistsValidator {
     constructor() { }
     /**
@@ -84,12 +85,36 @@ let IsExistsValidator = class IsExistsValidator {
         const entityName = name.charAt(0).toUpperCase() + name.slice(1) + 'Entity';
         return entityName;
     }
+    /**
+     * 获取数据源
+     * 支持多数据源配置，保持向后兼容
+     */
+    getDataSource(dataSourceOrName) {
+        // 1. 如果传入的是 DataSource 实例，直接返回
+        if (dataSourceOrName && typeof dataSourceOrName !== 'string') {
+            return dataSourceOrName;
+        }
+        // 2. 如果传入的是数据源名称，获取对应的数据源
+        if (typeof dataSourceOrName === 'string') {
+            return (0, transaction_2.getDataSource)(dataSourceOrName);
+        }
+        // 3. 默认：尝试从请求上下文获取，然后获取 default 数据源
+        try {
+            return transaction_1.TransactionContextService.getDataSource();
+        }
+        catch (_a) {
+            return (0, transaction_2.getDataSource)('default');
+        }
+    }
     validate(value, args) {
         return __awaiter(this, void 0, void 0, function* () {
-            const [entityRef, findCondition] = args.constraints;
+            const [entityRef, findCondition, options] = args.constraints;
             // 解析 Entity 引用
             const entityClass = this.resolveEntity(entityRef, args.property, args.object);
-            const repository = transaction_1.TransactionContextService.getDataSource().getRepository(entityClass);
+            // 获取数据源（支持多数据源）
+            const dataSourceName = (options === null || options === void 0 ? void 0 : options.dataSource) || 'default';
+            const dataSource = this.getDataSource(dataSourceName);
+            const repository = dataSource.getRepository(entityClass);
             args.value = value;
             return ((yield repository.count({
                 where: findCondition(args),

package/validators/is-unique.validator.d.ts

@@ -1,5 +1,5 @@
 import type { ValidationArguments, ValidationOptions, ValidatorConstraintInterface } from 'class-validator';
-import { EntitySchema, FindOptionsWhere, ObjectType } from 'typeorm';
+import type { DataSource, EntitySchema, FindOptionsWhere, ObjectType } from 'typeorm';
 export declare class IsUniqueValidator implements ValidatorConstraintInterface {
     constructor();
     /**
@@ -15,6 +15,11 @@ export declare class IsUniqueValidator implements ValidatorConstraintInterface {
      * 例如：appId -> App, userId -> User
      */
     private inferEntityNameFromProperty;
+    /**
+     * 获取数据源
+     * 支持多数据源配置，保持向后兼容
+     */
+    private getDataSource;
     validate<E>(value: string, args: IUniqueValidationArguments<E>): Promise<boolean>;
     defaultMessage(args: ValidationArguments): string;
 }
@@ -28,6 +33,12 @@ export interface IsUniqueOptions {
      * 例如：uniqueKeys: ['id'] 或 uniqueKeys: ['id', 'version']
      */
     uniqueKeys?: string[];
+    /**
+     * 数据源名称或 DataSource 实例
+     * 用于多数据源场景，指定使用哪个数据源进行验证
+     * 例如：'readonly', 'report', 或直接传入 DataSource 实例
+     */
+    dataSource?: string | DataSource;
 }
 type UniqueValidationConstraints<E> = [
     (ObjectType<E> | EntitySchema<E> | string | (() => ObjectType<E> | EntitySchema<E> | string)),

package/validators/is-unique.validator.js

@@ -24,6 +24,7 @@ const class_validator_1 = require("class-validator");
 const nestjs_i18n_1 = require("nestjs-i18n");
 const common_1 = require("@nestjs/common");
 const transaction_1 = require("../transaction");
+const transaction_2 = require("../transaction");
 let IsUniqueValidator = class IsUniqueValidator {
     constructor() { }
     /**
@@ -80,12 +81,36 @@ let IsUniqueValidator = class IsUniqueValidator {
         const entityName = name.charAt(0).toUpperCase() + name.slice(1) + 'Entity';
         return entityName;
     }
+    /**
+     * 获取数据源
+     * 支持多数据源配置，保持向后兼容
+     */
+    getDataSource(dataSourceOrName) {
+        // 1. 如果传入的是 DataSource 实例，直接返回
+        if (dataSourceOrName && typeof dataSourceOrName !== 'string') {
+            return dataSourceOrName;
+        }
+        // 2. 如果传入的是数据源名称，获取对应的数据源
+        if (typeof dataSourceOrName === 'string') {
+            return (0, transaction_2.getDataSource)(dataSourceOrName);
+        }
+        // 3. 默认：尝试从请求上下文获取，然后获取 default 数据源
+        try {
+            return transaction_1.TransactionContextService.getDataSource();
+        }
+        catch (_a) {
+            return (0, transaction_2.getDataSource)('default');
+        }
+    }
     validate(value, args) {
         return __awaiter(this, void 0, void 0, function* () {
             const [entityRef, findCondition, options] = args.constraints;
             // 解析 Entity 引用
             const entityClass = this.resolveEntity(entityRef, args.property, args.object);
-            const repository = transaction_1.TransactionContextService.getDataSource().getRepository(entityClass);
+            // 获取数据源（支持多数据源）
+            const dataSourceName = (options === null || options === void 0 ? void 0 : options.dataSource) || 'default';
+            const dataSource = this.getDataSource(dataSourceName);
+            const repository = dataSource.getRepository(entityClass);
             args.value = value;
             let exists;
             const defCon = findCondition(args);