@lark-apaas/nestjs-datapaas 1.0.8-alpha.5 → 1.0.8-alpha.7

package/dist/index.cjs

@@ -105,6 +105,15 @@ function genPostgresProxy(sql3, observable, requestContext) {
             return async (onfulfilled, onrejected) => {
               const sqlStr = query.strings instanceof Array ? query.strings[0] : "";
               const span = observable.startTrace(sqlStr, requestContext?.getContext()?.requestRootSpan);
+              const spanContext = {
+                traceId: span.spanContext().traceId,
+                spanId: span.spanContext().spanId
+              };
+              const logContext = {
+                source_type: "platform",
+                paas_attributes_module: "data",
+                paas_parent_span_context: spanContext
+              };
               span.setAttribute("module", "data");
               span.setAttribute("source_type", "platform");
               const start = Date.now();
@@ -120,11 +129,7 @@ function genPostgresProxy(sql3, observable, requestContext) {
                     result_count: result.length ?? 0,
                     error_message: ""
                   });
-                  nestLogger.log(parsedParams, {
-                    source_type: "platform",
-                    paas_attributes_module: "data",
-                    paas_parent_span: span
-                  });
+                  nestLogger.log(parsedParams, logContext);
                 } else {
                   observable.dropTrace(span);
                 }
@@ -140,11 +145,7 @@ function genPostgresProxy(sql3, observable, requestContext) {
                   result_count: 0,
                   error_message: error instanceof Error ? error.message : String(error)
                 });
-                nestLogger.log(parsedParams, {
-                  source_type: "platform",
-                  paas_attributes_module: "data",
-                  paas_parent_span: span
-                });
+                nestLogger.log(parsedParams, logContext);
                 span.end();
                 return onrejected ? onrejected(error) : Promise.reject(error);
               }

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/database/drizzle-manager.ts","../src/database/const.ts","../src/utils.ts","../src/database/postgres-proxy.ts","../src/database/database.module.ts","../src/database/database.service.ts","../src/database/sql-exection-context.middleware.ts","../src/database/database-utils.ts"],"sourcesContent":["// 导出所有公共API\n\nexport * from './database';\nexport * from './types';\n","import { Injectable, OnModuleInit, OnModuleDestroy, Inject } from '@nestjs/common';\nimport { drizzle, type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { sql } from 'drizzle-orm';\nimport postgres from 'postgres';\nimport { DATAPAAS_CONFIG, DATAPAAS_LOGGER_SERVICE } from './const';\nimport { Observable } from '@lark-apaas/nestjs-observable';\nimport { RequestContextService } from '@lark-apaas/nestjs-common';\n\nimport { type DataPaasConfig, type DatabaseConnectionInfo } from '../types';\nimport { parseConnectionInfo } from '../utils';\nimport { genPostgresProxy } from \"./postgres-proxy\";\n@Injectable()\nexport class DrizzleDatabaseManager implements OnModuleInit, OnModuleDestroy {\n  private db: PostgresJsDatabase | null = null;\n  private currentConnectionInfo: DatabaseConnectionInfo | null = null;\n  private isConnected: boolean = false;\n  private skipInitDbConnection: boolean = false;\n  constructor(\n    @Inject(DATAPAAS_CONFIG)\n    private readonly config: DataPaasConfig,\n    @Inject(DATAPAAS_LOGGER_SERVICE)\n    private readonly observable: Observable,\n    private readonly requestContext: RequestContextService,\n  ) { \n    this.skipInitDbConnection = !!process.env.DEPRECATED_SKIP_INIT_DB_CONNECTION;\n  }\n\n  async onModuleInit() {\n    await this.initialize();\n  }\n\n  async onModuleDestroy() {\n    await this.disconnect();\n  }\n\n  /**\n   * 初始化数据库连接\n   * @returns Drizzle 数据库实例\n   */\n  async initialize(): Promise<PostgresJsDatabase> {\n    if (this.skipInitDbConnection) {\n      this.isConnected = true;\n      this.db = {name: 'skipInitDbConnection'} as unknown as PostgresJsDatabase;\n      return this.db;\n    }\n    try {\n      const { connectionString, connectionConfig } = parseConnectionInfo(this.config.connectionString, this.config);\n      const baseSql = postgres(connectionString, connectionConfig);\n      const sqlProxy = genPostgresProxy(baseSql, this.observable, this.requestContext);\n      // 创建 Drizzle 实例\n      this.db = drizzle(sqlProxy, { schema: this.config.schema, logger: false });\n      this.currentConnectionInfo = {\n        connectionString,\n        connectionConfig: {\n          ssl: connectionConfig.ssl,\n          connectionTimeout: connectionConfig.connect_timeout,\n          idleTimeout: connectionConfig.idle_timeout,\n          maxConnections: connectionConfig.max,\n        },\n      };\n      this.isConnected = true;\n      return this.db;\n    } catch (error) {\n      this.isConnected = false;\n      throw new Error(`Failed to initialize database connection: ${error instanceof Error ? error.message : 'Unknown error'}`);\n    }\n  }\n\n  /**\n   * 获取数据库实例\n   * @returns Drizzle 数据库实例\n   */\n  getDatabase(): PostgresJsDatabase {\n    if (!this.db || !this.isConnected) {\n      throw new Error('Database not initialized. Call initialize() first.');\n    }\n    return this.db;\n  }\n\n  /**\n   * 测试数据库连接\n   * @returns 连接是否正常\n   */\n  async testConnection(): Promise<boolean> {\n    if (!this.db) {\n      return false;\n    }\n\n    try {\n      await this.db.execute(sql`SELECT 1`);\n      return true;\n    } catch (error) {\n      console.error('Database connection test failed:', error);\n      this.isConnected = false;\n      return false;\n    }\n  }\n\n  /**\n   * 重新连接数据库\n   * @param forceRefresh 是否强制刷新连接信息\n   * @returns 新的数据库实例\n   */\n  async reconnect(_forceRefresh: boolean = false): Promise<PostgresJsDatabase> {\n    await this.disconnect();\n    return this.initialize();\n  }\n\n  /**\n   * 断开数据库连接\n   */\n  async disconnect(): Promise<void> {\n    if (this.db) {\n      try {\n        // postgres-js 会自动处理连接关闭\n        this.db = null;\n        this.isConnected = false;\n        this.currentConnectionInfo = null;\n      } catch (error) {\n        console.error('Error disconnecting from database:', error);\n      }\n    }\n  }\n\n  /**\n   * 获取当前连接信息\n   * @returns 当前连接信息\n   */\n  getConnectionInfo(): DatabaseConnectionInfo | null {\n    return this.currentConnectionInfo;\n  }\n\n  /**\n   * 检查连接状态\n   * @returns 是否已连接\n   */\n  isDatabaseConnected(): boolean {\n    return this.isConnected;\n  }\n\n\n  /**\n   * 开始事务\n   * @param callback 事务回调函数\n   * @returns 事务结果\n   */\n  async transaction<T>(callback: (tx: PostgresJsDatabase) => Promise<T>): Promise<T> {\n    const db = this.getDatabase();\n    return db.transaction(callback);\n  }\n\n  /**\n   * 获取连接池状态\n   * @returns 连接池状态信息\n   */\n  getConnectionPoolStatus(): { isConnected: boolean; connectionInfo: DatabaseConnectionInfo | null } {\n    return {\n      isConnected: this.isConnected,\n      connectionInfo: this.currentConnectionInfo,\n    };\n  }\n}","export const DRIZZLE_DATABASE = 'DRIZZLE_DATABASE';\nexport const DATAPAAS_CONFIG = 'DATAPAAS_CONFIG';\nexport const DATAPAAS_LOGGER_SERVICE = Symbol('DATAPAAS_LOGGER_SERVICE');","import type postgres from 'postgres';\nimport { type Logger as DrizzleLogger } from 'drizzle-orm';\nimport { PostgresOptions } from './types';\n\ninterface ConnectionConfig {\n  maxConnections?: number;\n  idleTimeout?: number;\n  connectionTimeout?: number;\n  ssl?: PostgresOptions['ssl'];\n  logger?: boolean | DrizzleLogger;\n}\nexport function parseConnectionInfo(connectionString: string, config?: ConnectionConfig) {\n  const { maxConnections, idleTimeout, connectionTimeout, ssl, logger } = config || {};\n  // 创建 postgres 连接\n  let parsedConfigFromUrl: URL;\n  try {\n    parsedConfigFromUrl = new URL(connectionString);\n    const temp = new URL(connectionString);\n    temp.searchParams.delete('schema'); // 移除schema参数，避免 postgres 解析错误\n    connectionString = temp.toString();\n  } catch (error) {\n    throw new Error('Failed to parse connection info');\n  }\n  const { username, password, hostname, port, pathname } = parsedConfigFromUrl;\n  const connectionConfig = {\n    username: decodeURIComponent(username),\n    password: decodeURIComponent(password),\n    host: [hostname] as unknown as string,\n    port: [parseInt(port) || 5432] as unknown as number,\n    pathname: decodeURIComponent(pathname),\n    max: maxConnections || 1,\n    idle_timeout: idleTimeout || 20,\n    connect_timeout: connectionTimeout || 10,\n    ssl,\n    onnotice: logger ? console.log : undefined,\n  }\n  return { connectionString, connectionConfig };\n}","import type postgres from 'postgres';\nimport { Logger } from '@nestjs/common';\nimport { Span, Observable } from '@lark-apaas/nestjs-observable';\nimport { RequestContextService } from '@lark-apaas/nestjs-common';\n\nconst nestLogger = new Logger('Database');\n\n/**\n * 包装 postgres 实例以记录 SQL 和结果\n */\nexport function genPostgresProxy(sql: postgres.Sql, observable: Observable, requestContext: RequestContextService): postgres.Sql {\n  const wrapQuery = <T extends readonly (postgres.Row | undefined)[]>(query: postgres.PendingQuery<T>): postgres.PendingQuery<T> => {\n    if (query && typeof query.then === 'function') {\n      return new Proxy(query, {\n        get(target, prop, receiver) {\n          if (prop === 'then') {\n            return async (onfulfilled?: (value: postgres.RowList<T>) => unknown, onrejected?: (reason: unknown) => unknown) => {\n              const sqlStr = (query as unknown as Record<string, unknown>).strings instanceof Array ? (query as unknown as { strings: string[] }).strings[0] : '';\n              const span = observable.startTrace(sqlStr, requestContext?.getContext()?.requestRootSpan as Span);\n\n              span.setAttribute(\"module\", \"data\");\n              span.setAttribute(\"source_type\", \"platform\");\n\n              const start = Date.now();\n              try {\n                const result = await target;\n                const isSetRoleSql = result?.command === 'SET';\n\n                // rls SET ROLE 不上报日志\n                if (!isSetRoleSql) {\n                  const duration = Date.now() - start;\n\n                  const parsedParams = JSON.stringify({\n                    operation: result?.command,\n                    sql: result?.statement?.string,\n                    duration_ms: duration,\n                    result_count: result.length ?? 0,\n                    error_message: ''\n                  });\n\n                  nestLogger.log(parsedParams, {\n                    source_type: 'platform',\n                    paas_attributes_module: 'data',\n                    paas_parent_span: span,\n                  });\n                } else {\n                  observable.dropTrace(span);\n                }\n\n                span.end();\n                return onfulfilled ? onfulfilled(result) : result;\n              } catch (error) {\n                const duration = Date.now() - start;\n                const sqlStr = (query as unknown as Record<string, unknown>).strings instanceof Array ? (query as unknown as { strings: string[] }).strings[0] : '';\n                const parsedParams = JSON.stringify({\n                  operation: '',\n                  sql: sqlStr,\n                  duration_ms: duration,\n                  result_count: 0,\n                  error_message: error instanceof Error ? error.message : String(error)\n                });\n                nestLogger.log(parsedParams, {\n                  source_type: 'platform',\n                  paas_attributes_module: 'data',\n                  paas_parent_span: span,\n                });\n\n                span.end();\n                return onrejected ? onrejected(error) : Promise.reject(error);\n              }\n            };\n          }\n          const value = Reflect.get(target, prop, receiver);\n          return typeof value === 'function' ? value.bind(receiver) : value;\n        },\n      }) as postgres.PendingQuery<T>;\n    }\n    return query;\n  };\n\n  return new Proxy(sql, {\n    apply(target, thisArg, argArray) {\n      const query = Reflect.apply(target, thisArg, argArray);\n      return wrapQuery(query);\n    },\n    get(target, prop, receiver) {\n      const value = Reflect.get(target, prop, receiver);\n      if (prop === 'begin' && typeof value === 'function') {\n        return <T>(arg: string | ((tx: postgres.TransactionSql) => T | Promise<T>), cb?: (tx: postgres.TransactionSql) => T | Promise<T>) => {\n          if (typeof arg === 'function') {\n            return value.call(target, (tx: postgres.TransactionSql) => arg(genPostgresProxy(tx as unknown as postgres.Sql, observable, requestContext) as unknown as postgres.TransactionSql));\n          }\n          if (typeof cb === 'function') {\n            return value.call(target, arg, (tx: postgres.TransactionSql) => cb(genPostgresProxy(tx as unknown as postgres.Sql, observable, requestContext) as unknown as postgres.TransactionSql));\n          }\n          return value.call(target, arg);\n        };\n      }\n      if (prop === 'unsafe' && typeof value === 'function') {\n        return <T extends readonly (postgres.Row | undefined)[] = postgres.Row[]>(queryStr: string, args?: unknown[], options?: postgres.UnsafeQueryOptions): postgres.PendingQuery<T> => {\n          const query = value.call(target, queryStr, args, options) as postgres.PendingQuery<T>;\n          return wrapQuery(query);\n        };\n      }\n      return typeof value === 'function' ? value.bind(receiver) : value;\n    },\n  }) as postgres.Sql;\n}","import {\n  Module,\n  DynamicModule,\n  Provider,\n  MiddlewareConsumer,\n  InjectionToken,\n  ModuleMetadata,\n} from '@nestjs/common';\nimport { DrizzleDatabaseManager } from './drizzle-manager';\nimport { DataPaasDatabaseService } from './database.service';\nimport { SqlExecutionContextMiddleware } from './sql-exection-context.middleware';\nimport { type DataPaasConfig } from '@/types';\nimport { DATAPAAS_CONFIG, DRIZZLE_DATABASE, DATAPAAS_LOGGER_SERVICE } from './const';\nexport { type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { RequestContextService, OBSERVABLE_SERVICE, ObservableService } from '@lark-apaas/nestjs-common';\nimport { Observable } from '@lark-apaas/nestjs-observable';\n\nexport interface DataPaasModuleOptions extends DataPaasConfig {}\n\n/**\n * 异步配置选项\n */\nexport interface DataPaasModuleAsyncOptions {\n  /**\n   * 需要导入的模块（例如 ConfigModule）\n   */\n  imports?: ModuleMetadata['imports'];\n  /**\n   * 使用工厂函数创建配置\n   */\n  useFactory?: (...args: unknown[]) => Promise<DataPaasConfig> | DataPaasConfig;\n  /**\n   * 工厂函数需要注入的依赖\n   */\n  inject?: InjectionToken[];\n}\n\nfunction verifyDrizzlePatched(): void {\n  try {\n    // 直接加载被 patch 的 CJS 模块并检查导出是否存在\n    // 注意：此路径与补丁文件一致：drizzle-orm/postgres-js/session.cjs\n    // 若未来包结构变化，请同步更新补丁与此校验\n    // eslint-disable-next-line @typescript-eslint/no-var-requires\n    const postgres = require('drizzle-orm/postgres-js');\n    const ok =\n      postgres &&\n      typeof postgres.getAuthContextStore === 'function' &&\n      typeof postgres.runWithAuthContext === 'function';\n    if (!ok) {\n      throw new Error(\n        'drizzle-orm 未检测到补丁导出：getAuthContextStore/runWithAuthContext',\n      );\n    }\n  } catch (err) {\n    const reason = err instanceof Error ? err.message : String(err);\n    throw new Error(`drizzle-orm 补丁未生效或版本不匹配：${reason}`);\n  }\n}\n\n// 在模块初始化阶段进行一次 drizzle-orm 补丁自检\nverifyDrizzlePatched();\n\n@Module({\n  providers: [{\n    provide: DATAPAAS_LOGGER_SERVICE,\n    useFactory: (observable: ObservableService | null, rtx: RequestContextService) => {\n      return observable ?? new Observable(rtx);\n    },\n    inject: [{\n      token: OBSERVABLE_SERVICE,\n      optional: true,\n    }, RequestContextService],\n  }],\n})\nexport class DataPaasModule {\n  configure(_consumer: MiddlewareConsumer) {}\n  /**\n   * 注册数据平台模块（同步方式）\n   * @param options 模块配置选项\n   * @returns 动态模块\n   */\n  static forRoot(options: DataPaasConfig): DynamicModule {\n    let roleSchema = '';\n    let sslModeRequired = false;\n    if (options.connectionString) {\n      try {\n        const url = new URL(options.connectionString);\n\n        roleSchema = url.searchParams.get(\n          'schema'\n        ) as string;\n        sslModeRequired = url.searchParams.get('sslmode') === \"require\";\n      } catch (err) {\n        roleSchema = '';\n      }\n    }\n    const config: DataPaasConfig = {\n      connectionString: options.connectionString,\n      schema: options.schema,\n      logger: options.logger || false,\n      timeout: options.timeout || 10000,\n      maxConnections: options.maxConnections || 1,\n      idleTimeout: options.idleTimeout || 20,\n      connectionTimeout: options.connectionTimeout || 10,\n      ssl: options.ssl || (sslModeRequired ? 'require' : false),\n      autoContext: options.autoContext == null ? true : !!options.autoContext,\n      roleSchema,\n    };\n\n    const providers: Provider[] = [\n      // 配置提供者 - 使用 useValue\n      {\n        provide: DATAPAAS_CONFIG,\n        useValue: config,\n      },\n      // 数据库管理器 - 使用 useClass，依赖注入连接服务和配置服务\n      {\n        provide: DrizzleDatabaseManager,\n        useClass: DrizzleDatabaseManager,\n      },\n      // 数据库服务 - 使用 useClass，依赖注入管理器\n      {\n        provide: DataPaasDatabaseService,\n        useClass: DataPaasDatabaseService,\n      },\n      {\n        provide: SqlExecutionContextMiddleware,\n        useClass: SqlExecutionContextMiddleware,\n      },\n      // Drizzle 数据库实例 - 使用 useFactory 获取管理器中的数据库实例\n      {\n        provide: DRIZZLE_DATABASE,\n        useFactory: async (manager: DrizzleDatabaseManager) => {\n          await manager.initialize();\n          return manager.getDatabase();\n        },\n        inject: [DrizzleDatabaseManager],\n      },\n    ];\n\n    return {\n      module: DataPaasModule,\n      providers,\n      exports: [\n        DrizzleDatabaseManager,\n        DataPaasDatabaseService,\n        SqlExecutionContextMiddleware,\n        DRIZZLE_DATABASE,\n        DATAPAAS_CONFIG,\n      ],\n      global: true,\n    };\n  }\n\n  /**\n   * 注册数据平台模块（异步方式）\n   * 支持从 ConfigModule 等异步源获取配置\n   * @param options 异步配置选项\n   * @returns 动态模块\n   * @example\n   * ```typescript\n   * DataPaasModule.forRootAsync({\n   *   imports: [ConfigModule],\n   *   useFactory: async (configService: ConfigService) => ({\n   *     connectionString: configService.get('DATABASE_URL'),\n   *   }),\n   *   inject: [ConfigService],\n   * })\n   * ```\n   */\n  static forRootAsync(options: DataPaasModuleAsyncOptions): DynamicModule {\n    const asyncProviders = this.createAsyncProviders(options);\n\n    const providers: Provider[] = [\n      ...asyncProviders,\n      // 数据库管理器 - 使用 useClass，依赖注入连接服务和配置服务\n      {\n        provide: DrizzleDatabaseManager,\n        useClass: DrizzleDatabaseManager,\n      },\n      // 数据库服务 - 使用 useClass，依赖注入管理器\n      {\n        provide: DataPaasDatabaseService,\n        useClass: DataPaasDatabaseService,\n      },\n      {\n        provide: SqlExecutionContextMiddleware,\n        useClass: SqlExecutionContextMiddleware,\n      },\n      // Drizzle 数据库实例 - 使用 useFactory 获取管理器中的数据库实例\n      {\n        provide: DRIZZLE_DATABASE,\n        useFactory: async (manager: DrizzleDatabaseManager) => {\n          await manager.initialize();\n          return manager.getDatabase();\n        },\n        inject: [DrizzleDatabaseManager],\n      },\n    ];\n\n    return {\n      module: DataPaasModule,\n      imports: options.imports || [],\n      providers,\n      exports: [\n        DrizzleDatabaseManager,\n        DataPaasDatabaseService,\n        SqlExecutionContextMiddleware,\n        DRIZZLE_DATABASE,\n        DATAPAAS_CONFIG,\n      ],\n      global: true,\n    };\n  }\n\n  /**\n   * 创建异步配置 providers\n   */\n  private static createAsyncProviders(\n    options: DataPaasModuleAsyncOptions\n  ): Provider[] {\n    if (options.useFactory) {\n      return [this.createAsyncConfigProvider(options)];\n    }\n\n    throw new Error('DataPaasModule.forRootAsync 需要指定 useFactory');\n  }\n\n  /**\n   * 创建异步配置 provider\n   */\n  private static createAsyncConfigProvider(\n    options: DataPaasModuleAsyncOptions\n  ): Provider {\n    if (options.useFactory) {\n      return {\n        provide: DATAPAAS_CONFIG,\n        useFactory: async (...args: unknown[]) => {\n          const config = await options.useFactory!(...args);\n          return this.normalizeConfig(config);\n        },\n        inject: options.inject || [],\n      };\n    }\n\n    throw new Error('Invalid async options');\n  }\n\n  /**\n   * 标准化配置，设置默认值\n   */\n  private static normalizeConfig(config: DataPaasConfig): DataPaasConfig {\n    let roleSchema = '';\n    let sslModeRequired = false;\n    if (config.connectionString) {\n      try {\n        const url = new URL(config.connectionString);\n        roleSchema = url.searchParams.get(\n          'schema'\n        ) as string;\n        sslModeRequired = url.searchParams.get('sslmode') === \"require\";\n      } catch (err) {\n        roleSchema = '';\n      }\n    }\n    return {\n      connectionString: config.connectionString,\n      schema: config.schema,\n      logger: config.logger || false,\n      timeout: config.timeout || 10000,\n      maxConnections: config.maxConnections || 1,\n      idleTimeout: config.idleTimeout || 20,\n      connectionTimeout: config.connectionTimeout || 10,\n      ssl: config.ssl || (sslModeRequired ? 'require' : false),\n      autoContext: config.autoContext == null ? true : !!config.autoContext,\n      roleSchema,\n    };\n  }\n}\n","import { Injectable, OnModuleDestroy } from '@nestjs/common';\nimport { type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { DrizzleDatabaseManager } from './drizzle-manager';\nimport { DatabaseConnectionInfo } from '../types';\n\n@Injectable()\nexport class DataPaasDatabaseService implements OnModuleDestroy {\n  constructor(\n    private readonly databaseManager: DrizzleDatabaseManager,\n  ) {}\n\n  /**\n   * 获取 Drizzle 数据库实例\n   * @returns Drizzle 数据库实例\n   */\n  getDatabase(): PostgresJsDatabase {\n    return this.databaseManager.getDatabase();\n  }\n\n  /**\n   * 获取数据库管理器\n   * @returns 数据库管理器实例\n   */\n  getDatabaseManager(): DrizzleDatabaseManager {\n    return this.databaseManager;\n  }\n\n\n  /**\n   * 测试数据库连接\n   * @returns 连接是否正常\n   */\n  async testConnection(): Promise<boolean> {\n    return this.databaseManager.testConnection();\n  }\n\n  /**\n   * 重新连接数据库\n   * @param forceRefresh 是否强制刷新连接信息\n   * @returns 新的数据库实例\n   */\n  async reconnect(forceRefresh: boolean = false): Promise<PostgresJsDatabase> {\n    return this.databaseManager.reconnect(forceRefresh);\n  }\n\n  /**\n   * 获取当前连接信息\n   * @returns 当前连接信息\n   */\n  getConnectionInfo(): DatabaseConnectionInfo | null {\n    return this.databaseManager.getConnectionInfo();\n  }\n\n  /**\n   * 检查连接状态\n   * @returns 是否已连接\n   */\n  isConnected(): boolean {\n    return this.databaseManager.isDatabaseConnected();\n  }\n\n  /**\n   * 开始事务\n   * @param callback 事务回调函数\n   * @returns 事务结果\n   */\n  async transaction<T>(callback: (tx: PostgresJsDatabase) => Promise<T>): Promise<T> {\n    return this.databaseManager.transaction(callback);\n  }\n\n  /**\n   * 获取连接池状态\n   * @returns 连接池状态信息\n   */\n  getConnectionPoolStatus(): { isConnected: boolean; connectionInfo: DatabaseConnectionInfo | null } {\n    return this.databaseManager.getConnectionPoolStatus();\n  }\n\n\n  /**\n   * 模块销毁时断开数据库连接\n   */\n  async onModuleDestroy(): Promise<void> {\n    await this.databaseManager.disconnect();\n  }\n}\n","import { Injectable, NestMiddleware, Inject } from '@nestjs/common';\nimport { Request, Response, NextFunction } from 'express';\nimport * as session from 'drizzle-orm/postgres-js';\nimport { DATAPAAS_CONFIG } from './const';\nimport { type DataPaasConfig } from '../types';\n\ninterface UserContext {\n  userId?: string;\n  tenantId?: number;\n  appId?: string;\n  // 用户角色（如果有）\n  userRoles?: string[];\n  // 是否为系统账号\n  isSystemAccount: boolean;\n  // 用户类型\n  userType?: string;\n}\n\n@Injectable()\nexport class SqlExecutionContextMiddleware implements NestMiddleware {\n  constructor(\n    @Inject(DATAPAAS_CONFIG)\n    private readonly config: DataPaasConfig,\n  ) {}\n\n  use(req: Request, _res: Response, next: NextFunction) {\n    const roleSchema = this.config.roleSchema ?? '';\n    const userContext =\n      (req as unknown as { userContext: UserContext }).userContext || {};\n    const roles = (userContext.userRoles || []).join(',');\n    const userId = userContext.userId;\n    const isSystemAccount = userContext.isSystemAccount;\n    const userType = userContext.userType;\n    const sqls = [\n      // 设置用户 ID, 匿名用户时设置为空字符串\n      userId\n        ? `SET LOCAL app.user_id = '${userId}'`\n        : `SET LOCAL app.user_id = ''`,\n      // 系统用户设置为service_role, 登录用户设置为 authenticated_xxx 角色，匿名用户设置为 anon_xxx 角色\n      isSystemAccount ? `SET LOCAL ROLE 'service_role_${roleSchema}'` :  (\n        userId\n          ? `SET LOCAL ROLE 'authenticated_${roleSchema}'`\n          : `SET LOCAL ROLE 'anon_${roleSchema}'`),\n      // 自定义角色\n      roles && `SET LOCAL app.role_ids = '${roles}'`,\n      // 用户类型\n      userType\n        ? `SET LOCAL app.user_type = '${userType}'`\n        : `SET LOCAL app.user_type = ''`,\n    ]\n      .filter(Boolean)\n      .join(';');\n\n    // 需要确认角色设置是否对后续 SQL 执行有影响，预期是只影响当前会话\n    session.runWithAuthContext({ preSql: sqls }, () => {\n      next();\n    });\n  }\n}\n","import { type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { sql } from 'drizzle-orm';\n\nimport { DataPaasDatabaseService } from './database.service';\n\n/**\n * 数据库工具函数集合\n */\nexport class DatabaseUtils {\n  /**\n   * 安全执行数据库操作\n   * @param databaseService 数据库服务实例\n   * @param operation 数据库操作函数\n   * @param fallback 失败时的回退值\n   * @returns 操作结果或回退值\n   */\n  static async safeExecute<T>(\n    databaseService: DataPaasDatabaseService,\n    operation: (db: PostgresJsDatabase) => Promise<T>,\n    fallback?: T\n  ): Promise<T | undefined> {\n    try {\n      if (!databaseService.isConnected()) {\n        throw new Error('Database not connected');\n      }\n      \n      const db = databaseService.getDatabase();\n      return await operation(db);\n    } catch (error) {\n      console.error('Database operation failed:', error);\n      return fallback;\n    }\n  }\n\n  /**\n   * 批量执行数据库操作\n   * @param databaseService 数据库服务实例\n   * @param operations 操作数组\n   * @param batchSize 批处理大小\n   * @returns 所有操作的结果\n   */\n  static async batchExecute<T>(\n    databaseService: DataPaasDatabaseService,\n    operations: Array<(db: PostgresJsDatabase) => Promise<T>>,\n    batchSize: number = 10\n  ): Promise<T[]> {\n    const results: T[] = [];\n    const db = databaseService.getDatabase();\n\n    for (let i = 0; i < operations.length; i += batchSize) {\n      const batch = operations.slice(i, i + batchSize);\n      const batchResults = await Promise.all(\n        batch.map(operation => operation(db))\n      );\n      results.push(...batchResults);\n    }\n\n    return results;\n  }\n\n  /**\n   * 执行带重试的数据库操作\n   * @param databaseService 数据库服务实例\n   * @param operation 数据库操作函数\n   * @param maxRetries 最大重试次数\n   * @param retryDelay 重试延迟（毫秒）\n   * @returns 操作结果\n   */\n  static async executeWithRetry<T>(\n    databaseService: DataPaasDatabaseService,\n    operation: (db: PostgresJsDatabase) => Promise<T>,\n    maxRetries: number = 3,\n    retryDelay: number = 1000\n  ): Promise<T> {\n    let lastError: Error;\n\n    for (let attempt = 0; attempt <= maxRetries; attempt++) {\n      try {\n        const db = databaseService.getDatabase();\n        return await operation(db);\n      } catch (error) {\n        lastError = error as Error;\n        \n        if (attempt === maxRetries) {\n          break;\n        }\n\n        console.warn(`Database operation failed (attempt ${attempt + 1}/${maxRetries + 1}):`, error);\n        \n        // 等待后重试\n        await new Promise(resolve => setTimeout(resolve, retryDelay * (attempt + 1)));\n      }\n    }\n\n    throw lastError!;\n  }\n\n  /**\n   * 检查表是否存在\n   * @param databaseService 数据库服务实例\n   * @param tableName 表名\n   * @returns 表是否存在\n   */\n  static async tableExists(\n    db: PostgresJsDatabase,\n    tableName: string\n  ): Promise<boolean> {\n    try {\n      const result = await db.execute(\n        sql`SELECT EXISTS (\n          SELECT FROM information_schema.tables \n          WHERE table_schema = 'public' \n          AND table_name = ${tableName}\n        )`\n      );\n      \n      return result[0]?.exists as boolean || false;\n    } catch (error) {\n      console.error('Failed to check table existence:', error);\n      return false;\n    }\n  }\n\n  /**\n   * 获取表结构信息\n   * @param databaseService 数据库服务实例\n   * @param tableName 表名\n   * @returns 表结构信息\n   */\n  static async getTableStructure(\n    db: PostgresJsDatabase,\n    tableName: string\n  ): Promise<any[]> {\n    try {\n      const result = await db.execute(\n        sql`SELECT \n          column_name,\n          data_type,\n          is_nullable,\n          column_default,\n          character_maximum_length\n        FROM information_schema.columns \n        WHERE table_schema = 'public' \n        AND table_name = ${tableName}\n        ORDER BY ordinal_position`,\n      );\n      \n      return result;\n    } catch (error) {\n      console.error('Failed to get table structure:', error);\n      return [];\n    }\n  }\n\n  /**\n   * 获取数据库版本信息\n   * @param databaseService 数据库服务实例\n   * @returns 数据库版本\n   */\n  static async getDatabaseVersion(\n    db: PostgresJsDatabase\n  ): Promise<string> {\n    try {\n      const result = await db.execute(sql`SELECT version()`);\n      return result[0]?.version as string || 'Unknown';\n    } catch (error) {\n      console.error('Failed to get database version:', error);\n      return 'Unknown';\n    }\n  }\n\n  /**\n   * 获取连接统计信息\n   * @param databaseService 数据库服务实例\n   * @returns 连接统计信息\n   */\n  static async getConnectionStats(\n    db: PostgresJsDatabase\n  ): Promise<any> {\n    try {\n      const result = await db.execute(sql`\n        SELECT \n          count(*) as total_connections,\n          count(*) FILTER (WHERE state = 'active') as active_connections,\n          count(*) FILTER (WHERE state = 'idle') as idle_connections\n        FROM pg_stat_activity \n        WHERE datname = current_database()\n      `);\n      \n      return result[0] || {};\n    } catch (error) {\n      console.error('Failed to get connection stats:', error);\n      return {};\n    }\n  }\n\n  /**\n   * 健康检查\n   * @param databaseService 数据库服务实例\n   * @returns 健康状态\n   */\n  static async healthCheck(\n    databaseService: DataPaasDatabaseService\n  ): Promise<{\n    isHealthy: boolean;\n    isConnected: boolean;\n    version?: string;\n    connectionInfo?: any;\n    error?: string;\n  }> {\n    try {\n      const isConnected = databaseService.isConnected();\n      const connectionInfo = databaseService.getConnectionInfo();\n\n      const db = databaseService.getDatabase();\n      if (!isConnected) {\n        return {\n          isHealthy: false,\n          isConnected: false,\n          error: 'Database not connected'\n        };\n      }\n\n      const version = await this.getDatabaseVersion(db);\n      \n      return {\n        isHealthy: true,\n        isConnected: true,\n        version,\n        connectionInfo\n      };\n    } catch (error) {\n      return {\n        isHealthy: false,\n        isConnected: false,\n        error: error instanceof Error ? error.message : 'Unknown error'\n      };\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;;;;;;;;;;;;;;ACAA,IAAAA,iBAAkE;AAClE,yBAAiD;AACjD,yBAAoB;AACpB,sBAAqB;;;ACHd,IAAMC,mBAAmB;AACzB,IAAMC,kBAAkB;AACxB,IAAMC,0BAA0BC,uBAAO,yBAAA;;;ADG9C,+BAA2B;AAC3B,2BAAsC;;;AEK/B,SAASC,oBAAoBC,kBAA0BC,QAAyB;AACrF,QAAM,EAAEC,gBAAgBC,aAAaC,mBAAmBC,KAAKC,OAAM,IAAKL,UAAU,CAAC;AAEnF,MAAIM;AACJ,MAAI;AACFA,0BAAsB,IAAIC,IAAIR,gBAAAA;AAC9B,UAAMS,OAAO,IAAID,IAAIR,gBAAAA;AACrBS,SAAKC,aAAaC,OAAO,QAAA;AACzBX,uBAAmBS,KAAKG,SAAQ;EAClC,SAASC,OAAO;AACd,UAAM,IAAIC,MAAM,iCAAA;EAClB;AACA,QAAM,EAAEC,UAAUC,UAAUC,UAAUC,MAAMC,SAAQ,IAAKZ;AACzD,QAAMa,mBAAmB;IACvBL,UAAUM,mBAAmBN,QAAAA;IAC7BC,UAAUK,mBAAmBL,QAAAA;IAC7BM,MAAM;MAACL;;IACPC,MAAM;MAACK,SAASL,IAAAA,KAAS;;IACzBC,UAAUE,mBAAmBF,QAAAA;IAC7BK,KAAKtB,kBAAkB;IACvBuB,cAActB,eAAe;IAC7BuB,iBAAiBtB,qBAAqB;IACtCC;IACAsB,UAAUrB,SAASsB,QAAQC,MAAMC;EACnC;AACA,SAAO;IAAE9B;IAAkBoB;EAAiB;AAC9C;AA1BgBrB;;;ACVhB,oBAAuB;AAIvB,IAAMgC,aAAa,IAAIC,qBAAO,UAAA;AAKvB,SAASC,iBAAiBC,MAAmBC,YAAwBC,gBAAqC;AAC/G,QAAMC,YAAY,wBAAkDC,UAAAA;AAClE,QAAIA,SAAS,OAAOA,MAAMC,SAAS,YAAY;AAC7C,aAAO,IAAIC,MAAMF,OAAO;QACtBG,IAAIC,QAAQC,MAAMC,UAAQ;AACxB,cAAID,SAAS,QAAQ;AACnB,mBAAO,OAAOE,aAAuDC,eAAAA;AACnE,oBAAMC,SAAUT,MAA6CU,mBAAmBC,QAASX,MAA2CU,QAAQ,CAAA,IAAK;AACjJ,oBAAME,OAAOf,WAAWgB,WAAWJ,QAAQX,gBAAgBgB,WAAAA,GAAcC,eAAAA;AAEzEH,mBAAKI,aAAa,UAAU,MAAA;AAC5BJ,mBAAKI,aAAa,eAAe,UAAA;AAEjC,oBAAMC,QAAQC,KAAKC,IAAG;AACtB,kBAAI;AACF,sBAAMC,SAAS,MAAMhB;AACrB,sBAAMiB,eAAeD,QAAQE,YAAY;AAGzC,oBAAI,CAACD,cAAc;AACjB,wBAAME,WAAWL,KAAKC,IAAG,IAAKF;AAE9B,wBAAMO,eAAeC,KAAKC,UAAU;oBAClCC,WAAWP,QAAQE;oBACnB1B,KAAKwB,QAAQQ,WAAWC;oBACxBC,aAAaP;oBACbQ,cAAcX,OAAOY,UAAU;oBAC/BC,eAAe;kBACjB,CAAA;AAEAxC,6BAAWyC,IAAIV,cAAc;oBAC3BW,aAAa;oBACbC,wBAAwB;oBACxBC,kBAAkBzB;kBACpB,CAAA;gBACF,OAAO;AACLf,6BAAWyC,UAAU1B,IAAAA;gBACvB;AAEAA,qBAAK2B,IAAG;AACR,uBAAOhC,cAAcA,YAAYa,MAAAA,IAAUA;cAC7C,SAASoB,OAAO;AACd,sBAAMjB,WAAWL,KAAKC,IAAG,IAAKF;AAC9B,sBAAMR,UAAUT,MAA6CU,mBAAmBC,QAASX,MAA2CU,QAAQ,CAAA,IAAK;AACjJ,sBAAMc,eAAeC,KAAKC,UAAU;kBAClCC,WAAW;kBACX/B,KAAKa;kBACLqB,aAAaP;kBACbQ,cAAc;kBACdE,eAAeO,iBAAiBC,QAAQD,MAAME,UAAUC,OAAOH,KAAAA;gBACjE,CAAA;AACA/C,2BAAWyC,IAAIV,cAAc;kBAC3BW,aAAa;kBACbC,wBAAwB;kBACxBC,kBAAkBzB;gBACpB,CAAA;AAEAA,qBAAK2B,IAAG;AACR,uBAAO/B,aAAaA,WAAWgC,KAAAA,IAASI,QAAQC,OAAOL,KAAAA;cACzD;YACF;UACF;AACA,gBAAMM,QAAQC,QAAQ5C,IAAIC,QAAQC,MAAMC,QAAAA;AACxC,iBAAO,OAAOwC,UAAU,aAAaA,MAAME,KAAK1C,QAAAA,IAAYwC;QAC9D;MACF,CAAA;IACF;AACA,WAAO9C;EACT,GAnEkB;AAqElB,SAAO,IAAIE,MAAMN,MAAK;IACpBqD,MAAM7C,QAAQ8C,SAASC,UAAQ;AAC7B,YAAMnD,QAAQ+C,QAAQE,MAAM7C,QAAQ8C,SAASC,QAAAA;AAC7C,aAAOpD,UAAUC,KAAAA;IACnB;IACAG,IAAIC,QAAQC,MAAMC,UAAQ;AACxB,YAAMwC,QAAQC,QAAQ5C,IAAIC,QAAQC,MAAMC,QAAAA;AACxC,UAAID,SAAS,WAAW,OAAOyC,UAAU,YAAY;AACnD,eAAO,CAAIM,KAAiEC,OAAAA;AAC1E,cAAI,OAAOD,QAAQ,YAAY;AAC7B,mBAAON,MAAMQ,KAAKlD,QAAQ,CAACmD,OAAgCH,IAAIzD,iBAAiB4D,IAA+B1D,YAAYC,cAAAA,CAAAA,CAAAA;UAC7H;AACA,cAAI,OAAOuD,OAAO,YAAY;AAC5B,mBAAOP,MAAMQ,KAAKlD,QAAQgD,KAAK,CAACG,OAAgCF,GAAG1D,iBAAiB4D,IAA+B1D,YAAYC,cAAAA,CAAAA,CAAAA;UACjI;AACA,iBAAOgD,MAAMQ,KAAKlD,QAAQgD,GAAAA;QAC5B;MACF;AACA,UAAI/C,SAAS,YAAY,OAAOyC,UAAU,YAAY;AACpD,eAAO,CAAmEU,UAAkBC,MAAkBC,YAAAA;AAC5G,gBAAM1D,QAAQ8C,MAAMQ,KAAKlD,QAAQoD,UAAUC,MAAMC,OAAAA;AACjD,iBAAO3D,UAAUC,KAAAA;QACnB;MACF;AACA,aAAO,OAAO8C,UAAU,aAAaA,MAAME,KAAK1C,QAAAA,IAAYwC;IAC9D;EACF,CAAA;AACF;AAjGgBnD;;;;;;;;;;;;;;;;;;;;AHET,IAAMgE,yBAAN,MAAMA;SAAAA;;;;;;EACHC,KAAgC;EAChCC,wBAAuD;EACvDC,cAAuB;EACvBC,uBAAgC;EACxC,YAEmBC,QAEAC,YACAC,gBACjB;SAJiBF,SAAAA;SAEAC,aAAAA;SACAC,iBAAAA;AAEjB,SAAKH,uBAAuB,CAAC,CAACI,QAAQC,IAAIC;EAC5C;EAEA,MAAMC,eAAe;AACnB,UAAM,KAAKC,WAAU;EACvB;EAEA,MAAMC,kBAAkB;AACtB,UAAM,KAAKC,WAAU;EACvB;;;;;EAMA,MAAMF,aAA0C;AAC9C,QAAI,KAAKR,sBAAsB;AAC7B,WAAKD,cAAc;AACnB,WAAKF,KAAK;QAACc,MAAM;MAAsB;AACvC,aAAO,KAAKd;IACd;AACA,QAAI;AACF,YAAM,EAAEe,kBAAkBC,iBAAgB,IAAKC,oBAAoB,KAAKb,OAAOW,kBAAkB,KAAKX,MAAM;AAC5G,YAAMc,cAAUC,gBAAAA,SAASJ,kBAAkBC,gBAAAA;AAC3C,YAAMI,WAAWC,iBAAiBH,SAAS,KAAKb,YAAY,KAAKC,cAAc;AAE/E,WAAKN,SAAKsB,4BAAQF,UAAU;QAAEG,QAAQ,KAAKnB,OAAOmB;QAAQC,QAAQ;MAAM,CAAA;AACxE,WAAKvB,wBAAwB;QAC3Bc;QACAC,kBAAkB;UAChBS,KAAKT,iBAAiBS;UACtBC,mBAAmBV,iBAAiBW;UACpCC,aAAaZ,iBAAiBa;UAC9BC,gBAAgBd,iBAAiBe;QACnC;MACF;AACA,WAAK7B,cAAc;AACnB,aAAO,KAAKF;IACd,SAASgC,OAAO;AACd,WAAK9B,cAAc;AACnB,YAAM,IAAI+B,MAAM,6CAA6CD,iBAAiBC,QAAQD,MAAME,UAAU,eAAA,EAAiB;IACzH;EACF;;;;;EAMAC,cAAkC;AAChC,QAAI,CAAC,KAAKnC,MAAM,CAAC,KAAKE,aAAa;AACjC,YAAM,IAAI+B,MAAM,oDAAA;IAClB;AACA,WAAO,KAAKjC;EACd;;;;;EAMA,MAAMoC,iBAAmC;AACvC,QAAI,CAAC,KAAKpC,IAAI;AACZ,aAAO;IACT;AAEA,QAAI;AACF,YAAM,KAAKA,GAAGqC,QAAQC,gCAAa;AACnC,aAAO;IACT,SAASN,OAAO;AACdO,cAAQP,MAAM,oCAAoCA,KAAAA;AAClD,WAAK9B,cAAc;AACnB,aAAO;IACT;EACF;;;;;;EAOA,MAAMsC,UAAUC,gBAAyB,OAAoC;AAC3E,UAAM,KAAK5B,WAAU;AACrB,WAAO,KAAKF,WAAU;EACxB;;;;EAKA,MAAME,aAA4B;AAChC,QAAI,KAAKb,IAAI;AACX,UAAI;AAEF,aAAKA,KAAK;AACV,aAAKE,cAAc;AACnB,aAAKD,wBAAwB;MAC/B,SAAS+B,OAAO;AACdO,gBAAQP,MAAM,sCAAsCA,KAAAA;MACtD;IACF;EACF;;;;;EAMAU,oBAAmD;AACjD,WAAO,KAAKzC;EACd;;;;;EAMA0C,sBAA+B;AAC7B,WAAO,KAAKzC;EACd;;;;;;EAQA,MAAM0C,YAAeC,UAA8D;AACjF,UAAM7C,KAAK,KAAKmC,YAAW;AAC3B,WAAOnC,GAAG4C,YAAYC,QAAAA;EACxB;;;;;EAMAC,0BAAmG;AACjG,WAAO;MACL5C,aAAa,KAAKA;MAClB6C,gBAAgB,KAAK9C;IACvB;EACF;AACF;;;;;;;;;;;;;;AIjKA,IAAA+C,iBAOO;;;ACPP,IAAAC,iBAA4C;;;;;;;;;;;;AAMrC,IAAMC,0BAAN,MAAMA;SAAAA;;;;EACX,YACmBC,iBACjB;SADiBA,kBAAAA;EAChB;;;;;EAMHC,cAAkC;AAChC,WAAO,KAAKD,gBAAgBC,YAAW;EACzC;;;;;EAMAC,qBAA6C;AAC3C,WAAO,KAAKF;EACd;;;;;EAOA,MAAMG,iBAAmC;AACvC,WAAO,KAAKH,gBAAgBG,eAAc;EAC5C;;;;;;EAOA,MAAMC,UAAUC,eAAwB,OAAoC;AAC1E,WAAO,KAAKL,gBAAgBI,UAAUC,YAAAA;EACxC;;;;;EAMAC,oBAAmD;AACjD,WAAO,KAAKN,gBAAgBM,kBAAiB;EAC/C;;;;;EAMAC,cAAuB;AACrB,WAAO,KAAKP,gBAAgBQ,oBAAmB;EACjD;;;;;;EAOA,MAAMC,YAAeC,UAA8D;AACjF,WAAO,KAAKV,gBAAgBS,YAAYC,QAAAA;EAC1C;;;;;EAMAC,0BAAmG;AACjG,WAAO,KAAKX,gBAAgBW,wBAAuB;EACrD;;;;EAMA,MAAMC,kBAAiC;AACrC,UAAM,KAAKZ,gBAAgBa,WAAU;EACvC;AACF;;;;;;;;;;ACrFA,IAAAC,iBAAmD;AAEnD,cAAyB;;;;;;;;;;;;;;;;;;AAiBlB,IAAMC,gCAAN,MAAMA;SAAAA;;;;EACX,YAEmBC,QACjB;SADiBA,SAAAA;EAChB;EAEHC,IAAIC,KAAcC,MAAgBC,MAAoB;AACpD,UAAMC,aAAa,KAAKL,OAAOK,cAAc;AAC7C,UAAMC,cACHJ,IAAgDI,eAAe,CAAC;AACnE,UAAMC,SAASD,YAAYE,aAAa,CAAA,GAAIC,KAAK,GAAA;AACjD,UAAMC,SAASJ,YAAYI;AAC3B,UAAMC,kBAAkBL,YAAYK;AACpC,UAAMC,WAAWN,YAAYM;AAC7B,UAAMC,OAAO;;MAEXH,SACI,4BAA4BA,MAAAA,MAC5B;;MAEJC,kBAAkB,gCAAgCN,UAAAA,MAChDK,SACI,iCAAiCL,UAAAA,MACjC,wBAAwBA,UAAAA;;MAE9BE,SAAS,6BAA6BA,KAAAA;;MAEtCK,WACI,8BAA8BA,QAAAA,MAC9B;MAEHE,OAAOC,OAAAA,EACPN,KAAK,GAAA;AAGRO,IAAQC,2BAAmB;MAAEC,QAAQL;IAAK,GAAG,MAAA;AAC3CT,WAAAA;IACF,CAAA;EACF;AACF;;;;;;;;;;;AF5CA,IAAAe,wBAA6E;AAC7E,IAAAC,4BAA2B;;;;;;;;AAsB3B,SAASC,uBAAAA;AACP,MAAI;AAKF,UAAMC,YAAWC,QAAQ,yBAAA;AACzB,UAAMC,KACJF,aACA,OAAOA,UAASG,wBAAwB,cACxC,OAAOH,UAASI,uBAAuB;AACzC,QAAI,CAACF,IAAI;AACP,YAAM,IAAIG,MACR,0GAAA;IAEJ;EACF,SAASC,KAAK;AACZ,UAAMC,SAASD,eAAeD,QAAQC,IAAIE,UAAUC,OAAOH,GAAAA;AAC3D,UAAM,IAAID,MAAM,uFAA2BE,MAAAA,EAAQ;EACrD;AACF;AApBSR;AAuBTA,qBAAAA;AAcO,IAAMW,iBAAN,MAAMA,gBAAAA;SAAAA;;;EACXC,UAAUC,WAA+B;EAAC;;;;;;EAM1C,OAAOC,QAAQC,SAAwC;AACrD,QAAIC,aAAa;AACjB,QAAIC,kBAAkB;AACtB,QAAIF,QAAQG,kBAAkB;AAC5B,UAAI;AACF,cAAMC,MAAM,IAAIC,IAAIL,QAAQG,gBAAgB;AAE5CF,qBAAaG,IAAIE,aAAaC,IAC5B,QAAA;AAEFL,0BAAkBE,IAAIE,aAAaC,IAAI,SAAA,MAAe;MACxD,SAASf,KAAK;AACZS,qBAAa;MACf;IACF;AACA,UAAMO,SAAyB;MAC7BL,kBAAkBH,QAAQG;MAC1BM,QAAQT,QAAQS;MAChBC,QAAQV,QAAQU,UAAU;MAC1BC,SAASX,QAAQW,WAAW;MAC5BC,gBAAgBZ,QAAQY,kBAAkB;MAC1CC,aAAab,QAAQa,eAAe;MACpCC,mBAAmBd,QAAQc,qBAAqB;MAChDC,KAAKf,QAAQe,QAAQb,kBAAkB,YAAY;MACnDc,aAAahB,QAAQgB,eAAe,OAAO,OAAO,CAAC,CAAChB,QAAQgB;MAC5Df;IACF;AAEA,UAAMgB,YAAwB;;MAE5B;QACEC,SAASC;QACTC,UAAUZ;MACZ;;MAEA;QACEU,SAASG;QACTC,UAAUD;MACZ;;MAEA;QACEH,SAASK;QACTD,UAAUC;MACZ;MACA;QACEL,SAASM;QACTF,UAAUE;MACZ;;MAEA;QACEN,SAASO;QACTC,YAAY,8BAAOC,YAAAA;AACjB,gBAAMA,QAAQC,WAAU;AACxB,iBAAOD,QAAQE,YAAW;QAC5B,GAHY;QAIZC,QAAQ;UAACT;;MACX;;AAGF,WAAO;MACLU,QAAQnC;MACRqB;MACAe,SAAS;QACPX;QACAE;QACAC;QACAC;QACAN;;MAEFc,QAAQ;IACV;EACF;;;;;;;;;;;;;;;;;EAkBA,OAAOC,aAAalC,SAAoD;AACtE,UAAMmC,iBAAiB,KAAKC,qBAAqBpC,OAAAA;AAEjD,UAAMiB,YAAwB;SACzBkB;;MAEH;QACEjB,SAASG;QACTC,UAAUD;MACZ;;MAEA;QACEH,SAASK;QACTD,UAAUC;MACZ;MACA;QACEL,SAASM;QACTF,UAAUE;MACZ;;MAEA;QACEN,SAASO;QACTC,YAAY,8BAAOC,YAAAA;AACjB,gBAAMA,QAAQC,WAAU;AACxB,iBAAOD,QAAQE,YAAW;QAC5B,GAHY;QAIZC,QAAQ;UAACT;;MACX;;AAGF,WAAO;MACLU,QAAQnC;MACRyC,SAASrC,QAAQqC,WAAW,CAAA;MAC5BpB;MACAe,SAAS;QACPX;QACAE;QACAC;QACAC;QACAN;;MAEFc,QAAQ;IACV;EACF;;;;EAKA,OAAeG,qBACbpC,SACY;AACZ,QAAIA,QAAQ0B,YAAY;AACtB,aAAO;QAAC,KAAKY,0BAA0BtC,OAAAA;;IACzC;AAEA,UAAM,IAAIT,MAAM,iEAAA;EAClB;;;;EAKA,OAAe+C,0BACbtC,SACU;AACV,QAAIA,QAAQ0B,YAAY;AACtB,aAAO;QACLR,SAASC;QACTO,YAAY,iCAAUa,SAAAA;AACpB,gBAAM/B,SAAS,MAAMR,QAAQ0B,WAAU,GAAKa,IAAAA;AAC5C,iBAAO,KAAKC,gBAAgBhC,MAAAA;QAC9B,GAHY;QAIZsB,QAAQ9B,QAAQ8B,UAAU,CAAA;MAC5B;IACF;AAEA,UAAM,IAAIvC,MAAM,uBAAA;EAClB;;;;EAKA,OAAeiD,gBAAgBhC,QAAwC;AACrE,QAAIP,aAAa;AACjB,QAAIC,kBAAkB;AACtB,QAAIM,OAAOL,kBAAkB;AAC3B,UAAI;AACF,cAAMC,MAAM,IAAIC,IAAIG,OAAOL,gBAAgB;AAC3CF,qBAAaG,IAAIE,aAAaC,IAC5B,QAAA;AAEFL,0BAAkBE,IAAIE,aAAaC,IAAI,SAAA,MAAe;MACxD,SAASf,KAAK;AACZS,qBAAa;MACf;IACF;AACA,WAAO;MACLE,kBAAkBK,OAAOL;MACzBM,QAAQD,OAAOC;MACfC,QAAQF,OAAOE,UAAU;MACzBC,SAASH,OAAOG,WAAW;MAC3BC,gBAAgBJ,OAAOI,kBAAkB;MACzCC,aAAaL,OAAOK,eAAe;MACnCC,mBAAmBN,OAAOM,qBAAqB;MAC/CC,KAAKP,OAAOO,QAAQb,kBAAkB,YAAY;MAClDc,aAAaR,OAAOQ,eAAe,OAAO,OAAO,CAAC,CAACR,OAAOQ;MAC1Df;IACF;EACF;AACF;;;IAvNEgB,WAAW;MAAC;QACVC,SAASuB;QACTf,YAAY,wBAACgB,YAAsCC,QAAAA;AACjD,iBAAOD,cAAc,IAAIE,qCAAWD,GAAAA;QACtC,GAFY;QAGZb,QAAQ;UAAC;YACPe,OAAOC;YACPC,UAAU;UACZ;UAAGC;;MACL;;;;;;AGvEF,IAAAC,sBAAoB;AAOb,IAAMC,gBAAN,MAAMA;EAPb,OAOaA;;;;;;;;;;EAQX,aAAaC,YACXC,iBACAC,WACAC,UACwB;AACxB,QAAI;AACF,UAAI,CAACF,gBAAgBG,YAAW,GAAI;AAClC,cAAM,IAAIC,MAAM,wBAAA;MAClB;AAEA,YAAMC,KAAKL,gBAAgBM,YAAW;AACtC,aAAO,MAAML,UAAUI,EAAAA;IACzB,SAASE,OAAO;AACdC,cAAQD,MAAM,8BAA8BA,KAAAA;AAC5C,aAAOL;IACT;EACF;;;;;;;;EASA,aAAaO,aACXT,iBACAU,YACAC,YAAoB,IACN;AACd,UAAMC,UAAe,CAAA;AACrB,UAAMP,KAAKL,gBAAgBM,YAAW;AAEtC,aAASO,IAAI,GAAGA,IAAIH,WAAWI,QAAQD,KAAKF,WAAW;AACrD,YAAMI,QAAQL,WAAWM,MAAMH,GAAGA,IAAIF,SAAAA;AACtC,YAAMM,eAAe,MAAMC,QAAQC,IACjCJ,MAAMK,IAAInB,CAAAA,cAAaA,UAAUI,EAAAA,CAAAA,CAAAA;AAEnCO,cAAQS,KAAI,GAAIJ,YAAAA;IAClB;AAEA,WAAOL;EACT;;;;;;;;;EAUA,aAAaU,iBACXtB,iBACAC,WACAsB,aAAqB,GACrBC,aAAqB,KACT;AACZ,QAAIC;AAEJ,aAASC,UAAU,GAAGA,WAAWH,YAAYG,WAAW;AACtD,UAAI;AACF,cAAMrB,KAAKL,gBAAgBM,YAAW;AACtC,eAAO,MAAML,UAAUI,EAAAA;MACzB,SAASE,OAAO;AACdkB,oBAAYlB;AAEZ,YAAImB,YAAYH,YAAY;AAC1B;QACF;AAEAf,gBAAQmB,KAAK,sCAAsCD,UAAU,CAAA,IAAKH,aAAa,CAAA,MAAOhB,KAAAA;AAGtF,cAAM,IAAIW,QAAQU,CAAAA,YAAWC,WAAWD,SAASJ,cAAcE,UAAU,EAAA,CAAA;MAC3E;IACF;AAEA,UAAMD;EACR;;;;;;;EAQA,aAAaK,YACXzB,IACA0B,WACkB;AAClB,QAAI;AACF,YAAMC,SAAS,MAAM3B,GAAG4B,QACtBC;;;6BAGqBH,SAAAA;UACnB;AAGJ,aAAOC,OAAO,CAAA,GAAIG,UAAqB;IACzC,SAAS5B,OAAO;AACdC,cAAQD,MAAM,oCAAoCA,KAAAA;AAClD,aAAO;IACT;EACF;;;;;;;EAQA,aAAa6B,kBACX/B,IACA0B,WACgB;AAChB,QAAI;AACF,YAAMC,SAAS,MAAM3B,GAAG4B,QACtBC;;;;;;;;2BAQmBH,SAAAA;kCACO;AAG5B,aAAOC;IACT,SAASzB,OAAO;AACdC,cAAQD,MAAM,kCAAkCA,KAAAA;AAChD,aAAO,CAAA;IACT;EACF;;;;;;EAOA,aAAa8B,mBACXhC,IACiB;AACjB,QAAI;AACF,YAAM2B,SAAS,MAAM3B,GAAG4B,QAAQC,yCAAqB;AACrD,aAAOF,OAAO,CAAA,GAAIM,WAAqB;IACzC,SAAS/B,OAAO;AACdC,cAAQD,MAAM,mCAAmCA,KAAAA;AACjD,aAAO;IACT;EACF;;;;;;EAOA,aAAagC,mBACXlC,IACc;AACd,QAAI;AACF,YAAM2B,SAAS,MAAM3B,GAAG4B,QAAQC;;;;;;;OAO/B;AAED,aAAOF,OAAO,CAAA,KAAM,CAAC;IACvB,SAASzB,OAAO;AACdC,cAAQD,MAAM,mCAAmCA,KAAAA;AACjD,aAAO,CAAC;IACV;EACF;;;;;;EAOA,aAAaiC,YACXxC,iBAOC;AACD,QAAI;AACF,YAAMG,cAAcH,gBAAgBG,YAAW;AAC/C,YAAMsC,iBAAiBzC,gBAAgB0C,kBAAiB;AAExD,YAAMrC,KAAKL,gBAAgBM,YAAW;AACtC,UAAI,CAACH,aAAa;AAChB,eAAO;UACLwC,WAAW;UACXxC,aAAa;UACbI,OAAO;QACT;MACF;AAEA,YAAM+B,UAAU,MAAM,KAAKD,mBAAmBhC,EAAAA;AAE9C,aAAO;QACLsC,WAAW;QACXxC,aAAa;QACbmC;QACAG;MACF;IACF,SAASlC,OAAO;AACd,aAAO;QACLoC,WAAW;QACXxC,aAAa;QACbI,OAAOA,iBAAiBH,QAAQG,MAAMqC,UAAU;MAClD;IACF;EACF;AACF;","names":["import_common","DRIZZLE_DATABASE","DATAPAAS_CONFIG","DATAPAAS_LOGGER_SERVICE","Symbol","parseConnectionInfo","connectionString","config","maxConnections","idleTimeout","connectionTimeout","ssl","logger","parsedConfigFromUrl","URL","temp","searchParams","delete","toString","error","Error","username","password","hostname","port","pathname","connectionConfig","decodeURIComponent","host","parseInt","max","idle_timeout","connect_timeout","onnotice","console","log","undefined","nestLogger","Logger","genPostgresProxy","sql","observable","requestContext","wrapQuery","query","then","Proxy","get","target","prop","receiver","onfulfilled","onrejected","sqlStr","strings","Array","span","startTrace","getContext","requestRootSpan","setAttribute","start","Date","now","result","isSetRoleSql","command","duration","parsedParams","JSON","stringify","operation","statement","string","duration_ms","result_count","length","error_message","log","source_type","paas_attributes_module","paas_parent_span","dropTrace","end","error","Error","message","String","Promise","reject","value","Reflect","bind","apply","thisArg","argArray","arg","cb","call","tx","queryStr","args","options","DrizzleDatabaseManager","db","currentConnectionInfo","isConnected","skipInitDbConnection","config","observable","requestContext","process","env","DEPRECATED_SKIP_INIT_DB_CONNECTION","onModuleInit","initialize","onModuleDestroy","disconnect","name","connectionString","connectionConfig","parseConnectionInfo","baseSql","postgres","sqlProxy","genPostgresProxy","drizzle","schema","logger","ssl","connectionTimeout","connect_timeout","idleTimeout","idle_timeout","maxConnections","max","error","Error","message","getDatabase","testConnection","execute","sql","console","reconnect","_forceRefresh","getConnectionInfo","isDatabaseConnected","transaction","callback","getConnectionPoolStatus","connectionInfo","import_common","import_common","DataPaasDatabaseService","databaseManager","getDatabase","getDatabaseManager","testConnection","reconnect","forceRefresh","getConnectionInfo","isConnected","isDatabaseConnected","transaction","callback","getConnectionPoolStatus","onModuleDestroy","disconnect","import_common","SqlExecutionContextMiddleware","config","use","req","_res","next","roleSchema","userContext","roles","userRoles","join","userId","isSystemAccount","userType","sqls","filter","Boolean","session","runWithAuthContext","preSql","import_nestjs_common","import_nestjs_observable","verifyDrizzlePatched","postgres","require","ok","getAuthContextStore","runWithAuthContext","Error","err","reason","message","String","DataPaasModule","configure","_consumer","forRoot","options","roleSchema","sslModeRequired","connectionString","url","URL","searchParams","get","config","schema","logger","timeout","maxConnections","idleTimeout","connectionTimeout","ssl","autoContext","providers","provide","DATAPAAS_CONFIG","useValue","DrizzleDatabaseManager","useClass","DataPaasDatabaseService","SqlExecutionContextMiddleware","DRIZZLE_DATABASE","useFactory","manager","initialize","getDatabase","inject","module","exports","global","forRootAsync","asyncProviders","createAsyncProviders","imports","createAsyncConfigProvider","args","normalizeConfig","DATAPAAS_LOGGER_SERVICE","observable","rtx","Observable","token","OBSERVABLE_SERVICE","optional","RequestContextService","import_drizzle_orm","DatabaseUtils","safeExecute","databaseService","operation","fallback","isConnected","Error","db","getDatabase","error","console","batchExecute","operations","batchSize","results","i","length","batch","slice","batchResults","Promise","all","map","push","executeWithRetry","maxRetries","retryDelay","lastError","attempt","warn","resolve","setTimeout","tableExists","tableName","result","execute","sql","exists","getTableStructure","getDatabaseVersion","version","getConnectionStats","healthCheck","connectionInfo","getConnectionInfo","isHealthy","message"]}
+{"version":3,"sources":["../src/index.ts","../src/database/drizzle-manager.ts","../src/database/const.ts","../src/utils.ts","../src/database/postgres-proxy.ts","../src/database/database.module.ts","../src/database/database.service.ts","../src/database/sql-exection-context.middleware.ts","../src/database/database-utils.ts"],"sourcesContent":["// 导出所有公共API\n\nexport * from './database';\nexport * from './types';\n","import { Injectable, OnModuleInit, OnModuleDestroy, Inject } from '@nestjs/common';\nimport { drizzle, type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { sql } from 'drizzle-orm';\nimport postgres from 'postgres';\nimport { DATAPAAS_CONFIG, DATAPAAS_LOGGER_SERVICE } from './const';\nimport { Observable } from '@lark-apaas/nestjs-observable';\nimport { RequestContextService } from '@lark-apaas/nestjs-common';\n\nimport { type DataPaasConfig, type DatabaseConnectionInfo } from '../types';\nimport { parseConnectionInfo } from '../utils';\nimport { genPostgresProxy } from \"./postgres-proxy\";\n@Injectable()\nexport class DrizzleDatabaseManager implements OnModuleInit, OnModuleDestroy {\n  private db: PostgresJsDatabase | null = null;\n  private currentConnectionInfo: DatabaseConnectionInfo | null = null;\n  private isConnected: boolean = false;\n  private skipInitDbConnection: boolean = false;\n  constructor(\n    @Inject(DATAPAAS_CONFIG)\n    private readonly config: DataPaasConfig,\n    @Inject(DATAPAAS_LOGGER_SERVICE)\n    private readonly observable: Observable,\n    private readonly requestContext: RequestContextService,\n  ) { \n    this.skipInitDbConnection = !!process.env.DEPRECATED_SKIP_INIT_DB_CONNECTION;\n  }\n\n  async onModuleInit() {\n    await this.initialize();\n  }\n\n  async onModuleDestroy() {\n    await this.disconnect();\n  }\n\n  /**\n   * 初始化数据库连接\n   * @returns Drizzle 数据库实例\n   */\n  async initialize(): Promise<PostgresJsDatabase> {\n    if (this.skipInitDbConnection) {\n      this.isConnected = true;\n      this.db = {name: 'skipInitDbConnection'} as unknown as PostgresJsDatabase;\n      return this.db;\n    }\n    try {\n      const { connectionString, connectionConfig } = parseConnectionInfo(this.config.connectionString, this.config);\n      const baseSql = postgres(connectionString, connectionConfig);\n      const sqlProxy = genPostgresProxy(baseSql, this.observable, this.requestContext);\n      // 创建 Drizzle 实例\n      this.db = drizzle(sqlProxy, { schema: this.config.schema, logger: false });\n      this.currentConnectionInfo = {\n        connectionString,\n        connectionConfig: {\n          ssl: connectionConfig.ssl,\n          connectionTimeout: connectionConfig.connect_timeout,\n          idleTimeout: connectionConfig.idle_timeout,\n          maxConnections: connectionConfig.max,\n        },\n      };\n      this.isConnected = true;\n      return this.db;\n    } catch (error) {\n      this.isConnected = false;\n      throw new Error(`Failed to initialize database connection: ${error instanceof Error ? error.message : 'Unknown error'}`);\n    }\n  }\n\n  /**\n   * 获取数据库实例\n   * @returns Drizzle 数据库实例\n   */\n  getDatabase(): PostgresJsDatabase {\n    if (!this.db || !this.isConnected) {\n      throw new Error('Database not initialized. Call initialize() first.');\n    }\n    return this.db;\n  }\n\n  /**\n   * 测试数据库连接\n   * @returns 连接是否正常\n   */\n  async testConnection(): Promise<boolean> {\n    if (!this.db) {\n      return false;\n    }\n\n    try {\n      await this.db.execute(sql`SELECT 1`);\n      return true;\n    } catch (error) {\n      console.error('Database connection test failed:', error);\n      this.isConnected = false;\n      return false;\n    }\n  }\n\n  /**\n   * 重新连接数据库\n   * @param forceRefresh 是否强制刷新连接信息\n   * @returns 新的数据库实例\n   */\n  async reconnect(_forceRefresh: boolean = false): Promise<PostgresJsDatabase> {\n    await this.disconnect();\n    return this.initialize();\n  }\n\n  /**\n   * 断开数据库连接\n   */\n  async disconnect(): Promise<void> {\n    if (this.db) {\n      try {\n        // postgres-js 会自动处理连接关闭\n        this.db = null;\n        this.isConnected = false;\n        this.currentConnectionInfo = null;\n      } catch (error) {\n        console.error('Error disconnecting from database:', error);\n      }\n    }\n  }\n\n  /**\n   * 获取当前连接信息\n   * @returns 当前连接信息\n   */\n  getConnectionInfo(): DatabaseConnectionInfo | null {\n    return this.currentConnectionInfo;\n  }\n\n  /**\n   * 检查连接状态\n   * @returns 是否已连接\n   */\n  isDatabaseConnected(): boolean {\n    return this.isConnected;\n  }\n\n\n  /**\n   * 开始事务\n   * @param callback 事务回调函数\n   * @returns 事务结果\n   */\n  async transaction<T>(callback: (tx: PostgresJsDatabase) => Promise<T>): Promise<T> {\n    const db = this.getDatabase();\n    return db.transaction(callback);\n  }\n\n  /**\n   * 获取连接池状态\n   * @returns 连接池状态信息\n   */\n  getConnectionPoolStatus(): { isConnected: boolean; connectionInfo: DatabaseConnectionInfo | null } {\n    return {\n      isConnected: this.isConnected,\n      connectionInfo: this.currentConnectionInfo,\n    };\n  }\n}","export const DRIZZLE_DATABASE = 'DRIZZLE_DATABASE';\nexport const DATAPAAS_CONFIG = 'DATAPAAS_CONFIG';\nexport const DATAPAAS_LOGGER_SERVICE = Symbol('DATAPAAS_LOGGER_SERVICE');","import type postgres from 'postgres';\nimport { type Logger as DrizzleLogger } from 'drizzle-orm';\nimport { PostgresOptions } from './types';\n\ninterface ConnectionConfig {\n  maxConnections?: number;\n  idleTimeout?: number;\n  connectionTimeout?: number;\n  ssl?: PostgresOptions['ssl'];\n  logger?: boolean | DrizzleLogger;\n}\nexport function parseConnectionInfo(connectionString: string, config?: ConnectionConfig) {\n  const { maxConnections, idleTimeout, connectionTimeout, ssl, logger } = config || {};\n  // 创建 postgres 连接\n  let parsedConfigFromUrl: URL;\n  try {\n    parsedConfigFromUrl = new URL(connectionString);\n    const temp = new URL(connectionString);\n    temp.searchParams.delete('schema'); // 移除schema参数，避免 postgres 解析错误\n    connectionString = temp.toString();\n  } catch (error) {\n    throw new Error('Failed to parse connection info');\n  }\n  const { username, password, hostname, port, pathname } = parsedConfigFromUrl;\n  const connectionConfig = {\n    username: decodeURIComponent(username),\n    password: decodeURIComponent(password),\n    host: [hostname] as unknown as string,\n    port: [parseInt(port) || 5432] as unknown as number,\n    pathname: decodeURIComponent(pathname),\n    max: maxConnections || 1,\n    idle_timeout: idleTimeout || 20,\n    connect_timeout: connectionTimeout || 10,\n    ssl,\n    onnotice: logger ? console.log : undefined,\n  }\n  return { connectionString, connectionConfig };\n}","import type postgres from 'postgres';\nimport { Logger } from '@nestjs/common';\nimport { Span, Observable } from '@lark-apaas/nestjs-observable';\nimport { RequestContextService } from '@lark-apaas/nestjs-common';\n\nconst nestLogger = new Logger('Database');\n\n/**\n * 包装 postgres 实例以记录 SQL 和结果\n */\nexport function genPostgresProxy(sql: postgres.Sql, observable: Observable, requestContext: RequestContextService): postgres.Sql {\n  const wrapQuery = <T extends readonly (postgres.Row | undefined)[]>(query: postgres.PendingQuery<T>): postgres.PendingQuery<T> => {\n    if (query && typeof query.then === 'function') {\n      return new Proxy(query, {\n        get(target, prop, receiver) {\n          if (prop === 'then') {\n            return async (onfulfilled?: (value: postgres.RowList<T>) => unknown, onrejected?: (reason: unknown) => unknown) => {\n              const sqlStr = (query as unknown as Record<string, unknown>).strings instanceof Array ? (query as unknown as { strings: string[] }).strings[0] : '';\n              const span = observable.startTrace(sqlStr, requestContext?.getContext()?.requestRootSpan as Span);\n              const spanContext = {\n                traceId: span.spanContext().traceId,\n                spanId: span.spanContext().spanId\n              }\n              const logContext = {\n                source_type: 'platform',\n                paas_attributes_module: 'data',\n                paas_parent_span_context: spanContext,\n              }\n              span.setAttribute(\"module\", \"data\");\n              span.setAttribute(\"source_type\", \"platform\");\n\n              const start = Date.now();\n              try {\n                const result = await target;\n                const isSetRoleSql = result?.command === 'SET';\n\n                // rls SET ROLE 不上报日志\n                if (!isSetRoleSql) {\n                  const duration = Date.now() - start;\n\n                  const parsedParams = JSON.stringify({\n                    operation: result?.command,\n                    sql: result?.statement?.string,\n                    duration_ms: duration,\n                    result_count: result.length ?? 0,\n                    error_message: ''\n                  });\n\n                  nestLogger.log(parsedParams, logContext);\n                } else {\n                  observable.dropTrace(span);\n                }\n\n                span.end();\n                return onfulfilled ? onfulfilled(result) : result;\n              } catch (error) {\n                const duration = Date.now() - start;\n                const sqlStr = (query as unknown as Record<string, unknown>).strings instanceof Array ? (query as unknown as { strings: string[] }).strings[0] : '';\n                const parsedParams = JSON.stringify({\n                  operation: '',\n                  sql: sqlStr,\n                  duration_ms: duration,\n                  result_count: 0,\n                  error_message: error instanceof Error ? error.message : String(error)\n                });\n                nestLogger.log(parsedParams, logContext);\n\n                span.end();\n                return onrejected ? onrejected(error) : Promise.reject(error);\n              }\n            };\n          }\n          const value = Reflect.get(target, prop, receiver);\n          return typeof value === 'function' ? value.bind(receiver) : value;\n        },\n      }) as postgres.PendingQuery<T>;\n    }\n    return query;\n  };\n\n  return new Proxy(sql, {\n    apply(target, thisArg, argArray) {\n      const query = Reflect.apply(target, thisArg, argArray);\n      return wrapQuery(query);\n    },\n    get(target, prop, receiver) {\n      const value = Reflect.get(target, prop, receiver);\n      if (prop === 'begin' && typeof value === 'function') {\n        return <T>(arg: string | ((tx: postgres.TransactionSql) => T | Promise<T>), cb?: (tx: postgres.TransactionSql) => T | Promise<T>) => {\n          if (typeof arg === 'function') {\n            return value.call(target, (tx: postgres.TransactionSql) => arg(genPostgresProxy(tx as unknown as postgres.Sql, observable, requestContext) as unknown as postgres.TransactionSql));\n          }\n          if (typeof cb === 'function') {\n            return value.call(target, arg, (tx: postgres.TransactionSql) => cb(genPostgresProxy(tx as unknown as postgres.Sql, observable, requestContext) as unknown as postgres.TransactionSql));\n          }\n          return value.call(target, arg);\n        };\n      }\n      if (prop === 'unsafe' && typeof value === 'function') {\n        return <T extends readonly (postgres.Row | undefined)[] = postgres.Row[]>(queryStr: string, args?: unknown[], options?: postgres.UnsafeQueryOptions): postgres.PendingQuery<T> => {\n          const query = value.call(target, queryStr, args, options) as postgres.PendingQuery<T>;\n          return wrapQuery(query);\n        };\n      }\n      return typeof value === 'function' ? value.bind(receiver) : value;\n    },\n  }) as postgres.Sql;\n}","import {\n  Module,\n  DynamicModule,\n  Provider,\n  MiddlewareConsumer,\n  InjectionToken,\n  ModuleMetadata,\n} from '@nestjs/common';\nimport { DrizzleDatabaseManager } from './drizzle-manager';\nimport { DataPaasDatabaseService } from './database.service';\nimport { SqlExecutionContextMiddleware } from './sql-exection-context.middleware';\nimport { type DataPaasConfig } from '@/types';\nimport { DATAPAAS_CONFIG, DRIZZLE_DATABASE, DATAPAAS_LOGGER_SERVICE } from './const';\nexport { type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { RequestContextService, OBSERVABLE_SERVICE, ObservableService } from '@lark-apaas/nestjs-common';\nimport { Observable } from '@lark-apaas/nestjs-observable';\n\nexport interface DataPaasModuleOptions extends DataPaasConfig {}\n\n/**\n * 异步配置选项\n */\nexport interface DataPaasModuleAsyncOptions {\n  /**\n   * 需要导入的模块（例如 ConfigModule）\n   */\n  imports?: ModuleMetadata['imports'];\n  /**\n   * 使用工厂函数创建配置\n   */\n  useFactory?: (...args: unknown[]) => Promise<DataPaasConfig> | DataPaasConfig;\n  /**\n   * 工厂函数需要注入的依赖\n   */\n  inject?: InjectionToken[];\n}\n\nfunction verifyDrizzlePatched(): void {\n  try {\n    // 直接加载被 patch 的 CJS 模块并检查导出是否存在\n    // 注意：此路径与补丁文件一致：drizzle-orm/postgres-js/session.cjs\n    // 若未来包结构变化，请同步更新补丁与此校验\n    // eslint-disable-next-line @typescript-eslint/no-var-requires\n    const postgres = require('drizzle-orm/postgres-js');\n    const ok =\n      postgres &&\n      typeof postgres.getAuthContextStore === 'function' &&\n      typeof postgres.runWithAuthContext === 'function';\n    if (!ok) {\n      throw new Error(\n        'drizzle-orm 未检测到补丁导出：getAuthContextStore/runWithAuthContext',\n      );\n    }\n  } catch (err) {\n    const reason = err instanceof Error ? err.message : String(err);\n    throw new Error(`drizzle-orm 补丁未生效或版本不匹配：${reason}`);\n  }\n}\n\n// 在模块初始化阶段进行一次 drizzle-orm 补丁自检\nverifyDrizzlePatched();\n\n@Module({\n  providers: [{\n    provide: DATAPAAS_LOGGER_SERVICE,\n    useFactory: (observable: ObservableService | null, rtx: RequestContextService) => {\n      return observable ?? new Observable(rtx);\n    },\n    inject: [{\n      token: OBSERVABLE_SERVICE,\n      optional: true,\n    }, RequestContextService],\n  }],\n})\nexport class DataPaasModule {\n  configure(_consumer: MiddlewareConsumer) {}\n  /**\n   * 注册数据平台模块（同步方式）\n   * @param options 模块配置选项\n   * @returns 动态模块\n   */\n  static forRoot(options: DataPaasConfig): DynamicModule {\n    let roleSchema = '';\n    let sslModeRequired = false;\n    if (options.connectionString) {\n      try {\n        const url = new URL(options.connectionString);\n\n        roleSchema = url.searchParams.get(\n          'schema'\n        ) as string;\n        sslModeRequired = url.searchParams.get('sslmode') === \"require\";\n      } catch (err) {\n        roleSchema = '';\n      }\n    }\n    const config: DataPaasConfig = {\n      connectionString: options.connectionString,\n      schema: options.schema,\n      logger: options.logger || false,\n      timeout: options.timeout || 10000,\n      maxConnections: options.maxConnections || 1,\n      idleTimeout: options.idleTimeout || 20,\n      connectionTimeout: options.connectionTimeout || 10,\n      ssl: options.ssl || (sslModeRequired ? 'require' : false),\n      autoContext: options.autoContext == null ? true : !!options.autoContext,\n      roleSchema,\n    };\n\n    const providers: Provider[] = [\n      // 配置提供者 - 使用 useValue\n      {\n        provide: DATAPAAS_CONFIG,\n        useValue: config,\n      },\n      // 数据库管理器 - 使用 useClass，依赖注入连接服务和配置服务\n      {\n        provide: DrizzleDatabaseManager,\n        useClass: DrizzleDatabaseManager,\n      },\n      // 数据库服务 - 使用 useClass，依赖注入管理器\n      {\n        provide: DataPaasDatabaseService,\n        useClass: DataPaasDatabaseService,\n      },\n      {\n        provide: SqlExecutionContextMiddleware,\n        useClass: SqlExecutionContextMiddleware,\n      },\n      // Drizzle 数据库实例 - 使用 useFactory 获取管理器中的数据库实例\n      {\n        provide: DRIZZLE_DATABASE,\n        useFactory: async (manager: DrizzleDatabaseManager) => {\n          await manager.initialize();\n          return manager.getDatabase();\n        },\n        inject: [DrizzleDatabaseManager],\n      },\n    ];\n\n    return {\n      module: DataPaasModule,\n      providers,\n      exports: [\n        DrizzleDatabaseManager,\n        DataPaasDatabaseService,\n        SqlExecutionContextMiddleware,\n        DRIZZLE_DATABASE,\n        DATAPAAS_CONFIG,\n      ],\n      global: true,\n    };\n  }\n\n  /**\n   * 注册数据平台模块（异步方式）\n   * 支持从 ConfigModule 等异步源获取配置\n   * @param options 异步配置选项\n   * @returns 动态模块\n   * @example\n   * ```typescript\n   * DataPaasModule.forRootAsync({\n   *   imports: [ConfigModule],\n   *   useFactory: async (configService: ConfigService) => ({\n   *     connectionString: configService.get('DATABASE_URL'),\n   *   }),\n   *   inject: [ConfigService],\n   * })\n   * ```\n   */\n  static forRootAsync(options: DataPaasModuleAsyncOptions): DynamicModule {\n    const asyncProviders = this.createAsyncProviders(options);\n\n    const providers: Provider[] = [\n      ...asyncProviders,\n      // 数据库管理器 - 使用 useClass，依赖注入连接服务和配置服务\n      {\n        provide: DrizzleDatabaseManager,\n        useClass: DrizzleDatabaseManager,\n      },\n      // 数据库服务 - 使用 useClass，依赖注入管理器\n      {\n        provide: DataPaasDatabaseService,\n        useClass: DataPaasDatabaseService,\n      },\n      {\n        provide: SqlExecutionContextMiddleware,\n        useClass: SqlExecutionContextMiddleware,\n      },\n      // Drizzle 数据库实例 - 使用 useFactory 获取管理器中的数据库实例\n      {\n        provide: DRIZZLE_DATABASE,\n        useFactory: async (manager: DrizzleDatabaseManager) => {\n          await manager.initialize();\n          return manager.getDatabase();\n        },\n        inject: [DrizzleDatabaseManager],\n      },\n    ];\n\n    return {\n      module: DataPaasModule,\n      imports: options.imports || [],\n      providers,\n      exports: [\n        DrizzleDatabaseManager,\n        DataPaasDatabaseService,\n        SqlExecutionContextMiddleware,\n        DRIZZLE_DATABASE,\n        DATAPAAS_CONFIG,\n      ],\n      global: true,\n    };\n  }\n\n  /**\n   * 创建异步配置 providers\n   */\n  private static createAsyncProviders(\n    options: DataPaasModuleAsyncOptions\n  ): Provider[] {\n    if (options.useFactory) {\n      return [this.createAsyncConfigProvider(options)];\n    }\n\n    throw new Error('DataPaasModule.forRootAsync 需要指定 useFactory');\n  }\n\n  /**\n   * 创建异步配置 provider\n   */\n  private static createAsyncConfigProvider(\n    options: DataPaasModuleAsyncOptions\n  ): Provider {\n    if (options.useFactory) {\n      return {\n        provide: DATAPAAS_CONFIG,\n        useFactory: async (...args: unknown[]) => {\n          const config = await options.useFactory!(...args);\n          return this.normalizeConfig(config);\n        },\n        inject: options.inject || [],\n      };\n    }\n\n    throw new Error('Invalid async options');\n  }\n\n  /**\n   * 标准化配置，设置默认值\n   */\n  private static normalizeConfig(config: DataPaasConfig): DataPaasConfig {\n    let roleSchema = '';\n    let sslModeRequired = false;\n    if (config.connectionString) {\n      try {\n        const url = new URL(config.connectionString);\n        roleSchema = url.searchParams.get(\n          'schema'\n        ) as string;\n        sslModeRequired = url.searchParams.get('sslmode') === \"require\";\n      } catch (err) {\n        roleSchema = '';\n      }\n    }\n    return {\n      connectionString: config.connectionString,\n      schema: config.schema,\n      logger: config.logger || false,\n      timeout: config.timeout || 10000,\n      maxConnections: config.maxConnections || 1,\n      idleTimeout: config.idleTimeout || 20,\n      connectionTimeout: config.connectionTimeout || 10,\n      ssl: config.ssl || (sslModeRequired ? 'require' : false),\n      autoContext: config.autoContext == null ? true : !!config.autoContext,\n      roleSchema,\n    };\n  }\n}\n","import { Injectable, OnModuleDestroy } from '@nestjs/common';\nimport { type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { DrizzleDatabaseManager } from './drizzle-manager';\nimport { DatabaseConnectionInfo } from '../types';\n\n@Injectable()\nexport class DataPaasDatabaseService implements OnModuleDestroy {\n  constructor(\n    private readonly databaseManager: DrizzleDatabaseManager,\n  ) {}\n\n  /**\n   * 获取 Drizzle 数据库实例\n   * @returns Drizzle 数据库实例\n   */\n  getDatabase(): PostgresJsDatabase {\n    return this.databaseManager.getDatabase();\n  }\n\n  /**\n   * 获取数据库管理器\n   * @returns 数据库管理器实例\n   */\n  getDatabaseManager(): DrizzleDatabaseManager {\n    return this.databaseManager;\n  }\n\n\n  /**\n   * 测试数据库连接\n   * @returns 连接是否正常\n   */\n  async testConnection(): Promise<boolean> {\n    return this.databaseManager.testConnection();\n  }\n\n  /**\n   * 重新连接数据库\n   * @param forceRefresh 是否强制刷新连接信息\n   * @returns 新的数据库实例\n   */\n  async reconnect(forceRefresh: boolean = false): Promise<PostgresJsDatabase> {\n    return this.databaseManager.reconnect(forceRefresh);\n  }\n\n  /**\n   * 获取当前连接信息\n   * @returns 当前连接信息\n   */\n  getConnectionInfo(): DatabaseConnectionInfo | null {\n    return this.databaseManager.getConnectionInfo();\n  }\n\n  /**\n   * 检查连接状态\n   * @returns 是否已连接\n   */\n  isConnected(): boolean {\n    return this.databaseManager.isDatabaseConnected();\n  }\n\n  /**\n   * 开始事务\n   * @param callback 事务回调函数\n   * @returns 事务结果\n   */\n  async transaction<T>(callback: (tx: PostgresJsDatabase) => Promise<T>): Promise<T> {\n    return this.databaseManager.transaction(callback);\n  }\n\n  /**\n   * 获取连接池状态\n   * @returns 连接池状态信息\n   */\n  getConnectionPoolStatus(): { isConnected: boolean; connectionInfo: DatabaseConnectionInfo | null } {\n    return this.databaseManager.getConnectionPoolStatus();\n  }\n\n\n  /**\n   * 模块销毁时断开数据库连接\n   */\n  async onModuleDestroy(): Promise<void> {\n    await this.databaseManager.disconnect();\n  }\n}\n","import { Injectable, NestMiddleware, Inject } from '@nestjs/common';\nimport { Request, Response, NextFunction } from 'express';\nimport * as session from 'drizzle-orm/postgres-js';\nimport { DATAPAAS_CONFIG } from './const';\nimport { type DataPaasConfig } from '../types';\n\ninterface UserContext {\n  userId?: string;\n  tenantId?: number;\n  appId?: string;\n  // 用户角色（如果有）\n  userRoles?: string[];\n  // 是否为系统账号\n  isSystemAccount: boolean;\n  // 用户类型\n  userType?: string;\n}\n\n@Injectable()\nexport class SqlExecutionContextMiddleware implements NestMiddleware {\n  constructor(\n    @Inject(DATAPAAS_CONFIG)\n    private readonly config: DataPaasConfig,\n  ) {}\n\n  use(req: Request, _res: Response, next: NextFunction) {\n    const roleSchema = this.config.roleSchema ?? '';\n    const userContext =\n      (req as unknown as { userContext: UserContext }).userContext || {};\n    const roles = (userContext.userRoles || []).join(',');\n    const userId = userContext.userId;\n    const isSystemAccount = userContext.isSystemAccount;\n    const userType = userContext.userType;\n    const sqls = [\n      // 设置用户 ID, 匿名用户时设置为空字符串\n      userId\n        ? `SET LOCAL app.user_id = '${userId}'`\n        : `SET LOCAL app.user_id = ''`,\n      // 系统用户设置为service_role, 登录用户设置为 authenticated_xxx 角色，匿名用户设置为 anon_xxx 角色\n      isSystemAccount ? `SET LOCAL ROLE 'service_role_${roleSchema}'` :  (\n        userId\n          ? `SET LOCAL ROLE 'authenticated_${roleSchema}'`\n          : `SET LOCAL ROLE 'anon_${roleSchema}'`),\n      // 自定义角色\n      roles && `SET LOCAL app.role_ids = '${roles}'`,\n      // 用户类型\n      userType\n        ? `SET LOCAL app.user_type = '${userType}'`\n        : `SET LOCAL app.user_type = ''`,\n    ]\n      .filter(Boolean)\n      .join(';');\n\n    // 需要确认角色设置是否对后续 SQL 执行有影响，预期是只影响当前会话\n    session.runWithAuthContext({ preSql: sqls }, () => {\n      next();\n    });\n  }\n}\n","import { type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { sql } from 'drizzle-orm';\n\nimport { DataPaasDatabaseService } from './database.service';\n\n/**\n * 数据库工具函数集合\n */\nexport class DatabaseUtils {\n  /**\n   * 安全执行数据库操作\n   * @param databaseService 数据库服务实例\n   * @param operation 数据库操作函数\n   * @param fallback 失败时的回退值\n   * @returns 操作结果或回退值\n   */\n  static async safeExecute<T>(\n    databaseService: DataPaasDatabaseService,\n    operation: (db: PostgresJsDatabase) => Promise<T>,\n    fallback?: T\n  ): Promise<T | undefined> {\n    try {\n      if (!databaseService.isConnected()) {\n        throw new Error('Database not connected');\n      }\n      \n      const db = databaseService.getDatabase();\n      return await operation(db);\n    } catch (error) {\n      console.error('Database operation failed:', error);\n      return fallback;\n    }\n  }\n\n  /**\n   * 批量执行数据库操作\n   * @param databaseService 数据库服务实例\n   * @param operations 操作数组\n   * @param batchSize 批处理大小\n   * @returns 所有操作的结果\n   */\n  static async batchExecute<T>(\n    databaseService: DataPaasDatabaseService,\n    operations: Array<(db: PostgresJsDatabase) => Promise<T>>,\n    batchSize: number = 10\n  ): Promise<T[]> {\n    const results: T[] = [];\n    const db = databaseService.getDatabase();\n\n    for (let i = 0; i < operations.length; i += batchSize) {\n      const batch = operations.slice(i, i + batchSize);\n      const batchResults = await Promise.all(\n        batch.map(operation => operation(db))\n      );\n      results.push(...batchResults);\n    }\n\n    return results;\n  }\n\n  /**\n   * 执行带重试的数据库操作\n   * @param databaseService 数据库服务实例\n   * @param operation 数据库操作函数\n   * @param maxRetries 最大重试次数\n   * @param retryDelay 重试延迟（毫秒）\n   * @returns 操作结果\n   */\n  static async executeWithRetry<T>(\n    databaseService: DataPaasDatabaseService,\n    operation: (db: PostgresJsDatabase) => Promise<T>,\n    maxRetries: number = 3,\n    retryDelay: number = 1000\n  ): Promise<T> {\n    let lastError: Error;\n\n    for (let attempt = 0; attempt <= maxRetries; attempt++) {\n      try {\n        const db = databaseService.getDatabase();\n        return await operation(db);\n      } catch (error) {\n        lastError = error as Error;\n        \n        if (attempt === maxRetries) {\n          break;\n        }\n\n        console.warn(`Database operation failed (attempt ${attempt + 1}/${maxRetries + 1}):`, error);\n        \n        // 等待后重试\n        await new Promise(resolve => setTimeout(resolve, retryDelay * (attempt + 1)));\n      }\n    }\n\n    throw lastError!;\n  }\n\n  /**\n   * 检查表是否存在\n   * @param databaseService 数据库服务实例\n   * @param tableName 表名\n   * @returns 表是否存在\n   */\n  static async tableExists(\n    db: PostgresJsDatabase,\n    tableName: string\n  ): Promise<boolean> {\n    try {\n      const result = await db.execute(\n        sql`SELECT EXISTS (\n          SELECT FROM information_schema.tables \n          WHERE table_schema = 'public' \n          AND table_name = ${tableName}\n        )`\n      );\n      \n      return result[0]?.exists as boolean || false;\n    } catch (error) {\n      console.error('Failed to check table existence:', error);\n      return false;\n    }\n  }\n\n  /**\n   * 获取表结构信息\n   * @param databaseService 数据库服务实例\n   * @param tableName 表名\n   * @returns 表结构信息\n   */\n  static async getTableStructure(\n    db: PostgresJsDatabase,\n    tableName: string\n  ): Promise<any[]> {\n    try {\n      const result = await db.execute(\n        sql`SELECT \n          column_name,\n          data_type,\n          is_nullable,\n          column_default,\n          character_maximum_length\n        FROM information_schema.columns \n        WHERE table_schema = 'public' \n        AND table_name = ${tableName}\n        ORDER BY ordinal_position`,\n      );\n      \n      return result;\n    } catch (error) {\n      console.error('Failed to get table structure:', error);\n      return [];\n    }\n  }\n\n  /**\n   * 获取数据库版本信息\n   * @param databaseService 数据库服务实例\n   * @returns 数据库版本\n   */\n  static async getDatabaseVersion(\n    db: PostgresJsDatabase\n  ): Promise<string> {\n    try {\n      const result = await db.execute(sql`SELECT version()`);\n      return result[0]?.version as string || 'Unknown';\n    } catch (error) {\n      console.error('Failed to get database version:', error);\n      return 'Unknown';\n    }\n  }\n\n  /**\n   * 获取连接统计信息\n   * @param databaseService 数据库服务实例\n   * @returns 连接统计信息\n   */\n  static async getConnectionStats(\n    db: PostgresJsDatabase\n  ): Promise<any> {\n    try {\n      const result = await db.execute(sql`\n        SELECT \n          count(*) as total_connections,\n          count(*) FILTER (WHERE state = 'active') as active_connections,\n          count(*) FILTER (WHERE state = 'idle') as idle_connections\n        FROM pg_stat_activity \n        WHERE datname = current_database()\n      `);\n      \n      return result[0] || {};\n    } catch (error) {\n      console.error('Failed to get connection stats:', error);\n      return {};\n    }\n  }\n\n  /**\n   * 健康检查\n   * @param databaseService 数据库服务实例\n   * @returns 健康状态\n   */\n  static async healthCheck(\n    databaseService: DataPaasDatabaseService\n  ): Promise<{\n    isHealthy: boolean;\n    isConnected: boolean;\n    version?: string;\n    connectionInfo?: any;\n    error?: string;\n  }> {\n    try {\n      const isConnected = databaseService.isConnected();\n      const connectionInfo = databaseService.getConnectionInfo();\n\n      const db = databaseService.getDatabase();\n      if (!isConnected) {\n        return {\n          isHealthy: false,\n          isConnected: false,\n          error: 'Database not connected'\n        };\n      }\n\n      const version = await this.getDatabaseVersion(db);\n      \n      return {\n        isHealthy: true,\n        isConnected: true,\n        version,\n        connectionInfo\n      };\n    } catch (error) {\n      return {\n        isHealthy: false,\n        isConnected: false,\n        error: error instanceof Error ? error.message : 'Unknown error'\n      };\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;;;;;;;;;;;;;;ACAA,IAAAA,iBAAkE;AAClE,yBAAiD;AACjD,yBAAoB;AACpB,sBAAqB;;;ACHd,IAAMC,mBAAmB;AACzB,IAAMC,kBAAkB;AACxB,IAAMC,0BAA0BC,uBAAO,yBAAA;;;ADG9C,+BAA2B;AAC3B,2BAAsC;;;AEK/B,SAASC,oBAAoBC,kBAA0BC,QAAyB;AACrF,QAAM,EAAEC,gBAAgBC,aAAaC,mBAAmBC,KAAKC,OAAM,IAAKL,UAAU,CAAC;AAEnF,MAAIM;AACJ,MAAI;AACFA,0BAAsB,IAAIC,IAAIR,gBAAAA;AAC9B,UAAMS,OAAO,IAAID,IAAIR,gBAAAA;AACrBS,SAAKC,aAAaC,OAAO,QAAA;AACzBX,uBAAmBS,KAAKG,SAAQ;EAClC,SAASC,OAAO;AACd,UAAM,IAAIC,MAAM,iCAAA;EAClB;AACA,QAAM,EAAEC,UAAUC,UAAUC,UAAUC,MAAMC,SAAQ,IAAKZ;AACzD,QAAMa,mBAAmB;IACvBL,UAAUM,mBAAmBN,QAAAA;IAC7BC,UAAUK,mBAAmBL,QAAAA;IAC7BM,MAAM;MAACL;;IACPC,MAAM;MAACK,SAASL,IAAAA,KAAS;;IACzBC,UAAUE,mBAAmBF,QAAAA;IAC7BK,KAAKtB,kBAAkB;IACvBuB,cAActB,eAAe;IAC7BuB,iBAAiBtB,qBAAqB;IACtCC;IACAsB,UAAUrB,SAASsB,QAAQC,MAAMC;EACnC;AACA,SAAO;IAAE9B;IAAkBoB;EAAiB;AAC9C;AA1BgBrB;;;ACVhB,oBAAuB;AAIvB,IAAMgC,aAAa,IAAIC,qBAAO,UAAA;AAKvB,SAASC,iBAAiBC,MAAmBC,YAAwBC,gBAAqC;AAC/G,QAAMC,YAAY,wBAAkDC,UAAAA;AAClE,QAAIA,SAAS,OAAOA,MAAMC,SAAS,YAAY;AAC7C,aAAO,IAAIC,MAAMF,OAAO;QACtBG,IAAIC,QAAQC,MAAMC,UAAQ;AACxB,cAAID,SAAS,QAAQ;AACnB,mBAAO,OAAOE,aAAuDC,eAAAA;AACnE,oBAAMC,SAAUT,MAA6CU,mBAAmBC,QAASX,MAA2CU,QAAQ,CAAA,IAAK;AACjJ,oBAAME,OAAOf,WAAWgB,WAAWJ,QAAQX,gBAAgBgB,WAAAA,GAAcC,eAAAA;AACzE,oBAAMC,cAAc;gBAClBC,SAASL,KAAKI,YAAW,EAAGC;gBAC5BC,QAAQN,KAAKI,YAAW,EAAGE;cAC7B;AACA,oBAAMC,aAAa;gBACjBC,aAAa;gBACbC,wBAAwB;gBACxBC,0BAA0BN;cAC5B;AACAJ,mBAAKW,aAAa,UAAU,MAAA;AAC5BX,mBAAKW,aAAa,eAAe,UAAA;AAEjC,oBAAMC,QAAQC,KAAKC,IAAG;AACtB,kBAAI;AACF,sBAAMC,SAAS,MAAMvB;AACrB,sBAAMwB,eAAeD,QAAQE,YAAY;AAGzC,oBAAI,CAACD,cAAc;AACjB,wBAAME,WAAWL,KAAKC,IAAG,IAAKF;AAE9B,wBAAMO,eAAeC,KAAKC,UAAU;oBAClCC,WAAWP,QAAQE;oBACnBjC,KAAK+B,QAAQQ,WAAWC;oBACxBC,aAAaP;oBACbQ,cAAcX,OAAOY,UAAU;oBAC/BC,eAAe;kBACjB,CAAA;AAEA/C,6BAAWgD,IAAIV,cAAcZ,UAAAA;gBAC/B,OAAO;AACLtB,6BAAW6C,UAAU9B,IAAAA;gBACvB;AAEAA,qBAAK+B,IAAG;AACR,uBAAOpC,cAAcA,YAAYoB,MAAAA,IAAUA;cAC7C,SAASiB,OAAO;AACd,sBAAMd,WAAWL,KAAKC,IAAG,IAAKF;AAC9B,sBAAMf,UAAUT,MAA6CU,mBAAmBC,QAASX,MAA2CU,QAAQ,CAAA,IAAK;AACjJ,sBAAMqB,eAAeC,KAAKC,UAAU;kBAClCC,WAAW;kBACXtC,KAAKa;kBACL4B,aAAaP;kBACbQ,cAAc;kBACdE,eAAeI,iBAAiBC,QAAQD,MAAME,UAAUC,OAAOH,KAAAA;gBACjE,CAAA;AACAnD,2BAAWgD,IAAIV,cAAcZ,UAAAA;AAE7BP,qBAAK+B,IAAG;AACR,uBAAOnC,aAAaA,WAAWoC,KAAAA,IAASI,QAAQC,OAAOL,KAAAA;cACzD;YACF;UACF;AACA,gBAAMM,QAAQC,QAAQhD,IAAIC,QAAQC,MAAMC,QAAAA;AACxC,iBAAO,OAAO4C,UAAU,aAAaA,MAAME,KAAK9C,QAAAA,IAAY4C;QAC9D;MACF,CAAA;IACF;AACA,WAAOlD;EACT,GAnEkB;AAqElB,SAAO,IAAIE,MAAMN,MAAK;IACpByD,MAAMjD,QAAQkD,SAASC,UAAQ;AAC7B,YAAMvD,QAAQmD,QAAQE,MAAMjD,QAAQkD,SAASC,QAAAA;AAC7C,aAAOxD,UAAUC,KAAAA;IACnB;IACAG,IAAIC,QAAQC,MAAMC,UAAQ;AACxB,YAAM4C,QAAQC,QAAQhD,IAAIC,QAAQC,MAAMC,QAAAA;AACxC,UAAID,SAAS,WAAW,OAAO6C,UAAU,YAAY;AACnD,eAAO,CAAIM,KAAiEC,OAAAA;AAC1E,cAAI,OAAOD,QAAQ,YAAY;AAC7B,mBAAON,MAAMQ,KAAKtD,QAAQ,CAACuD,OAAgCH,IAAI7D,iBAAiBgE,IAA+B9D,YAAYC,cAAAA,CAAAA,CAAAA;UAC7H;AACA,cAAI,OAAO2D,OAAO,YAAY;AAC5B,mBAAOP,MAAMQ,KAAKtD,QAAQoD,KAAK,CAACG,OAAgCF,GAAG9D,iBAAiBgE,IAA+B9D,YAAYC,cAAAA,CAAAA,CAAAA;UACjI;AACA,iBAAOoD,MAAMQ,KAAKtD,QAAQoD,GAAAA;QAC5B;MACF;AACA,UAAInD,SAAS,YAAY,OAAO6C,UAAU,YAAY;AACpD,eAAO,CAAmEU,UAAkBC,MAAkBC,YAAAA;AAC5G,gBAAM9D,QAAQkD,MAAMQ,KAAKtD,QAAQwD,UAAUC,MAAMC,OAAAA;AACjD,iBAAO/D,UAAUC,KAAAA;QACnB;MACF;AACA,aAAO,OAAOkD,UAAU,aAAaA,MAAME,KAAK9C,QAAAA,IAAY4C;IAC9D;EACF,CAAA;AACF;AAjGgBvD;;;;;;;;;;;;;;;;;;;;AHET,IAAMoE,yBAAN,MAAMA;SAAAA;;;;;;EACHC,KAAgC;EAChCC,wBAAuD;EACvDC,cAAuB;EACvBC,uBAAgC;EACxC,YAEmBC,QAEAC,YACAC,gBACjB;SAJiBF,SAAAA;SAEAC,aAAAA;SACAC,iBAAAA;AAEjB,SAAKH,uBAAuB,CAAC,CAACI,QAAQC,IAAIC;EAC5C;EAEA,MAAMC,eAAe;AACnB,UAAM,KAAKC,WAAU;EACvB;EAEA,MAAMC,kBAAkB;AACtB,UAAM,KAAKC,WAAU;EACvB;;;;;EAMA,MAAMF,aAA0C;AAC9C,QAAI,KAAKR,sBAAsB;AAC7B,WAAKD,cAAc;AACnB,WAAKF,KAAK;QAACc,MAAM;MAAsB;AACvC,aAAO,KAAKd;IACd;AACA,QAAI;AACF,YAAM,EAAEe,kBAAkBC,iBAAgB,IAAKC,oBAAoB,KAAKb,OAAOW,kBAAkB,KAAKX,MAAM;AAC5G,YAAMc,cAAUC,gBAAAA,SAASJ,kBAAkBC,gBAAAA;AAC3C,YAAMI,WAAWC,iBAAiBH,SAAS,KAAKb,YAAY,KAAKC,cAAc;AAE/E,WAAKN,SAAKsB,4BAAQF,UAAU;QAAEG,QAAQ,KAAKnB,OAAOmB;QAAQC,QAAQ;MAAM,CAAA;AACxE,WAAKvB,wBAAwB;QAC3Bc;QACAC,kBAAkB;UAChBS,KAAKT,iBAAiBS;UACtBC,mBAAmBV,iBAAiBW;UACpCC,aAAaZ,iBAAiBa;UAC9BC,gBAAgBd,iBAAiBe;QACnC;MACF;AACA,WAAK7B,cAAc;AACnB,aAAO,KAAKF;IACd,SAASgC,OAAO;AACd,WAAK9B,cAAc;AACnB,YAAM,IAAI+B,MAAM,6CAA6CD,iBAAiBC,QAAQD,MAAME,UAAU,eAAA,EAAiB;IACzH;EACF;;;;;EAMAC,cAAkC;AAChC,QAAI,CAAC,KAAKnC,MAAM,CAAC,KAAKE,aAAa;AACjC,YAAM,IAAI+B,MAAM,oDAAA;IAClB;AACA,WAAO,KAAKjC;EACd;;;;;EAMA,MAAMoC,iBAAmC;AACvC,QAAI,CAAC,KAAKpC,IAAI;AACZ,aAAO;IACT;AAEA,QAAI;AACF,YAAM,KAAKA,GAAGqC,QAAQC,gCAAa;AACnC,aAAO;IACT,SAASN,OAAO;AACdO,cAAQP,MAAM,oCAAoCA,KAAAA;AAClD,WAAK9B,cAAc;AACnB,aAAO;IACT;EACF;;;;;;EAOA,MAAMsC,UAAUC,gBAAyB,OAAoC;AAC3E,UAAM,KAAK5B,WAAU;AACrB,WAAO,KAAKF,WAAU;EACxB;;;;EAKA,MAAME,aAA4B;AAChC,QAAI,KAAKb,IAAI;AACX,UAAI;AAEF,aAAKA,KAAK;AACV,aAAKE,cAAc;AACnB,aAAKD,wBAAwB;MAC/B,SAAS+B,OAAO;AACdO,gBAAQP,MAAM,sCAAsCA,KAAAA;MACtD;IACF;EACF;;;;;EAMAU,oBAAmD;AACjD,WAAO,KAAKzC;EACd;;;;;EAMA0C,sBAA+B;AAC7B,WAAO,KAAKzC;EACd;;;;;;EAQA,MAAM0C,YAAeC,UAA8D;AACjF,UAAM7C,KAAK,KAAKmC,YAAW;AAC3B,WAAOnC,GAAG4C,YAAYC,QAAAA;EACxB;;;;;EAMAC,0BAAmG;AACjG,WAAO;MACL5C,aAAa,KAAKA;MAClB6C,gBAAgB,KAAK9C;IACvB;EACF;AACF;;;;;;;;;;;;;;AIjKA,IAAA+C,iBAOO;;;ACPP,IAAAC,iBAA4C;;;;;;;;;;;;AAMrC,IAAMC,0BAAN,MAAMA;SAAAA;;;;EACX,YACmBC,iBACjB;SADiBA,kBAAAA;EAChB;;;;;EAMHC,cAAkC;AAChC,WAAO,KAAKD,gBAAgBC,YAAW;EACzC;;;;;EAMAC,qBAA6C;AAC3C,WAAO,KAAKF;EACd;;;;;EAOA,MAAMG,iBAAmC;AACvC,WAAO,KAAKH,gBAAgBG,eAAc;EAC5C;;;;;;EAOA,MAAMC,UAAUC,eAAwB,OAAoC;AAC1E,WAAO,KAAKL,gBAAgBI,UAAUC,YAAAA;EACxC;;;;;EAMAC,oBAAmD;AACjD,WAAO,KAAKN,gBAAgBM,kBAAiB;EAC/C;;;;;EAMAC,cAAuB;AACrB,WAAO,KAAKP,gBAAgBQ,oBAAmB;EACjD;;;;;;EAOA,MAAMC,YAAeC,UAA8D;AACjF,WAAO,KAAKV,gBAAgBS,YAAYC,QAAAA;EAC1C;;;;;EAMAC,0BAAmG;AACjG,WAAO,KAAKX,gBAAgBW,wBAAuB;EACrD;;;;EAMA,MAAMC,kBAAiC;AACrC,UAAM,KAAKZ,gBAAgBa,WAAU;EACvC;AACF;;;;;;;;;;ACrFA,IAAAC,iBAAmD;AAEnD,cAAyB;;;;;;;;;;;;;;;;;;AAiBlB,IAAMC,gCAAN,MAAMA;SAAAA;;;;EACX,YAEmBC,QACjB;SADiBA,SAAAA;EAChB;EAEHC,IAAIC,KAAcC,MAAgBC,MAAoB;AACpD,UAAMC,aAAa,KAAKL,OAAOK,cAAc;AAC7C,UAAMC,cACHJ,IAAgDI,eAAe,CAAC;AACnE,UAAMC,SAASD,YAAYE,aAAa,CAAA,GAAIC,KAAK,GAAA;AACjD,UAAMC,SAASJ,YAAYI;AAC3B,UAAMC,kBAAkBL,YAAYK;AACpC,UAAMC,WAAWN,YAAYM;AAC7B,UAAMC,OAAO;;MAEXH,SACI,4BAA4BA,MAAAA,MAC5B;;MAEJC,kBAAkB,gCAAgCN,UAAAA,MAChDK,SACI,iCAAiCL,UAAAA,MACjC,wBAAwBA,UAAAA;;MAE9BE,SAAS,6BAA6BA,KAAAA;;MAEtCK,WACI,8BAA8BA,QAAAA,MAC9B;MAEHE,OAAOC,OAAAA,EACPN,KAAK,GAAA;AAGRO,IAAQC,2BAAmB;MAAEC,QAAQL;IAAK,GAAG,MAAA;AAC3CT,WAAAA;IACF,CAAA;EACF;AACF;;;;;;;;;;;AF5CA,IAAAe,wBAA6E;AAC7E,IAAAC,4BAA2B;;;;;;;;AAsB3B,SAASC,uBAAAA;AACP,MAAI;AAKF,UAAMC,YAAWC,QAAQ,yBAAA;AACzB,UAAMC,KACJF,aACA,OAAOA,UAASG,wBAAwB,cACxC,OAAOH,UAASI,uBAAuB;AACzC,QAAI,CAACF,IAAI;AACP,YAAM,IAAIG,MACR,0GAAA;IAEJ;EACF,SAASC,KAAK;AACZ,UAAMC,SAASD,eAAeD,QAAQC,IAAIE,UAAUC,OAAOH,GAAAA;AAC3D,UAAM,IAAID,MAAM,uFAA2BE,MAAAA,EAAQ;EACrD;AACF;AApBSR;AAuBTA,qBAAAA;AAcO,IAAMW,iBAAN,MAAMA,gBAAAA;SAAAA;;;EACXC,UAAUC,WAA+B;EAAC;;;;;;EAM1C,OAAOC,QAAQC,SAAwC;AACrD,QAAIC,aAAa;AACjB,QAAIC,kBAAkB;AACtB,QAAIF,QAAQG,kBAAkB;AAC5B,UAAI;AACF,cAAMC,MAAM,IAAIC,IAAIL,QAAQG,gBAAgB;AAE5CF,qBAAaG,IAAIE,aAAaC,IAC5B,QAAA;AAEFL,0BAAkBE,IAAIE,aAAaC,IAAI,SAAA,MAAe;MACxD,SAASf,KAAK;AACZS,qBAAa;MACf;IACF;AACA,UAAMO,SAAyB;MAC7BL,kBAAkBH,QAAQG;MAC1BM,QAAQT,QAAQS;MAChBC,QAAQV,QAAQU,UAAU;MAC1BC,SAASX,QAAQW,WAAW;MAC5BC,gBAAgBZ,QAAQY,kBAAkB;MAC1CC,aAAab,QAAQa,eAAe;MACpCC,mBAAmBd,QAAQc,qBAAqB;MAChDC,KAAKf,QAAQe,QAAQb,kBAAkB,YAAY;MACnDc,aAAahB,QAAQgB,eAAe,OAAO,OAAO,CAAC,CAAChB,QAAQgB;MAC5Df;IACF;AAEA,UAAMgB,YAAwB;;MAE5B;QACEC,SAASC;QACTC,UAAUZ;MACZ;;MAEA;QACEU,SAASG;QACTC,UAAUD;MACZ;;MAEA;QACEH,SAASK;QACTD,UAAUC;MACZ;MACA;QACEL,SAASM;QACTF,UAAUE;MACZ;;MAEA;QACEN,SAASO;QACTC,YAAY,8BAAOC,YAAAA;AACjB,gBAAMA,QAAQC,WAAU;AACxB,iBAAOD,QAAQE,YAAW;QAC5B,GAHY;QAIZC,QAAQ;UAACT;;MACX;;AAGF,WAAO;MACLU,QAAQnC;MACRqB;MACAe,SAAS;QACPX;QACAE;QACAC;QACAC;QACAN;;MAEFc,QAAQ;IACV;EACF;;;;;;;;;;;;;;;;;EAkBA,OAAOC,aAAalC,SAAoD;AACtE,UAAMmC,iBAAiB,KAAKC,qBAAqBpC,OAAAA;AAEjD,UAAMiB,YAAwB;SACzBkB;;MAEH;QACEjB,SAASG;QACTC,UAAUD;MACZ;;MAEA;QACEH,SAASK;QACTD,UAAUC;MACZ;MACA;QACEL,SAASM;QACTF,UAAUE;MACZ;;MAEA;QACEN,SAASO;QACTC,YAAY,8BAAOC,YAAAA;AACjB,gBAAMA,QAAQC,WAAU;AACxB,iBAAOD,QAAQE,YAAW;QAC5B,GAHY;QAIZC,QAAQ;UAACT;;MACX;;AAGF,WAAO;MACLU,QAAQnC;MACRyC,SAASrC,QAAQqC,WAAW,CAAA;MAC5BpB;MACAe,SAAS;QACPX;QACAE;QACAC;QACAC;QACAN;;MAEFc,QAAQ;IACV;EACF;;;;EAKA,OAAeG,qBACbpC,SACY;AACZ,QAAIA,QAAQ0B,YAAY;AACtB,aAAO;QAAC,KAAKY,0BAA0BtC,OAAAA;;IACzC;AAEA,UAAM,IAAIT,MAAM,iEAAA;EAClB;;;;EAKA,OAAe+C,0BACbtC,SACU;AACV,QAAIA,QAAQ0B,YAAY;AACtB,aAAO;QACLR,SAASC;QACTO,YAAY,iCAAUa,SAAAA;AACpB,gBAAM/B,SAAS,MAAMR,QAAQ0B,WAAU,GAAKa,IAAAA;AAC5C,iBAAO,KAAKC,gBAAgBhC,MAAAA;QAC9B,GAHY;QAIZsB,QAAQ9B,QAAQ8B,UAAU,CAAA;MAC5B;IACF;AAEA,UAAM,IAAIvC,MAAM,uBAAA;EAClB;;;;EAKA,OAAeiD,gBAAgBhC,QAAwC;AACrE,QAAIP,aAAa;AACjB,QAAIC,kBAAkB;AACtB,QAAIM,OAAOL,kBAAkB;AAC3B,UAAI;AACF,cAAMC,MAAM,IAAIC,IAAIG,OAAOL,gBAAgB;AAC3CF,qBAAaG,IAAIE,aAAaC,IAC5B,QAAA;AAEFL,0BAAkBE,IAAIE,aAAaC,IAAI,SAAA,MAAe;MACxD,SAASf,KAAK;AACZS,qBAAa;MACf;IACF;AACA,WAAO;MACLE,kBAAkBK,OAAOL;MACzBM,QAAQD,OAAOC;MACfC,QAAQF,OAAOE,UAAU;MACzBC,SAASH,OAAOG,WAAW;MAC3BC,gBAAgBJ,OAAOI,kBAAkB;MACzCC,aAAaL,OAAOK,eAAe;MACnCC,mBAAmBN,OAAOM,qBAAqB;MAC/CC,KAAKP,OAAOO,QAAQb,kBAAkB,YAAY;MAClDc,aAAaR,OAAOQ,eAAe,OAAO,OAAO,CAAC,CAACR,OAAOQ;MAC1Df;IACF;EACF;AACF;;;IAvNEgB,WAAW;MAAC;QACVC,SAASuB;QACTf,YAAY,wBAACgB,YAAsCC,QAAAA;AACjD,iBAAOD,cAAc,IAAIE,qCAAWD,GAAAA;QACtC,GAFY;QAGZb,QAAQ;UAAC;YACPe,OAAOC;YACPC,UAAU;UACZ;UAAGC;;MACL;;;;;;AGvEF,IAAAC,sBAAoB;AAOb,IAAMC,gBAAN,MAAMA;EAPb,OAOaA;;;;;;;;;;EAQX,aAAaC,YACXC,iBACAC,WACAC,UACwB;AACxB,QAAI;AACF,UAAI,CAACF,gBAAgBG,YAAW,GAAI;AAClC,cAAM,IAAIC,MAAM,wBAAA;MAClB;AAEA,YAAMC,KAAKL,gBAAgBM,YAAW;AACtC,aAAO,MAAML,UAAUI,EAAAA;IACzB,SAASE,OAAO;AACdC,cAAQD,MAAM,8BAA8BA,KAAAA;AAC5C,aAAOL;IACT;EACF;;;;;;;;EASA,aAAaO,aACXT,iBACAU,YACAC,YAAoB,IACN;AACd,UAAMC,UAAe,CAAA;AACrB,UAAMP,KAAKL,gBAAgBM,YAAW;AAEtC,aAASO,IAAI,GAAGA,IAAIH,WAAWI,QAAQD,KAAKF,WAAW;AACrD,YAAMI,QAAQL,WAAWM,MAAMH,GAAGA,IAAIF,SAAAA;AACtC,YAAMM,eAAe,MAAMC,QAAQC,IACjCJ,MAAMK,IAAInB,CAAAA,cAAaA,UAAUI,EAAAA,CAAAA,CAAAA;AAEnCO,cAAQS,KAAI,GAAIJ,YAAAA;IAClB;AAEA,WAAOL;EACT;;;;;;;;;EAUA,aAAaU,iBACXtB,iBACAC,WACAsB,aAAqB,GACrBC,aAAqB,KACT;AACZ,QAAIC;AAEJ,aAASC,UAAU,GAAGA,WAAWH,YAAYG,WAAW;AACtD,UAAI;AACF,cAAMrB,KAAKL,gBAAgBM,YAAW;AACtC,eAAO,MAAML,UAAUI,EAAAA;MACzB,SAASE,OAAO;AACdkB,oBAAYlB;AAEZ,YAAImB,YAAYH,YAAY;AAC1B;QACF;AAEAf,gBAAQmB,KAAK,sCAAsCD,UAAU,CAAA,IAAKH,aAAa,CAAA,MAAOhB,KAAAA;AAGtF,cAAM,IAAIW,QAAQU,CAAAA,YAAWC,WAAWD,SAASJ,cAAcE,UAAU,EAAA,CAAA;MAC3E;IACF;AAEA,UAAMD;EACR;;;;;;;EAQA,aAAaK,YACXzB,IACA0B,WACkB;AAClB,QAAI;AACF,YAAMC,SAAS,MAAM3B,GAAG4B,QACtBC;;;6BAGqBH,SAAAA;UACnB;AAGJ,aAAOC,OAAO,CAAA,GAAIG,UAAqB;IACzC,SAAS5B,OAAO;AACdC,cAAQD,MAAM,oCAAoCA,KAAAA;AAClD,aAAO;IACT;EACF;;;;;;;EAQA,aAAa6B,kBACX/B,IACA0B,WACgB;AAChB,QAAI;AACF,YAAMC,SAAS,MAAM3B,GAAG4B,QACtBC;;;;;;;;2BAQmBH,SAAAA;kCACO;AAG5B,aAAOC;IACT,SAASzB,OAAO;AACdC,cAAQD,MAAM,kCAAkCA,KAAAA;AAChD,aAAO,CAAA;IACT;EACF;;;;;;EAOA,aAAa8B,mBACXhC,IACiB;AACjB,QAAI;AACF,YAAM2B,SAAS,MAAM3B,GAAG4B,QAAQC,yCAAqB;AACrD,aAAOF,OAAO,CAAA,GAAIM,WAAqB;IACzC,SAAS/B,OAAO;AACdC,cAAQD,MAAM,mCAAmCA,KAAAA;AACjD,aAAO;IACT;EACF;;;;;;EAOA,aAAagC,mBACXlC,IACc;AACd,QAAI;AACF,YAAM2B,SAAS,MAAM3B,GAAG4B,QAAQC;;;;;;;OAO/B;AAED,aAAOF,OAAO,CAAA,KAAM,CAAC;IACvB,SAASzB,OAAO;AACdC,cAAQD,MAAM,mCAAmCA,KAAAA;AACjD,aAAO,CAAC;IACV;EACF;;;;;;EAOA,aAAaiC,YACXxC,iBAOC;AACD,QAAI;AACF,YAAMG,cAAcH,gBAAgBG,YAAW;AAC/C,YAAMsC,iBAAiBzC,gBAAgB0C,kBAAiB;AAExD,YAAMrC,KAAKL,gBAAgBM,YAAW;AACtC,UAAI,CAACH,aAAa;AAChB,eAAO;UACLwC,WAAW;UACXxC,aAAa;UACbI,OAAO;QACT;MACF;AAEA,YAAM+B,UAAU,MAAM,KAAKD,mBAAmBhC,EAAAA;AAE9C,aAAO;QACLsC,WAAW;QACXxC,aAAa;QACbmC;QACAG;MACF;IACF,SAASlC,OAAO;AACd,aAAO;QACLoC,WAAW;QACXxC,aAAa;QACbI,OAAOA,iBAAiBH,QAAQG,MAAMqC,UAAU;MAClD;IACF;EACF;AACF;","names":["import_common","DRIZZLE_DATABASE","DATAPAAS_CONFIG","DATAPAAS_LOGGER_SERVICE","Symbol","parseConnectionInfo","connectionString","config","maxConnections","idleTimeout","connectionTimeout","ssl","logger","parsedConfigFromUrl","URL","temp","searchParams","delete","toString","error","Error","username","password","hostname","port","pathname","connectionConfig","decodeURIComponent","host","parseInt","max","idle_timeout","connect_timeout","onnotice","console","log","undefined","nestLogger","Logger","genPostgresProxy","sql","observable","requestContext","wrapQuery","query","then","Proxy","get","target","prop","receiver","onfulfilled","onrejected","sqlStr","strings","Array","span","startTrace","getContext","requestRootSpan","spanContext","traceId","spanId","logContext","source_type","paas_attributes_module","paas_parent_span_context","setAttribute","start","Date","now","result","isSetRoleSql","command","duration","parsedParams","JSON","stringify","operation","statement","string","duration_ms","result_count","length","error_message","log","dropTrace","end","error","Error","message","String","Promise","reject","value","Reflect","bind","apply","thisArg","argArray","arg","cb","call","tx","queryStr","args","options","DrizzleDatabaseManager","db","currentConnectionInfo","isConnected","skipInitDbConnection","config","observable","requestContext","process","env","DEPRECATED_SKIP_INIT_DB_CONNECTION","onModuleInit","initialize","onModuleDestroy","disconnect","name","connectionString","connectionConfig","parseConnectionInfo","baseSql","postgres","sqlProxy","genPostgresProxy","drizzle","schema","logger","ssl","connectionTimeout","connect_timeout","idleTimeout","idle_timeout","maxConnections","max","error","Error","message","getDatabase","testConnection","execute","sql","console","reconnect","_forceRefresh","getConnectionInfo","isDatabaseConnected","transaction","callback","getConnectionPoolStatus","connectionInfo","import_common","import_common","DataPaasDatabaseService","databaseManager","getDatabase","getDatabaseManager","testConnection","reconnect","forceRefresh","getConnectionInfo","isConnected","isDatabaseConnected","transaction","callback","getConnectionPoolStatus","onModuleDestroy","disconnect","import_common","SqlExecutionContextMiddleware","config","use","req","_res","next","roleSchema","userContext","roles","userRoles","join","userId","isSystemAccount","userType","sqls","filter","Boolean","session","runWithAuthContext","preSql","import_nestjs_common","import_nestjs_observable","verifyDrizzlePatched","postgres","require","ok","getAuthContextStore","runWithAuthContext","Error","err","reason","message","String","DataPaasModule","configure","_consumer","forRoot","options","roleSchema","sslModeRequired","connectionString","url","URL","searchParams","get","config","schema","logger","timeout","maxConnections","idleTimeout","connectionTimeout","ssl","autoContext","providers","provide","DATAPAAS_CONFIG","useValue","DrizzleDatabaseManager","useClass","DataPaasDatabaseService","SqlExecutionContextMiddleware","DRIZZLE_DATABASE","useFactory","manager","initialize","getDatabase","inject","module","exports","global","forRootAsync","asyncProviders","createAsyncProviders","imports","createAsyncConfigProvider","args","normalizeConfig","DATAPAAS_LOGGER_SERVICE","observable","rtx","Observable","token","OBSERVABLE_SERVICE","optional","RequestContextService","import_drizzle_orm","DatabaseUtils","safeExecute","databaseService","operation","fallback","isConnected","Error","db","getDatabase","error","console","batchExecute","operations","batchSize","results","i","length","batch","slice","batchResults","Promise","all","map","push","executeWithRetry","maxRetries","retryDelay","lastError","attempt","warn","resolve","setTimeout","tableExists","tableName","result","execute","sql","exists","getTableStructure","getDatabaseVersion","version","getConnectionStats","healthCheck","connectionInfo","getConnectionInfo","isHealthy","message"]}

package/dist/index.js

@@ -70,6 +70,15 @@ function genPostgresProxy(sql3, observable, requestContext) {
             return async (onfulfilled, onrejected) => {
               const sqlStr = query.strings instanceof Array ? query.strings[0] : "";
               const span = observable.startTrace(sqlStr, requestContext?.getContext()?.requestRootSpan);
+              const spanContext = {
+                traceId: span.spanContext().traceId,
+                spanId: span.spanContext().spanId
+              };
+              const logContext = {
+                source_type: "platform",
+                paas_attributes_module: "data",
+                paas_parent_span_context: spanContext
+              };
               span.setAttribute("module", "data");
               span.setAttribute("source_type", "platform");
               const start = Date.now();
@@ -85,11 +94,7 @@ function genPostgresProxy(sql3, observable, requestContext) {
                     result_count: result.length ?? 0,
                     error_message: ""
                   });
-                  nestLogger.log(parsedParams, {
-                    source_type: "platform",
-                    paas_attributes_module: "data",
-                    paas_parent_span: span
-                  });
+                  nestLogger.log(parsedParams, logContext);
                 } else {
                   observable.dropTrace(span);
                 }
@@ -105,11 +110,7 @@ function genPostgresProxy(sql3, observable, requestContext) {
                   result_count: 0,
                   error_message: error instanceof Error ? error.message : String(error)
                 });
-                nestLogger.log(parsedParams, {
-                  source_type: "platform",
-                  paas_attributes_module: "data",
-                  paas_parent_span: span
-                });
+                nestLogger.log(parsedParams, logContext);
                 span.end();
                 return onrejected ? onrejected(error) : Promise.reject(error);
               }

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/database/drizzle-manager.ts","../src/database/const.ts","../src/utils.ts","../src/database/postgres-proxy.ts","../src/database/database.module.ts","../src/database/database.service.ts","../src/database/sql-exection-context.middleware.ts","../src/database/database-utils.ts"],"sourcesContent":["import { Injectable, OnModuleInit, OnModuleDestroy, Inject } from '@nestjs/common';\nimport { drizzle, type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { sql } from 'drizzle-orm';\nimport postgres from 'postgres';\nimport { DATAPAAS_CONFIG, DATAPAAS_LOGGER_SERVICE } from './const';\nimport { Observable } from '@lark-apaas/nestjs-observable';\nimport { RequestContextService } from '@lark-apaas/nestjs-common';\n\nimport { type DataPaasConfig, type DatabaseConnectionInfo } from '../types';\nimport { parseConnectionInfo } from '../utils';\nimport { genPostgresProxy } from \"./postgres-proxy\";\n@Injectable()\nexport class DrizzleDatabaseManager implements OnModuleInit, OnModuleDestroy {\n  private db: PostgresJsDatabase | null = null;\n  private currentConnectionInfo: DatabaseConnectionInfo | null = null;\n  private isConnected: boolean = false;\n  private skipInitDbConnection: boolean = false;\n  constructor(\n    @Inject(DATAPAAS_CONFIG)\n    private readonly config: DataPaasConfig,\n    @Inject(DATAPAAS_LOGGER_SERVICE)\n    private readonly observable: Observable,\n    private readonly requestContext: RequestContextService,\n  ) { \n    this.skipInitDbConnection = !!process.env.DEPRECATED_SKIP_INIT_DB_CONNECTION;\n  }\n\n  async onModuleInit() {\n    await this.initialize();\n  }\n\n  async onModuleDestroy() {\n    await this.disconnect();\n  }\n\n  /**\n   * 初始化数据库连接\n   * @returns Drizzle 数据库实例\n   */\n  async initialize(): Promise<PostgresJsDatabase> {\n    if (this.skipInitDbConnection) {\n      this.isConnected = true;\n      this.db = {name: 'skipInitDbConnection'} as unknown as PostgresJsDatabase;\n      return this.db;\n    }\n    try {\n      const { connectionString, connectionConfig } = parseConnectionInfo(this.config.connectionString, this.config);\n      const baseSql = postgres(connectionString, connectionConfig);\n      const sqlProxy = genPostgresProxy(baseSql, this.observable, this.requestContext);\n      // 创建 Drizzle 实例\n      this.db = drizzle(sqlProxy, { schema: this.config.schema, logger: false });\n      this.currentConnectionInfo = {\n        connectionString,\n        connectionConfig: {\n          ssl: connectionConfig.ssl,\n          connectionTimeout: connectionConfig.connect_timeout,\n          idleTimeout: connectionConfig.idle_timeout,\n          maxConnections: connectionConfig.max,\n        },\n      };\n      this.isConnected = true;\n      return this.db;\n    } catch (error) {\n      this.isConnected = false;\n      throw new Error(`Failed to initialize database connection: ${error instanceof Error ? error.message : 'Unknown error'}`);\n    }\n  }\n\n  /**\n   * 获取数据库实例\n   * @returns Drizzle 数据库实例\n   */\n  getDatabase(): PostgresJsDatabase {\n    if (!this.db || !this.isConnected) {\n      throw new Error('Database not initialized. Call initialize() first.');\n    }\n    return this.db;\n  }\n\n  /**\n   * 测试数据库连接\n   * @returns 连接是否正常\n   */\n  async testConnection(): Promise<boolean> {\n    if (!this.db) {\n      return false;\n    }\n\n    try {\n      await this.db.execute(sql`SELECT 1`);\n      return true;\n    } catch (error) {\n      console.error('Database connection test failed:', error);\n      this.isConnected = false;\n      return false;\n    }\n  }\n\n  /**\n   * 重新连接数据库\n   * @param forceRefresh 是否强制刷新连接信息\n   * @returns 新的数据库实例\n   */\n  async reconnect(_forceRefresh: boolean = false): Promise<PostgresJsDatabase> {\n    await this.disconnect();\n    return this.initialize();\n  }\n\n  /**\n   * 断开数据库连接\n   */\n  async disconnect(): Promise<void> {\n    if (this.db) {\n      try {\n        // postgres-js 会自动处理连接关闭\n        this.db = null;\n        this.isConnected = false;\n        this.currentConnectionInfo = null;\n      } catch (error) {\n        console.error('Error disconnecting from database:', error);\n      }\n    }\n  }\n\n  /**\n   * 获取当前连接信息\n   * @returns 当前连接信息\n   */\n  getConnectionInfo(): DatabaseConnectionInfo | null {\n    return this.currentConnectionInfo;\n  }\n\n  /**\n   * 检查连接状态\n   * @returns 是否已连接\n   */\n  isDatabaseConnected(): boolean {\n    return this.isConnected;\n  }\n\n\n  /**\n   * 开始事务\n   * @param callback 事务回调函数\n   * @returns 事务结果\n   */\n  async transaction<T>(callback: (tx: PostgresJsDatabase) => Promise<T>): Promise<T> {\n    const db = this.getDatabase();\n    return db.transaction(callback);\n  }\n\n  /**\n   * 获取连接池状态\n   * @returns 连接池状态信息\n   */\n  getConnectionPoolStatus(): { isConnected: boolean; connectionInfo: DatabaseConnectionInfo | null } {\n    return {\n      isConnected: this.isConnected,\n      connectionInfo: this.currentConnectionInfo,\n    };\n  }\n}","export const DRIZZLE_DATABASE = 'DRIZZLE_DATABASE';\nexport const DATAPAAS_CONFIG = 'DATAPAAS_CONFIG';\nexport const DATAPAAS_LOGGER_SERVICE = Symbol('DATAPAAS_LOGGER_SERVICE');","import type postgres from 'postgres';\nimport { type Logger as DrizzleLogger } from 'drizzle-orm';\nimport { PostgresOptions } from './types';\n\ninterface ConnectionConfig {\n  maxConnections?: number;\n  idleTimeout?: number;\n  connectionTimeout?: number;\n  ssl?: PostgresOptions['ssl'];\n  logger?: boolean | DrizzleLogger;\n}\nexport function parseConnectionInfo(connectionString: string, config?: ConnectionConfig) {\n  const { maxConnections, idleTimeout, connectionTimeout, ssl, logger } = config || {};\n  // 创建 postgres 连接\n  let parsedConfigFromUrl: URL;\n  try {\n    parsedConfigFromUrl = new URL(connectionString);\n    const temp = new URL(connectionString);\n    temp.searchParams.delete('schema'); // 移除schema参数，避免 postgres 解析错误\n    connectionString = temp.toString();\n  } catch (error) {\n    throw new Error('Failed to parse connection info');\n  }\n  const { username, password, hostname, port, pathname } = parsedConfigFromUrl;\n  const connectionConfig = {\n    username: decodeURIComponent(username),\n    password: decodeURIComponent(password),\n    host: [hostname] as unknown as string,\n    port: [parseInt(port) || 5432] as unknown as number,\n    pathname: decodeURIComponent(pathname),\n    max: maxConnections || 1,\n    idle_timeout: idleTimeout || 20,\n    connect_timeout: connectionTimeout || 10,\n    ssl,\n    onnotice: logger ? console.log : undefined,\n  }\n  return { connectionString, connectionConfig };\n}","import type postgres from 'postgres';\nimport { Logger } from '@nestjs/common';\nimport { Span, Observable } from '@lark-apaas/nestjs-observable';\nimport { RequestContextService } from '@lark-apaas/nestjs-common';\n\nconst nestLogger = new Logger('Database');\n\n/**\n * 包装 postgres 实例以记录 SQL 和结果\n */\nexport function genPostgresProxy(sql: postgres.Sql, observable: Observable, requestContext: RequestContextService): postgres.Sql {\n  const wrapQuery = <T extends readonly (postgres.Row | undefined)[]>(query: postgres.PendingQuery<T>): postgres.PendingQuery<T> => {\n    if (query && typeof query.then === 'function') {\n      return new Proxy(query, {\n        get(target, prop, receiver) {\n          if (prop === 'then') {\n            return async (onfulfilled?: (value: postgres.RowList<T>) => unknown, onrejected?: (reason: unknown) => unknown) => {\n              const sqlStr = (query as unknown as Record<string, unknown>).strings instanceof Array ? (query as unknown as { strings: string[] }).strings[0] : '';\n              const span = observable.startTrace(sqlStr, requestContext?.getContext()?.requestRootSpan as Span);\n\n              span.setAttribute(\"module\", \"data\");\n              span.setAttribute(\"source_type\", \"platform\");\n\n              const start = Date.now();\n              try {\n                const result = await target;\n                const isSetRoleSql = result?.command === 'SET';\n\n                // rls SET ROLE 不上报日志\n                if (!isSetRoleSql) {\n                  const duration = Date.now() - start;\n\n                  const parsedParams = JSON.stringify({\n                    operation: result?.command,\n                    sql: result?.statement?.string,\n                    duration_ms: duration,\n                    result_count: result.length ?? 0,\n                    error_message: ''\n                  });\n\n                  nestLogger.log(parsedParams, {\n                    source_type: 'platform',\n                    paas_attributes_module: 'data',\n                    paas_parent_span: span,\n                  });\n                } else {\n                  observable.dropTrace(span);\n                }\n\n                span.end();\n                return onfulfilled ? onfulfilled(result) : result;\n              } catch (error) {\n                const duration = Date.now() - start;\n                const sqlStr = (query as unknown as Record<string, unknown>).strings instanceof Array ? (query as unknown as { strings: string[] }).strings[0] : '';\n                const parsedParams = JSON.stringify({\n                  operation: '',\n                  sql: sqlStr,\n                  duration_ms: duration,\n                  result_count: 0,\n                  error_message: error instanceof Error ? error.message : String(error)\n                });\n                nestLogger.log(parsedParams, {\n                  source_type: 'platform',\n                  paas_attributes_module: 'data',\n                  paas_parent_span: span,\n                });\n\n                span.end();\n                return onrejected ? onrejected(error) : Promise.reject(error);\n              }\n            };\n          }\n          const value = Reflect.get(target, prop, receiver);\n          return typeof value === 'function' ? value.bind(receiver) : value;\n        },\n      }) as postgres.PendingQuery<T>;\n    }\n    return query;\n  };\n\n  return new Proxy(sql, {\n    apply(target, thisArg, argArray) {\n      const query = Reflect.apply(target, thisArg, argArray);\n      return wrapQuery(query);\n    },\n    get(target, prop, receiver) {\n      const value = Reflect.get(target, prop, receiver);\n      if (prop === 'begin' && typeof value === 'function') {\n        return <T>(arg: string | ((tx: postgres.TransactionSql) => T | Promise<T>), cb?: (tx: postgres.TransactionSql) => T | Promise<T>) => {\n          if (typeof arg === 'function') {\n            return value.call(target, (tx: postgres.TransactionSql) => arg(genPostgresProxy(tx as unknown as postgres.Sql, observable, requestContext) as unknown as postgres.TransactionSql));\n          }\n          if (typeof cb === 'function') {\n            return value.call(target, arg, (tx: postgres.TransactionSql) => cb(genPostgresProxy(tx as unknown as postgres.Sql, observable, requestContext) as unknown as postgres.TransactionSql));\n          }\n          return value.call(target, arg);\n        };\n      }\n      if (prop === 'unsafe' && typeof value === 'function') {\n        return <T extends readonly (postgres.Row | undefined)[] = postgres.Row[]>(queryStr: string, args?: unknown[], options?: postgres.UnsafeQueryOptions): postgres.PendingQuery<T> => {\n          const query = value.call(target, queryStr, args, options) as postgres.PendingQuery<T>;\n          return wrapQuery(query);\n        };\n      }\n      return typeof value === 'function' ? value.bind(receiver) : value;\n    },\n  }) as postgres.Sql;\n}","import {\n  Module,\n  DynamicModule,\n  Provider,\n  MiddlewareConsumer,\n  InjectionToken,\n  ModuleMetadata,\n} from '@nestjs/common';\nimport { DrizzleDatabaseManager } from './drizzle-manager';\nimport { DataPaasDatabaseService } from './database.service';\nimport { SqlExecutionContextMiddleware } from './sql-exection-context.middleware';\nimport { type DataPaasConfig } from '@/types';\nimport { DATAPAAS_CONFIG, DRIZZLE_DATABASE, DATAPAAS_LOGGER_SERVICE } from './const';\nexport { type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { RequestContextService, OBSERVABLE_SERVICE, ObservableService } from '@lark-apaas/nestjs-common';\nimport { Observable } from '@lark-apaas/nestjs-observable';\n\nexport interface DataPaasModuleOptions extends DataPaasConfig {}\n\n/**\n * 异步配置选项\n */\nexport interface DataPaasModuleAsyncOptions {\n  /**\n   * 需要导入的模块（例如 ConfigModule）\n   */\n  imports?: ModuleMetadata['imports'];\n  /**\n   * 使用工厂函数创建配置\n   */\n  useFactory?: (...args: unknown[]) => Promise<DataPaasConfig> | DataPaasConfig;\n  /**\n   * 工厂函数需要注入的依赖\n   */\n  inject?: InjectionToken[];\n}\n\nfunction verifyDrizzlePatched(): void {\n  try {\n    // 直接加载被 patch 的 CJS 模块并检查导出是否存在\n    // 注意：此路径与补丁文件一致：drizzle-orm/postgres-js/session.cjs\n    // 若未来包结构变化，请同步更新补丁与此校验\n    // eslint-disable-next-line @typescript-eslint/no-var-requires\n    const postgres = require('drizzle-orm/postgres-js');\n    const ok =\n      postgres &&\n      typeof postgres.getAuthContextStore === 'function' &&\n      typeof postgres.runWithAuthContext === 'function';\n    if (!ok) {\n      throw new Error(\n        'drizzle-orm 未检测到补丁导出：getAuthContextStore/runWithAuthContext',\n      );\n    }\n  } catch (err) {\n    const reason = err instanceof Error ? err.message : String(err);\n    throw new Error(`drizzle-orm 补丁未生效或版本不匹配：${reason}`);\n  }\n}\n\n// 在模块初始化阶段进行一次 drizzle-orm 补丁自检\nverifyDrizzlePatched();\n\n@Module({\n  providers: [{\n    provide: DATAPAAS_LOGGER_SERVICE,\n    useFactory: (observable: ObservableService | null, rtx: RequestContextService) => {\n      return observable ?? new Observable(rtx);\n    },\n    inject: [{\n      token: OBSERVABLE_SERVICE,\n      optional: true,\n    }, RequestContextService],\n  }],\n})\nexport class DataPaasModule {\n  configure(_consumer: MiddlewareConsumer) {}\n  /**\n   * 注册数据平台模块（同步方式）\n   * @param options 模块配置选项\n   * @returns 动态模块\n   */\n  static forRoot(options: DataPaasConfig): DynamicModule {\n    let roleSchema = '';\n    let sslModeRequired = false;\n    if (options.connectionString) {\n      try {\n        const url = new URL(options.connectionString);\n\n        roleSchema = url.searchParams.get(\n          'schema'\n        ) as string;\n        sslModeRequired = url.searchParams.get('sslmode') === \"require\";\n      } catch (err) {\n        roleSchema = '';\n      }\n    }\n    const config: DataPaasConfig = {\n      connectionString: options.connectionString,\n      schema: options.schema,\n      logger: options.logger || false,\n      timeout: options.timeout || 10000,\n      maxConnections: options.maxConnections || 1,\n      idleTimeout: options.idleTimeout || 20,\n      connectionTimeout: options.connectionTimeout || 10,\n      ssl: options.ssl || (sslModeRequired ? 'require' : false),\n      autoContext: options.autoContext == null ? true : !!options.autoContext,\n      roleSchema,\n    };\n\n    const providers: Provider[] = [\n      // 配置提供者 - 使用 useValue\n      {\n        provide: DATAPAAS_CONFIG,\n        useValue: config,\n      },\n      // 数据库管理器 - 使用 useClass，依赖注入连接服务和配置服务\n      {\n        provide: DrizzleDatabaseManager,\n        useClass: DrizzleDatabaseManager,\n      },\n      // 数据库服务 - 使用 useClass，依赖注入管理器\n      {\n        provide: DataPaasDatabaseService,\n        useClass: DataPaasDatabaseService,\n      },\n      {\n        provide: SqlExecutionContextMiddleware,\n        useClass: SqlExecutionContextMiddleware,\n      },\n      // Drizzle 数据库实例 - 使用 useFactory 获取管理器中的数据库实例\n      {\n        provide: DRIZZLE_DATABASE,\n        useFactory: async (manager: DrizzleDatabaseManager) => {\n          await manager.initialize();\n          return manager.getDatabase();\n        },\n        inject: [DrizzleDatabaseManager],\n      },\n    ];\n\n    return {\n      module: DataPaasModule,\n      providers,\n      exports: [\n        DrizzleDatabaseManager,\n        DataPaasDatabaseService,\n        SqlExecutionContextMiddleware,\n        DRIZZLE_DATABASE,\n        DATAPAAS_CONFIG,\n      ],\n      global: true,\n    };\n  }\n\n  /**\n   * 注册数据平台模块（异步方式）\n   * 支持从 ConfigModule 等异步源获取配置\n   * @param options 异步配置选项\n   * @returns 动态模块\n   * @example\n   * ```typescript\n   * DataPaasModule.forRootAsync({\n   *   imports: [ConfigModule],\n   *   useFactory: async (configService: ConfigService) => ({\n   *     connectionString: configService.get('DATABASE_URL'),\n   *   }),\n   *   inject: [ConfigService],\n   * })\n   * ```\n   */\n  static forRootAsync(options: DataPaasModuleAsyncOptions): DynamicModule {\n    const asyncProviders = this.createAsyncProviders(options);\n\n    const providers: Provider[] = [\n      ...asyncProviders,\n      // 数据库管理器 - 使用 useClass，依赖注入连接服务和配置服务\n      {\n        provide: DrizzleDatabaseManager,\n        useClass: DrizzleDatabaseManager,\n      },\n      // 数据库服务 - 使用 useClass，依赖注入管理器\n      {\n        provide: DataPaasDatabaseService,\n        useClass: DataPaasDatabaseService,\n      },\n      {\n        provide: SqlExecutionContextMiddleware,\n        useClass: SqlExecutionContextMiddleware,\n      },\n      // Drizzle 数据库实例 - 使用 useFactory 获取管理器中的数据库实例\n      {\n        provide: DRIZZLE_DATABASE,\n        useFactory: async (manager: DrizzleDatabaseManager) => {\n          await manager.initialize();\n          return manager.getDatabase();\n        },\n        inject: [DrizzleDatabaseManager],\n      },\n    ];\n\n    return {\n      module: DataPaasModule,\n      imports: options.imports || [],\n      providers,\n      exports: [\n        DrizzleDatabaseManager,\n        DataPaasDatabaseService,\n        SqlExecutionContextMiddleware,\n        DRIZZLE_DATABASE,\n        DATAPAAS_CONFIG,\n      ],\n      global: true,\n    };\n  }\n\n  /**\n   * 创建异步配置 providers\n   */\n  private static createAsyncProviders(\n    options: DataPaasModuleAsyncOptions\n  ): Provider[] {\n    if (options.useFactory) {\n      return [this.createAsyncConfigProvider(options)];\n    }\n\n    throw new Error('DataPaasModule.forRootAsync 需要指定 useFactory');\n  }\n\n  /**\n   * 创建异步配置 provider\n   */\n  private static createAsyncConfigProvider(\n    options: DataPaasModuleAsyncOptions\n  ): Provider {\n    if (options.useFactory) {\n      return {\n        provide: DATAPAAS_CONFIG,\n        useFactory: async (...args: unknown[]) => {\n          const config = await options.useFactory!(...args);\n          return this.normalizeConfig(config);\n        },\n        inject: options.inject || [],\n      };\n    }\n\n    throw new Error('Invalid async options');\n  }\n\n  /**\n   * 标准化配置，设置默认值\n   */\n  private static normalizeConfig(config: DataPaasConfig): DataPaasConfig {\n    let roleSchema = '';\n    let sslModeRequired = false;\n    if (config.connectionString) {\n      try {\n        const url = new URL(config.connectionString);\n        roleSchema = url.searchParams.get(\n          'schema'\n        ) as string;\n        sslModeRequired = url.searchParams.get('sslmode') === \"require\";\n      } catch (err) {\n        roleSchema = '';\n      }\n    }\n    return {\n      connectionString: config.connectionString,\n      schema: config.schema,\n      logger: config.logger || false,\n      timeout: config.timeout || 10000,\n      maxConnections: config.maxConnections || 1,\n      idleTimeout: config.idleTimeout || 20,\n      connectionTimeout: config.connectionTimeout || 10,\n      ssl: config.ssl || (sslModeRequired ? 'require' : false),\n      autoContext: config.autoContext == null ? true : !!config.autoContext,\n      roleSchema,\n    };\n  }\n}\n","import { Injectable, OnModuleDestroy } from '@nestjs/common';\nimport { type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { DrizzleDatabaseManager } from './drizzle-manager';\nimport { DatabaseConnectionInfo } from '../types';\n\n@Injectable()\nexport class DataPaasDatabaseService implements OnModuleDestroy {\n  constructor(\n    private readonly databaseManager: DrizzleDatabaseManager,\n  ) {}\n\n  /**\n   * 获取 Drizzle 数据库实例\n   * @returns Drizzle 数据库实例\n   */\n  getDatabase(): PostgresJsDatabase {\n    return this.databaseManager.getDatabase();\n  }\n\n  /**\n   * 获取数据库管理器\n   * @returns 数据库管理器实例\n   */\n  getDatabaseManager(): DrizzleDatabaseManager {\n    return this.databaseManager;\n  }\n\n\n  /**\n   * 测试数据库连接\n   * @returns 连接是否正常\n   */\n  async testConnection(): Promise<boolean> {\n    return this.databaseManager.testConnection();\n  }\n\n  /**\n   * 重新连接数据库\n   * @param forceRefresh 是否强制刷新连接信息\n   * @returns 新的数据库实例\n   */\n  async reconnect(forceRefresh: boolean = false): Promise<PostgresJsDatabase> {\n    return this.databaseManager.reconnect(forceRefresh);\n  }\n\n  /**\n   * 获取当前连接信息\n   * @returns 当前连接信息\n   */\n  getConnectionInfo(): DatabaseConnectionInfo | null {\n    return this.databaseManager.getConnectionInfo();\n  }\n\n  /**\n   * 检查连接状态\n   * @returns 是否已连接\n   */\n  isConnected(): boolean {\n    return this.databaseManager.isDatabaseConnected();\n  }\n\n  /**\n   * 开始事务\n   * @param callback 事务回调函数\n   * @returns 事务结果\n   */\n  async transaction<T>(callback: (tx: PostgresJsDatabase) => Promise<T>): Promise<T> {\n    return this.databaseManager.transaction(callback);\n  }\n\n  /**\n   * 获取连接池状态\n   * @returns 连接池状态信息\n   */\n  getConnectionPoolStatus(): { isConnected: boolean; connectionInfo: DatabaseConnectionInfo | null } {\n    return this.databaseManager.getConnectionPoolStatus();\n  }\n\n\n  /**\n   * 模块销毁时断开数据库连接\n   */\n  async onModuleDestroy(): Promise<void> {\n    await this.databaseManager.disconnect();\n  }\n}\n","import { Injectable, NestMiddleware, Inject } from '@nestjs/common';\nimport { Request, Response, NextFunction } from 'express';\nimport * as session from 'drizzle-orm/postgres-js';\nimport { DATAPAAS_CONFIG } from './const';\nimport { type DataPaasConfig } from '../types';\n\ninterface UserContext {\n  userId?: string;\n  tenantId?: number;\n  appId?: string;\n  // 用户角色（如果有）\n  userRoles?: string[];\n  // 是否为系统账号\n  isSystemAccount: boolean;\n  // 用户类型\n  userType?: string;\n}\n\n@Injectable()\nexport class SqlExecutionContextMiddleware implements NestMiddleware {\n  constructor(\n    @Inject(DATAPAAS_CONFIG)\n    private readonly config: DataPaasConfig,\n  ) {}\n\n  use(req: Request, _res: Response, next: NextFunction) {\n    const roleSchema = this.config.roleSchema ?? '';\n    const userContext =\n      (req as unknown as { userContext: UserContext }).userContext || {};\n    const roles = (userContext.userRoles || []).join(',');\n    const userId = userContext.userId;\n    const isSystemAccount = userContext.isSystemAccount;\n    const userType = userContext.userType;\n    const sqls = [\n      // 设置用户 ID, 匿名用户时设置为空字符串\n      userId\n        ? `SET LOCAL app.user_id = '${userId}'`\n        : `SET LOCAL app.user_id = ''`,\n      // 系统用户设置为service_role, 登录用户设置为 authenticated_xxx 角色，匿名用户设置为 anon_xxx 角色\n      isSystemAccount ? `SET LOCAL ROLE 'service_role_${roleSchema}'` :  (\n        userId\n          ? `SET LOCAL ROLE 'authenticated_${roleSchema}'`\n          : `SET LOCAL ROLE 'anon_${roleSchema}'`),\n      // 自定义角色\n      roles && `SET LOCAL app.role_ids = '${roles}'`,\n      // 用户类型\n      userType\n        ? `SET LOCAL app.user_type = '${userType}'`\n        : `SET LOCAL app.user_type = ''`,\n    ]\n      .filter(Boolean)\n      .join(';');\n\n    // 需要确认角色设置是否对后续 SQL 执行有影响，预期是只影响当前会话\n    session.runWithAuthContext({ preSql: sqls }, () => {\n      next();\n    });\n  }\n}\n","import { type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { sql } from 'drizzle-orm';\n\nimport { DataPaasDatabaseService } from './database.service';\n\n/**\n * 数据库工具函数集合\n */\nexport class DatabaseUtils {\n  /**\n   * 安全执行数据库操作\n   * @param databaseService 数据库服务实例\n   * @param operation 数据库操作函数\n   * @param fallback 失败时的回退值\n   * @returns 操作结果或回退值\n   */\n  static async safeExecute<T>(\n    databaseService: DataPaasDatabaseService,\n    operation: (db: PostgresJsDatabase) => Promise<T>,\n    fallback?: T\n  ): Promise<T | undefined> {\n    try {\n      if (!databaseService.isConnected()) {\n        throw new Error('Database not connected');\n      }\n      \n      const db = databaseService.getDatabase();\n      return await operation(db);\n    } catch (error) {\n      console.error('Database operation failed:', error);\n      return fallback;\n    }\n  }\n\n  /**\n   * 批量执行数据库操作\n   * @param databaseService 数据库服务实例\n   * @param operations 操作数组\n   * @param batchSize 批处理大小\n   * @returns 所有操作的结果\n   */\n  static async batchExecute<T>(\n    databaseService: DataPaasDatabaseService,\n    operations: Array<(db: PostgresJsDatabase) => Promise<T>>,\n    batchSize: number = 10\n  ): Promise<T[]> {\n    const results: T[] = [];\n    const db = databaseService.getDatabase();\n\n    for (let i = 0; i < operations.length; i += batchSize) {\n      const batch = operations.slice(i, i + batchSize);\n      const batchResults = await Promise.all(\n        batch.map(operation => operation(db))\n      );\n      results.push(...batchResults);\n    }\n\n    return results;\n  }\n\n  /**\n   * 执行带重试的数据库操作\n   * @param databaseService 数据库服务实例\n   * @param operation 数据库操作函数\n   * @param maxRetries 最大重试次数\n   * @param retryDelay 重试延迟（毫秒）\n   * @returns 操作结果\n   */\n  static async executeWithRetry<T>(\n    databaseService: DataPaasDatabaseService,\n    operation: (db: PostgresJsDatabase) => Promise<T>,\n    maxRetries: number = 3,\n    retryDelay: number = 1000\n  ): Promise<T> {\n    let lastError: Error;\n\n    for (let attempt = 0; attempt <= maxRetries; attempt++) {\n      try {\n        const db = databaseService.getDatabase();\n        return await operation(db);\n      } catch (error) {\n        lastError = error as Error;\n        \n        if (attempt === maxRetries) {\n          break;\n        }\n\n        console.warn(`Database operation failed (attempt ${attempt + 1}/${maxRetries + 1}):`, error);\n        \n        // 等待后重试\n        await new Promise(resolve => setTimeout(resolve, retryDelay * (attempt + 1)));\n      }\n    }\n\n    throw lastError!;\n  }\n\n  /**\n   * 检查表是否存在\n   * @param databaseService 数据库服务实例\n   * @param tableName 表名\n   * @returns 表是否存在\n   */\n  static async tableExists(\n    db: PostgresJsDatabase,\n    tableName: string\n  ): Promise<boolean> {\n    try {\n      const result = await db.execute(\n        sql`SELECT EXISTS (\n          SELECT FROM information_schema.tables \n          WHERE table_schema = 'public' \n          AND table_name = ${tableName}\n        )`\n      );\n      \n      return result[0]?.exists as boolean || false;\n    } catch (error) {\n      console.error('Failed to check table existence:', error);\n      return false;\n    }\n  }\n\n  /**\n   * 获取表结构信息\n   * @param databaseService 数据库服务实例\n   * @param tableName 表名\n   * @returns 表结构信息\n   */\n  static async getTableStructure(\n    db: PostgresJsDatabase,\n    tableName: string\n  ): Promise<any[]> {\n    try {\n      const result = await db.execute(\n        sql`SELECT \n          column_name,\n          data_type,\n          is_nullable,\n          column_default,\n          character_maximum_length\n        FROM information_schema.columns \n        WHERE table_schema = 'public' \n        AND table_name = ${tableName}\n        ORDER BY ordinal_position`,\n      );\n      \n      return result;\n    } catch (error) {\n      console.error('Failed to get table structure:', error);\n      return [];\n    }\n  }\n\n  /**\n   * 获取数据库版本信息\n   * @param databaseService 数据库服务实例\n   * @returns 数据库版本\n   */\n  static async getDatabaseVersion(\n    db: PostgresJsDatabase\n  ): Promise<string> {\n    try {\n      const result = await db.execute(sql`SELECT version()`);\n      return result[0]?.version as string || 'Unknown';\n    } catch (error) {\n      console.error('Failed to get database version:', error);\n      return 'Unknown';\n    }\n  }\n\n  /**\n   * 获取连接统计信息\n   * @param databaseService 数据库服务实例\n   * @returns 连接统计信息\n   */\n  static async getConnectionStats(\n    db: PostgresJsDatabase\n  ): Promise<any> {\n    try {\n      const result = await db.execute(sql`\n        SELECT \n          count(*) as total_connections,\n          count(*) FILTER (WHERE state = 'active') as active_connections,\n          count(*) FILTER (WHERE state = 'idle') as idle_connections\n        FROM pg_stat_activity \n        WHERE datname = current_database()\n      `);\n      \n      return result[0] || {};\n    } catch (error) {\n      console.error('Failed to get connection stats:', error);\n      return {};\n    }\n  }\n\n  /**\n   * 健康检查\n   * @param databaseService 数据库服务实例\n   * @returns 健康状态\n   */\n  static async healthCheck(\n    databaseService: DataPaasDatabaseService\n  ): Promise<{\n    isHealthy: boolean;\n    isConnected: boolean;\n    version?: string;\n    connectionInfo?: any;\n    error?: string;\n  }> {\n    try {\n      const isConnected = databaseService.isConnected();\n      const connectionInfo = databaseService.getConnectionInfo();\n\n      const db = databaseService.getDatabase();\n      if (!isConnected) {\n        return {\n          isHealthy: false,\n          isConnected: false,\n          error: 'Database not connected'\n        };\n      }\n\n      const version = await this.getDatabaseVersion(db);\n      \n      return {\n        isHealthy: true,\n        isConnected: true,\n        version,\n        connectionInfo\n      };\n    } catch (error) {\n      return {\n        isHealthy: false,\n        isConnected: false,\n        error: error instanceof Error ? error.message : 'Unknown error'\n      };\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;AAAA,SAASA,YAA2CC,cAAc;AAClE,SAASC,eAAwC;AACjD,SAASC,WAAW;AACpB,OAAOC,cAAc;;;ACHd,IAAMC,mBAAmB;AACzB,IAAMC,kBAAkB;AACxB,IAAMC,0BAA0BC,uBAAO,yBAAA;;;ADG9C,SAASC,kBAAkB;AAC3B,SAASC,6BAA6B;;;AEK/B,SAASC,oBAAoBC,kBAA0BC,QAAyB;AACrF,QAAM,EAAEC,gBAAgBC,aAAaC,mBAAmBC,KAAKC,OAAM,IAAKL,UAAU,CAAC;AAEnF,MAAIM;AACJ,MAAI;AACFA,0BAAsB,IAAIC,IAAIR,gBAAAA;AAC9B,UAAMS,OAAO,IAAID,IAAIR,gBAAAA;AACrBS,SAAKC,aAAaC,OAAO,QAAA;AACzBX,uBAAmBS,KAAKG,SAAQ;EAClC,SAASC,OAAO;AACd,UAAM,IAAIC,MAAM,iCAAA;EAClB;AACA,QAAM,EAAEC,UAAUC,UAAUC,UAAUC,MAAMC,SAAQ,IAAKZ;AACzD,QAAMa,mBAAmB;IACvBL,UAAUM,mBAAmBN,QAAAA;IAC7BC,UAAUK,mBAAmBL,QAAAA;IAC7BM,MAAM;MAACL;;IACPC,MAAM;MAACK,SAASL,IAAAA,KAAS;;IACzBC,UAAUE,mBAAmBF,QAAAA;IAC7BK,KAAKtB,kBAAkB;IACvBuB,cAActB,eAAe;IAC7BuB,iBAAiBtB,qBAAqB;IACtCC;IACAsB,UAAUrB,SAASsB,QAAQC,MAAMC;EACnC;AACA,SAAO;IAAE9B;IAAkBoB;EAAiB;AAC9C;AA1BgBrB;;;ACVhB,SAASgC,cAAc;AAIvB,IAAMC,aAAa,IAAIC,OAAO,UAAA;AAKvB,SAASC,iBAAiBC,MAAmBC,YAAwBC,gBAAqC;AAC/G,QAAMC,YAAY,wBAAkDC,UAAAA;AAClE,QAAIA,SAAS,OAAOA,MAAMC,SAAS,YAAY;AAC7C,aAAO,IAAIC,MAAMF,OAAO;QACtBG,IAAIC,QAAQC,MAAMC,UAAQ;AACxB,cAAID,SAAS,QAAQ;AACnB,mBAAO,OAAOE,aAAuDC,eAAAA;AACnE,oBAAMC,SAAUT,MAA6CU,mBAAmBC,QAASX,MAA2CU,QAAQ,CAAA,IAAK;AACjJ,oBAAME,OAAOf,WAAWgB,WAAWJ,QAAQX,gBAAgBgB,WAAAA,GAAcC,eAAAA;AAEzEH,mBAAKI,aAAa,UAAU,MAAA;AAC5BJ,mBAAKI,aAAa,eAAe,UAAA;AAEjC,oBAAMC,QAAQC,KAAKC,IAAG;AACtB,kBAAI;AACF,sBAAMC,SAAS,MAAMhB;AACrB,sBAAMiB,eAAeD,QAAQE,YAAY;AAGzC,oBAAI,CAACD,cAAc;AACjB,wBAAME,WAAWL,KAAKC,IAAG,IAAKF;AAE9B,wBAAMO,eAAeC,KAAKC,UAAU;oBAClCC,WAAWP,QAAQE;oBACnB1B,KAAKwB,QAAQQ,WAAWC;oBACxBC,aAAaP;oBACbQ,cAAcX,OAAOY,UAAU;oBAC/BC,eAAe;kBACjB,CAAA;AAEAxC,6BAAWyC,IAAIV,cAAc;oBAC3BW,aAAa;oBACbC,wBAAwB;oBACxBC,kBAAkBzB;kBACpB,CAAA;gBACF,OAAO;AACLf,6BAAWyC,UAAU1B,IAAAA;gBACvB;AAEAA,qBAAK2B,IAAG;AACR,uBAAOhC,cAAcA,YAAYa,MAAAA,IAAUA;cAC7C,SAASoB,OAAO;AACd,sBAAMjB,WAAWL,KAAKC,IAAG,IAAKF;AAC9B,sBAAMR,UAAUT,MAA6CU,mBAAmBC,QAASX,MAA2CU,QAAQ,CAAA,IAAK;AACjJ,sBAAMc,eAAeC,KAAKC,UAAU;kBAClCC,WAAW;kBACX/B,KAAKa;kBACLqB,aAAaP;kBACbQ,cAAc;kBACdE,eAAeO,iBAAiBC,QAAQD,MAAME,UAAUC,OAAOH,KAAAA;gBACjE,CAAA;AACA/C,2BAAWyC,IAAIV,cAAc;kBAC3BW,aAAa;kBACbC,wBAAwB;kBACxBC,kBAAkBzB;gBACpB,CAAA;AAEAA,qBAAK2B,IAAG;AACR,uBAAO/B,aAAaA,WAAWgC,KAAAA,IAASI,QAAQC,OAAOL,KAAAA;cACzD;YACF;UACF;AACA,gBAAMM,QAAQC,QAAQ5C,IAAIC,QAAQC,MAAMC,QAAAA;AACxC,iBAAO,OAAOwC,UAAU,aAAaA,MAAME,KAAK1C,QAAAA,IAAYwC;QAC9D;MACF,CAAA;IACF;AACA,WAAO9C;EACT,GAnEkB;AAqElB,SAAO,IAAIE,MAAMN,MAAK;IACpBqD,MAAM7C,QAAQ8C,SAASC,UAAQ;AAC7B,YAAMnD,QAAQ+C,QAAQE,MAAM7C,QAAQ8C,SAASC,QAAAA;AAC7C,aAAOpD,UAAUC,KAAAA;IACnB;IACAG,IAAIC,QAAQC,MAAMC,UAAQ;AACxB,YAAMwC,QAAQC,QAAQ5C,IAAIC,QAAQC,MAAMC,QAAAA;AACxC,UAAID,SAAS,WAAW,OAAOyC,UAAU,YAAY;AACnD,eAAO,CAAIM,KAAiEC,OAAAA;AAC1E,cAAI,OAAOD,QAAQ,YAAY;AAC7B,mBAAON,MAAMQ,KAAKlD,QAAQ,CAACmD,OAAgCH,IAAIzD,iBAAiB4D,IAA+B1D,YAAYC,cAAAA,CAAAA,CAAAA;UAC7H;AACA,cAAI,OAAOuD,OAAO,YAAY;AAC5B,mBAAOP,MAAMQ,KAAKlD,QAAQgD,KAAK,CAACG,OAAgCF,GAAG1D,iBAAiB4D,IAA+B1D,YAAYC,cAAAA,CAAAA,CAAAA;UACjI;AACA,iBAAOgD,MAAMQ,KAAKlD,QAAQgD,GAAAA;QAC5B;MACF;AACA,UAAI/C,SAAS,YAAY,OAAOyC,UAAU,YAAY;AACpD,eAAO,CAAmEU,UAAkBC,MAAkBC,YAAAA;AAC5G,gBAAM1D,QAAQ8C,MAAMQ,KAAKlD,QAAQoD,UAAUC,MAAMC,OAAAA;AACjD,iBAAO3D,UAAUC,KAAAA;QACnB;MACF;AACA,aAAO,OAAO8C,UAAU,aAAaA,MAAME,KAAK1C,QAAAA,IAAYwC;IAC9D;EACF,CAAA;AACF;AAjGgBnD;;;;;;;;;;;;;;;;;;;;AHET,IAAMgE,yBAAN,MAAMA;SAAAA;;;;;;EACHC,KAAgC;EAChCC,wBAAuD;EACvDC,cAAuB;EACvBC,uBAAgC;EACxC,YAEmBC,QAEAC,YACAC,gBACjB;SAJiBF,SAAAA;SAEAC,aAAAA;SACAC,iBAAAA;AAEjB,SAAKH,uBAAuB,CAAC,CAACI,QAAQC,IAAIC;EAC5C;EAEA,MAAMC,eAAe;AACnB,UAAM,KAAKC,WAAU;EACvB;EAEA,MAAMC,kBAAkB;AACtB,UAAM,KAAKC,WAAU;EACvB;;;;;EAMA,MAAMF,aAA0C;AAC9C,QAAI,KAAKR,sBAAsB;AAC7B,WAAKD,cAAc;AACnB,WAAKF,KAAK;QAACc,MAAM;MAAsB;AACvC,aAAO,KAAKd;IACd;AACA,QAAI;AACF,YAAM,EAAEe,kBAAkBC,iBAAgB,IAAKC,oBAAoB,KAAKb,OAAOW,kBAAkB,KAAKX,MAAM;AAC5G,YAAMc,UAAUC,SAASJ,kBAAkBC,gBAAAA;AAC3C,YAAMI,WAAWC,iBAAiBH,SAAS,KAAKb,YAAY,KAAKC,cAAc;AAE/E,WAAKN,KAAKsB,QAAQF,UAAU;QAAEG,QAAQ,KAAKnB,OAAOmB;QAAQC,QAAQ;MAAM,CAAA;AACxE,WAAKvB,wBAAwB;QAC3Bc;QACAC,kBAAkB;UAChBS,KAAKT,iBAAiBS;UACtBC,mBAAmBV,iBAAiBW;UACpCC,aAAaZ,iBAAiBa;UAC9BC,gBAAgBd,iBAAiBe;QACnC;MACF;AACA,WAAK7B,cAAc;AACnB,aAAO,KAAKF;IACd,SAASgC,OAAO;AACd,WAAK9B,cAAc;AACnB,YAAM,IAAI+B,MAAM,6CAA6CD,iBAAiBC,QAAQD,MAAME,UAAU,eAAA,EAAiB;IACzH;EACF;;;;;EAMAC,cAAkC;AAChC,QAAI,CAAC,KAAKnC,MAAM,CAAC,KAAKE,aAAa;AACjC,YAAM,IAAI+B,MAAM,oDAAA;IAClB;AACA,WAAO,KAAKjC;EACd;;;;;EAMA,MAAMoC,iBAAmC;AACvC,QAAI,CAAC,KAAKpC,IAAI;AACZ,aAAO;IACT;AAEA,QAAI;AACF,YAAM,KAAKA,GAAGqC,QAAQC,aAAa;AACnC,aAAO;IACT,SAASN,OAAO;AACdO,cAAQP,MAAM,oCAAoCA,KAAAA;AAClD,WAAK9B,cAAc;AACnB,aAAO;IACT;EACF;;;;;;EAOA,MAAMsC,UAAUC,gBAAyB,OAAoC;AAC3E,UAAM,KAAK5B,WAAU;AACrB,WAAO,KAAKF,WAAU;EACxB;;;;EAKA,MAAME,aAA4B;AAChC,QAAI,KAAKb,IAAI;AACX,UAAI;AAEF,aAAKA,KAAK;AACV,aAAKE,cAAc;AACnB,aAAKD,wBAAwB;MAC/B,SAAS+B,OAAO;AACdO,gBAAQP,MAAM,sCAAsCA,KAAAA;MACtD;IACF;EACF;;;;;EAMAU,oBAAmD;AACjD,WAAO,KAAKzC;EACd;;;;;EAMA0C,sBAA+B;AAC7B,WAAO,KAAKzC;EACd;;;;;;EAQA,MAAM0C,YAAeC,UAA8D;AACjF,UAAM7C,KAAK,KAAKmC,YAAW;AAC3B,WAAOnC,GAAG4C,YAAYC,QAAAA;EACxB;;;;;EAMAC,0BAAmG;AACjG,WAAO;MACL5C,aAAa,KAAKA;MAClB6C,gBAAgB,KAAK9C;IACvB;EACF;AACF;;;;;;;;;;;;;;AIjKA,SACE+C,cAMK;;;ACPP,SAASC,cAAAA,mBAAmC;;;;;;;;;;;;AAMrC,IAAMC,0BAAN,MAAMA;SAAAA;;;;EACX,YACmBC,iBACjB;SADiBA,kBAAAA;EAChB;;;;;EAMHC,cAAkC;AAChC,WAAO,KAAKD,gBAAgBC,YAAW;EACzC;;;;;EAMAC,qBAA6C;AAC3C,WAAO,KAAKF;EACd;;;;;EAOA,MAAMG,iBAAmC;AACvC,WAAO,KAAKH,gBAAgBG,eAAc;EAC5C;;;;;;EAOA,MAAMC,UAAUC,eAAwB,OAAoC;AAC1E,WAAO,KAAKL,gBAAgBI,UAAUC,YAAAA;EACxC;;;;;EAMAC,oBAAmD;AACjD,WAAO,KAAKN,gBAAgBM,kBAAiB;EAC/C;;;;;EAMAC,cAAuB;AACrB,WAAO,KAAKP,gBAAgBQ,oBAAmB;EACjD;;;;;;EAOA,MAAMC,YAAeC,UAA8D;AACjF,WAAO,KAAKV,gBAAgBS,YAAYC,QAAAA;EAC1C;;;;;EAMAC,0BAAmG;AACjG,WAAO,KAAKX,gBAAgBW,wBAAuB;EACrD;;;;EAMA,MAAMC,kBAAiC;AACrC,UAAM,KAAKZ,gBAAgBa,WAAU;EACvC;AACF;;;;;;;;;;ACrFA,SAASC,cAAAA,aAA4BC,UAAAA,eAAc;AAEnD,YAAYC,aAAa;;;;;;;;;;;;;;;;;;AAiBlB,IAAMC,gCAAN,MAAMA;SAAAA;;;;EACX,YAEmBC,QACjB;SADiBA,SAAAA;EAChB;EAEHC,IAAIC,KAAcC,MAAgBC,MAAoB;AACpD,UAAMC,aAAa,KAAKL,OAAOK,cAAc;AAC7C,UAAMC,cACHJ,IAAgDI,eAAe,CAAC;AACnE,UAAMC,SAASD,YAAYE,aAAa,CAAA,GAAIC,KAAK,GAAA;AACjD,UAAMC,SAASJ,YAAYI;AAC3B,UAAMC,kBAAkBL,YAAYK;AACpC,UAAMC,WAAWN,YAAYM;AAC7B,UAAMC,OAAO;;MAEXH,SACI,4BAA4BA,MAAAA,MAC5B;;MAEJC,kBAAkB,gCAAgCN,UAAAA,MAChDK,SACI,iCAAiCL,UAAAA,MACjC,wBAAwBA,UAAAA;;MAE9BE,SAAS,6BAA6BA,KAAAA;;MAEtCK,WACI,8BAA8BA,QAAAA,MAC9B;MAEHE,OAAOC,OAAAA,EACPN,KAAK,GAAA;AAGRO,IAAQC,2BAAmB;MAAEC,QAAQL;IAAK,GAAG,MAAA;AAC3CT,WAAAA;IACF,CAAA;EACF;AACF;;;;;;;;;;;AF5CA,SAASe,yBAAAA,wBAAuBC,0BAA6C;AAC7E,SAASC,cAAAA,mBAAkB;;;;;;;;AAsB3B,SAASC,uBAAAA;AACP,MAAI;AAKF,UAAMC,YAAWC,UAAQ,yBAAA;AACzB,UAAMC,KACJF,aACA,OAAOA,UAASG,wBAAwB,cACxC,OAAOH,UAASI,uBAAuB;AACzC,QAAI,CAACF,IAAI;AACP,YAAM,IAAIG,MACR,0GAAA;IAEJ;EACF,SAASC,KAAK;AACZ,UAAMC,SAASD,eAAeD,QAAQC,IAAIE,UAAUC,OAAOH,GAAAA;AAC3D,UAAM,IAAID,MAAM,uFAA2BE,MAAAA,EAAQ;EACrD;AACF;AApBSR;AAuBTA,qBAAAA;AAcO,IAAMW,iBAAN,MAAMA,gBAAAA;SAAAA;;;EACXC,UAAUC,WAA+B;EAAC;;;;;;EAM1C,OAAOC,QAAQC,SAAwC;AACrD,QAAIC,aAAa;AACjB,QAAIC,kBAAkB;AACtB,QAAIF,QAAQG,kBAAkB;AAC5B,UAAI;AACF,cAAMC,MAAM,IAAIC,IAAIL,QAAQG,gBAAgB;AAE5CF,qBAAaG,IAAIE,aAAaC,IAC5B,QAAA;AAEFL,0BAAkBE,IAAIE,aAAaC,IAAI,SAAA,MAAe;MACxD,SAASf,KAAK;AACZS,qBAAa;MACf;IACF;AACA,UAAMO,SAAyB;MAC7BL,kBAAkBH,QAAQG;MAC1BM,QAAQT,QAAQS;MAChBC,QAAQV,QAAQU,UAAU;MAC1BC,SAASX,QAAQW,WAAW;MAC5BC,gBAAgBZ,QAAQY,kBAAkB;MAC1CC,aAAab,QAAQa,eAAe;MACpCC,mBAAmBd,QAAQc,qBAAqB;MAChDC,KAAKf,QAAQe,QAAQb,kBAAkB,YAAY;MACnDc,aAAahB,QAAQgB,eAAe,OAAO,OAAO,CAAC,CAAChB,QAAQgB;MAC5Df;IACF;AAEA,UAAMgB,YAAwB;;MAE5B;QACEC,SAASC;QACTC,UAAUZ;MACZ;;MAEA;QACEU,SAASG;QACTC,UAAUD;MACZ;;MAEA;QACEH,SAASK;QACTD,UAAUC;MACZ;MACA;QACEL,SAASM;QACTF,UAAUE;MACZ;;MAEA;QACEN,SAASO;QACTC,YAAY,8BAAOC,YAAAA;AACjB,gBAAMA,QAAQC,WAAU;AACxB,iBAAOD,QAAQE,YAAW;QAC5B,GAHY;QAIZC,QAAQ;UAACT;;MACX;;AAGF,WAAO;MACLU,QAAQnC;MACRqB;MACAe,SAAS;QACPX;QACAE;QACAC;QACAC;QACAN;;MAEFc,QAAQ;IACV;EACF;;;;;;;;;;;;;;;;;EAkBA,OAAOC,aAAalC,SAAoD;AACtE,UAAMmC,iBAAiB,KAAKC,qBAAqBpC,OAAAA;AAEjD,UAAMiB,YAAwB;SACzBkB;;MAEH;QACEjB,SAASG;QACTC,UAAUD;MACZ;;MAEA;QACEH,SAASK;QACTD,UAAUC;MACZ;MACA;QACEL,SAASM;QACTF,UAAUE;MACZ;;MAEA;QACEN,SAASO;QACTC,YAAY,8BAAOC,YAAAA;AACjB,gBAAMA,QAAQC,WAAU;AACxB,iBAAOD,QAAQE,YAAW;QAC5B,GAHY;QAIZC,QAAQ;UAACT;;MACX;;AAGF,WAAO;MACLU,QAAQnC;MACRyC,SAASrC,QAAQqC,WAAW,CAAA;MAC5BpB;MACAe,SAAS;QACPX;QACAE;QACAC;QACAC;QACAN;;MAEFc,QAAQ;IACV;EACF;;;;EAKA,OAAeG,qBACbpC,SACY;AACZ,QAAIA,QAAQ0B,YAAY;AACtB,aAAO;QAAC,KAAKY,0BAA0BtC,OAAAA;;IACzC;AAEA,UAAM,IAAIT,MAAM,iEAAA;EAClB;;;;EAKA,OAAe+C,0BACbtC,SACU;AACV,QAAIA,QAAQ0B,YAAY;AACtB,aAAO;QACLR,SAASC;QACTO,YAAY,iCAAUa,SAAAA;AACpB,gBAAM/B,SAAS,MAAMR,QAAQ0B,WAAU,GAAKa,IAAAA;AAC5C,iBAAO,KAAKC,gBAAgBhC,MAAAA;QAC9B,GAHY;QAIZsB,QAAQ9B,QAAQ8B,UAAU,CAAA;MAC5B;IACF;AAEA,UAAM,IAAIvC,MAAM,uBAAA;EAClB;;;;EAKA,OAAeiD,gBAAgBhC,QAAwC;AACrE,QAAIP,aAAa;AACjB,QAAIC,kBAAkB;AACtB,QAAIM,OAAOL,kBAAkB;AAC3B,UAAI;AACF,cAAMC,MAAM,IAAIC,IAAIG,OAAOL,gBAAgB;AAC3CF,qBAAaG,IAAIE,aAAaC,IAC5B,QAAA;AAEFL,0BAAkBE,IAAIE,aAAaC,IAAI,SAAA,MAAe;MACxD,SAASf,KAAK;AACZS,qBAAa;MACf;IACF;AACA,WAAO;MACLE,kBAAkBK,OAAOL;MACzBM,QAAQD,OAAOC;MACfC,QAAQF,OAAOE,UAAU;MACzBC,SAASH,OAAOG,WAAW;MAC3BC,gBAAgBJ,OAAOI,kBAAkB;MACzCC,aAAaL,OAAOK,eAAe;MACnCC,mBAAmBN,OAAOM,qBAAqB;MAC/CC,KAAKP,OAAOO,QAAQb,kBAAkB,YAAY;MAClDc,aAAaR,OAAOQ,eAAe,OAAO,OAAO,CAAC,CAACR,OAAOQ;MAC1Df;IACF;EACF;AACF;;;IAvNEgB,WAAW;MAAC;QACVC,SAASuB;QACTf,YAAY,wBAACgB,YAAsCC,QAAAA;AACjD,iBAAOD,cAAc,IAAIE,YAAWD,GAAAA;QACtC,GAFY;QAGZb,QAAQ;UAAC;YACPe,OAAOC;YACPC,UAAU;UACZ;UAAGC;;MACL;;;;;;AGvEF,SAASC,OAAAA,YAAW;AAOb,IAAMC,gBAAN,MAAMA;EAPb,OAOaA;;;;;;;;;;EAQX,aAAaC,YACXC,iBACAC,WACAC,UACwB;AACxB,QAAI;AACF,UAAI,CAACF,gBAAgBG,YAAW,GAAI;AAClC,cAAM,IAAIC,MAAM,wBAAA;MAClB;AAEA,YAAMC,KAAKL,gBAAgBM,YAAW;AACtC,aAAO,MAAML,UAAUI,EAAAA;IACzB,SAASE,OAAO;AACdC,cAAQD,MAAM,8BAA8BA,KAAAA;AAC5C,aAAOL;IACT;EACF;;;;;;;;EASA,aAAaO,aACXT,iBACAU,YACAC,YAAoB,IACN;AACd,UAAMC,UAAe,CAAA;AACrB,UAAMP,KAAKL,gBAAgBM,YAAW;AAEtC,aAASO,IAAI,GAAGA,IAAIH,WAAWI,QAAQD,KAAKF,WAAW;AACrD,YAAMI,QAAQL,WAAWM,MAAMH,GAAGA,IAAIF,SAAAA;AACtC,YAAMM,eAAe,MAAMC,QAAQC,IACjCJ,MAAMK,IAAInB,CAAAA,cAAaA,UAAUI,EAAAA,CAAAA,CAAAA;AAEnCO,cAAQS,KAAI,GAAIJ,YAAAA;IAClB;AAEA,WAAOL;EACT;;;;;;;;;EAUA,aAAaU,iBACXtB,iBACAC,WACAsB,aAAqB,GACrBC,aAAqB,KACT;AACZ,QAAIC;AAEJ,aAASC,UAAU,GAAGA,WAAWH,YAAYG,WAAW;AACtD,UAAI;AACF,cAAMrB,KAAKL,gBAAgBM,YAAW;AACtC,eAAO,MAAML,UAAUI,EAAAA;MACzB,SAASE,OAAO;AACdkB,oBAAYlB;AAEZ,YAAImB,YAAYH,YAAY;AAC1B;QACF;AAEAf,gBAAQmB,KAAK,sCAAsCD,UAAU,CAAA,IAAKH,aAAa,CAAA,MAAOhB,KAAAA;AAGtF,cAAM,IAAIW,QAAQU,CAAAA,YAAWC,WAAWD,SAASJ,cAAcE,UAAU,EAAA,CAAA;MAC3E;IACF;AAEA,UAAMD;EACR;;;;;;;EAQA,aAAaK,YACXzB,IACA0B,WACkB;AAClB,QAAI;AACF,YAAMC,SAAS,MAAM3B,GAAG4B,QACtBC;;;6BAGqBH,SAAAA;UACnB;AAGJ,aAAOC,OAAO,CAAA,GAAIG,UAAqB;IACzC,SAAS5B,OAAO;AACdC,cAAQD,MAAM,oCAAoCA,KAAAA;AAClD,aAAO;IACT;EACF;;;;;;;EAQA,aAAa6B,kBACX/B,IACA0B,WACgB;AAChB,QAAI;AACF,YAAMC,SAAS,MAAM3B,GAAG4B,QACtBC;;;;;;;;2BAQmBH,SAAAA;kCACO;AAG5B,aAAOC;IACT,SAASzB,OAAO;AACdC,cAAQD,MAAM,kCAAkCA,KAAAA;AAChD,aAAO,CAAA;IACT;EACF;;;;;;EAOA,aAAa8B,mBACXhC,IACiB;AACjB,QAAI;AACF,YAAM2B,SAAS,MAAM3B,GAAG4B,QAAQC,sBAAqB;AACrD,aAAOF,OAAO,CAAA,GAAIM,WAAqB;IACzC,SAAS/B,OAAO;AACdC,cAAQD,MAAM,mCAAmCA,KAAAA;AACjD,aAAO;IACT;EACF;;;;;;EAOA,aAAagC,mBACXlC,IACc;AACd,QAAI;AACF,YAAM2B,SAAS,MAAM3B,GAAG4B,QAAQC;;;;;;;OAO/B;AAED,aAAOF,OAAO,CAAA,KAAM,CAAC;IACvB,SAASzB,OAAO;AACdC,cAAQD,MAAM,mCAAmCA,KAAAA;AACjD,aAAO,CAAC;IACV;EACF;;;;;;EAOA,aAAaiC,YACXxC,iBAOC;AACD,QAAI;AACF,YAAMG,cAAcH,gBAAgBG,YAAW;AAC/C,YAAMsC,iBAAiBzC,gBAAgB0C,kBAAiB;AAExD,YAAMrC,KAAKL,gBAAgBM,YAAW;AACtC,UAAI,CAACH,aAAa;AAChB,eAAO;UACLwC,WAAW;UACXxC,aAAa;UACbI,OAAO;QACT;MACF;AAEA,YAAM+B,UAAU,MAAM,KAAKD,mBAAmBhC,EAAAA;AAE9C,aAAO;QACLsC,WAAW;QACXxC,aAAa;QACbmC;QACAG;MACF;IACF,SAASlC,OAAO;AACd,aAAO;QACLoC,WAAW;QACXxC,aAAa;QACbI,OAAOA,iBAAiBH,QAAQG,MAAMqC,UAAU;MAClD;IACF;EACF;AACF;","names":["Injectable","Inject","drizzle","sql","postgres","DRIZZLE_DATABASE","DATAPAAS_CONFIG","DATAPAAS_LOGGER_SERVICE","Symbol","Observable","RequestContextService","parseConnectionInfo","connectionString","config","maxConnections","idleTimeout","connectionTimeout","ssl","logger","parsedConfigFromUrl","URL","temp","searchParams","delete","toString","error","Error","username","password","hostname","port","pathname","connectionConfig","decodeURIComponent","host","parseInt","max","idle_timeout","connect_timeout","onnotice","console","log","undefined","Logger","nestLogger","Logger","genPostgresProxy","sql","observable","requestContext","wrapQuery","query","then","Proxy","get","target","prop","receiver","onfulfilled","onrejected","sqlStr","strings","Array","span","startTrace","getContext","requestRootSpan","setAttribute","start","Date","now","result","isSetRoleSql","command","duration","parsedParams","JSON","stringify","operation","statement","string","duration_ms","result_count","length","error_message","log","source_type","paas_attributes_module","paas_parent_span","dropTrace","end","error","Error","message","String","Promise","reject","value","Reflect","bind","apply","thisArg","argArray","arg","cb","call","tx","queryStr","args","options","DrizzleDatabaseManager","db","currentConnectionInfo","isConnected","skipInitDbConnection","config","observable","requestContext","process","env","DEPRECATED_SKIP_INIT_DB_CONNECTION","onModuleInit","initialize","onModuleDestroy","disconnect","name","connectionString","connectionConfig","parseConnectionInfo","baseSql","postgres","sqlProxy","genPostgresProxy","drizzle","schema","logger","ssl","connectionTimeout","connect_timeout","idleTimeout","idle_timeout","maxConnections","max","error","Error","message","getDatabase","testConnection","execute","sql","console","reconnect","_forceRefresh","getConnectionInfo","isDatabaseConnected","transaction","callback","getConnectionPoolStatus","connectionInfo","Module","Injectable","DataPaasDatabaseService","databaseManager","getDatabase","getDatabaseManager","testConnection","reconnect","forceRefresh","getConnectionInfo","isConnected","isDatabaseConnected","transaction","callback","getConnectionPoolStatus","onModuleDestroy","disconnect","Injectable","Inject","session","SqlExecutionContextMiddleware","config","use","req","_res","next","roleSchema","userContext","roles","userRoles","join","userId","isSystemAccount","userType","sqls","filter","Boolean","session","runWithAuthContext","preSql","RequestContextService","OBSERVABLE_SERVICE","Observable","verifyDrizzlePatched","postgres","require","ok","getAuthContextStore","runWithAuthContext","Error","err","reason","message","String","DataPaasModule","configure","_consumer","forRoot","options","roleSchema","sslModeRequired","connectionString","url","URL","searchParams","get","config","schema","logger","timeout","maxConnections","idleTimeout","connectionTimeout","ssl","autoContext","providers","provide","DATAPAAS_CONFIG","useValue","DrizzleDatabaseManager","useClass","DataPaasDatabaseService","SqlExecutionContextMiddleware","DRIZZLE_DATABASE","useFactory","manager","initialize","getDatabase","inject","module","exports","global","forRootAsync","asyncProviders","createAsyncProviders","imports","createAsyncConfigProvider","args","normalizeConfig","DATAPAAS_LOGGER_SERVICE","observable","rtx","Observable","token","OBSERVABLE_SERVICE","optional","RequestContextService","sql","DatabaseUtils","safeExecute","databaseService","operation","fallback","isConnected","Error","db","getDatabase","error","console","batchExecute","operations","batchSize","results","i","length","batch","slice","batchResults","Promise","all","map","push","executeWithRetry","maxRetries","retryDelay","lastError","attempt","warn","resolve","setTimeout","tableExists","tableName","result","execute","sql","exists","getTableStructure","getDatabaseVersion","version","getConnectionStats","healthCheck","connectionInfo","getConnectionInfo","isHealthy","message"]}
+{"version":3,"sources":["../src/database/drizzle-manager.ts","../src/database/const.ts","../src/utils.ts","../src/database/postgres-proxy.ts","../src/database/database.module.ts","../src/database/database.service.ts","../src/database/sql-exection-context.middleware.ts","../src/database/database-utils.ts"],"sourcesContent":["import { Injectable, OnModuleInit, OnModuleDestroy, Inject } from '@nestjs/common';\nimport { drizzle, type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { sql } from 'drizzle-orm';\nimport postgres from 'postgres';\nimport { DATAPAAS_CONFIG, DATAPAAS_LOGGER_SERVICE } from './const';\nimport { Observable } from '@lark-apaas/nestjs-observable';\nimport { RequestContextService } from '@lark-apaas/nestjs-common';\n\nimport { type DataPaasConfig, type DatabaseConnectionInfo } from '../types';\nimport { parseConnectionInfo } from '../utils';\nimport { genPostgresProxy } from \"./postgres-proxy\";\n@Injectable()\nexport class DrizzleDatabaseManager implements OnModuleInit, OnModuleDestroy {\n  private db: PostgresJsDatabase | null = null;\n  private currentConnectionInfo: DatabaseConnectionInfo | null = null;\n  private isConnected: boolean = false;\n  private skipInitDbConnection: boolean = false;\n  constructor(\n    @Inject(DATAPAAS_CONFIG)\n    private readonly config: DataPaasConfig,\n    @Inject(DATAPAAS_LOGGER_SERVICE)\n    private readonly observable: Observable,\n    private readonly requestContext: RequestContextService,\n  ) { \n    this.skipInitDbConnection = !!process.env.DEPRECATED_SKIP_INIT_DB_CONNECTION;\n  }\n\n  async onModuleInit() {\n    await this.initialize();\n  }\n\n  async onModuleDestroy() {\n    await this.disconnect();\n  }\n\n  /**\n   * 初始化数据库连接\n   * @returns Drizzle 数据库实例\n   */\n  async initialize(): Promise<PostgresJsDatabase> {\n    if (this.skipInitDbConnection) {\n      this.isConnected = true;\n      this.db = {name: 'skipInitDbConnection'} as unknown as PostgresJsDatabase;\n      return this.db;\n    }\n    try {\n      const { connectionString, connectionConfig } = parseConnectionInfo(this.config.connectionString, this.config);\n      const baseSql = postgres(connectionString, connectionConfig);\n      const sqlProxy = genPostgresProxy(baseSql, this.observable, this.requestContext);\n      // 创建 Drizzle 实例\n      this.db = drizzle(sqlProxy, { schema: this.config.schema, logger: false });\n      this.currentConnectionInfo = {\n        connectionString,\n        connectionConfig: {\n          ssl: connectionConfig.ssl,\n          connectionTimeout: connectionConfig.connect_timeout,\n          idleTimeout: connectionConfig.idle_timeout,\n          maxConnections: connectionConfig.max,\n        },\n      };\n      this.isConnected = true;\n      return this.db;\n    } catch (error) {\n      this.isConnected = false;\n      throw new Error(`Failed to initialize database connection: ${error instanceof Error ? error.message : 'Unknown error'}`);\n    }\n  }\n\n  /**\n   * 获取数据库实例\n   * @returns Drizzle 数据库实例\n   */\n  getDatabase(): PostgresJsDatabase {\n    if (!this.db || !this.isConnected) {\n      throw new Error('Database not initialized. Call initialize() first.');\n    }\n    return this.db;\n  }\n\n  /**\n   * 测试数据库连接\n   * @returns 连接是否正常\n   */\n  async testConnection(): Promise<boolean> {\n    if (!this.db) {\n      return false;\n    }\n\n    try {\n      await this.db.execute(sql`SELECT 1`);\n      return true;\n    } catch (error) {\n      console.error('Database connection test failed:', error);\n      this.isConnected = false;\n      return false;\n    }\n  }\n\n  /**\n   * 重新连接数据库\n   * @param forceRefresh 是否强制刷新连接信息\n   * @returns 新的数据库实例\n   */\n  async reconnect(_forceRefresh: boolean = false): Promise<PostgresJsDatabase> {\n    await this.disconnect();\n    return this.initialize();\n  }\n\n  /**\n   * 断开数据库连接\n   */\n  async disconnect(): Promise<void> {\n    if (this.db) {\n      try {\n        // postgres-js 会自动处理连接关闭\n        this.db = null;\n        this.isConnected = false;\n        this.currentConnectionInfo = null;\n      } catch (error) {\n        console.error('Error disconnecting from database:', error);\n      }\n    }\n  }\n\n  /**\n   * 获取当前连接信息\n   * @returns 当前连接信息\n   */\n  getConnectionInfo(): DatabaseConnectionInfo | null {\n    return this.currentConnectionInfo;\n  }\n\n  /**\n   * 检查连接状态\n   * @returns 是否已连接\n   */\n  isDatabaseConnected(): boolean {\n    return this.isConnected;\n  }\n\n\n  /**\n   * 开始事务\n   * @param callback 事务回调函数\n   * @returns 事务结果\n   */\n  async transaction<T>(callback: (tx: PostgresJsDatabase) => Promise<T>): Promise<T> {\n    const db = this.getDatabase();\n    return db.transaction(callback);\n  }\n\n  /**\n   * 获取连接池状态\n   * @returns 连接池状态信息\n   */\n  getConnectionPoolStatus(): { isConnected: boolean; connectionInfo: DatabaseConnectionInfo | null } {\n    return {\n      isConnected: this.isConnected,\n      connectionInfo: this.currentConnectionInfo,\n    };\n  }\n}","export const DRIZZLE_DATABASE = 'DRIZZLE_DATABASE';\nexport const DATAPAAS_CONFIG = 'DATAPAAS_CONFIG';\nexport const DATAPAAS_LOGGER_SERVICE = Symbol('DATAPAAS_LOGGER_SERVICE');","import type postgres from 'postgres';\nimport { type Logger as DrizzleLogger } from 'drizzle-orm';\nimport { PostgresOptions } from './types';\n\ninterface ConnectionConfig {\n  maxConnections?: number;\n  idleTimeout?: number;\n  connectionTimeout?: number;\n  ssl?: PostgresOptions['ssl'];\n  logger?: boolean | DrizzleLogger;\n}\nexport function parseConnectionInfo(connectionString: string, config?: ConnectionConfig) {\n  const { maxConnections, idleTimeout, connectionTimeout, ssl, logger } = config || {};\n  // 创建 postgres 连接\n  let parsedConfigFromUrl: URL;\n  try {\n    parsedConfigFromUrl = new URL(connectionString);\n    const temp = new URL(connectionString);\n    temp.searchParams.delete('schema'); // 移除schema参数，避免 postgres 解析错误\n    connectionString = temp.toString();\n  } catch (error) {\n    throw new Error('Failed to parse connection info');\n  }\n  const { username, password, hostname, port, pathname } = parsedConfigFromUrl;\n  const connectionConfig = {\n    username: decodeURIComponent(username),\n    password: decodeURIComponent(password),\n    host: [hostname] as unknown as string,\n    port: [parseInt(port) || 5432] as unknown as number,\n    pathname: decodeURIComponent(pathname),\n    max: maxConnections || 1,\n    idle_timeout: idleTimeout || 20,\n    connect_timeout: connectionTimeout || 10,\n    ssl,\n    onnotice: logger ? console.log : undefined,\n  }\n  return { connectionString, connectionConfig };\n}","import type postgres from 'postgres';\nimport { Logger } from '@nestjs/common';\nimport { Span, Observable } from '@lark-apaas/nestjs-observable';\nimport { RequestContextService } from '@lark-apaas/nestjs-common';\n\nconst nestLogger = new Logger('Database');\n\n/**\n * 包装 postgres 实例以记录 SQL 和结果\n */\nexport function genPostgresProxy(sql: postgres.Sql, observable: Observable, requestContext: RequestContextService): postgres.Sql {\n  const wrapQuery = <T extends readonly (postgres.Row | undefined)[]>(query: postgres.PendingQuery<T>): postgres.PendingQuery<T> => {\n    if (query && typeof query.then === 'function') {\n      return new Proxy(query, {\n        get(target, prop, receiver) {\n          if (prop === 'then') {\n            return async (onfulfilled?: (value: postgres.RowList<T>) => unknown, onrejected?: (reason: unknown) => unknown) => {\n              const sqlStr = (query as unknown as Record<string, unknown>).strings instanceof Array ? (query as unknown as { strings: string[] }).strings[0] : '';\n              const span = observable.startTrace(sqlStr, requestContext?.getContext()?.requestRootSpan as Span);\n              const spanContext = {\n                traceId: span.spanContext().traceId,\n                spanId: span.spanContext().spanId\n              }\n              const logContext = {\n                source_type: 'platform',\n                paas_attributes_module: 'data',\n                paas_parent_span_context: spanContext,\n              }\n              span.setAttribute(\"module\", \"data\");\n              span.setAttribute(\"source_type\", \"platform\");\n\n              const start = Date.now();\n              try {\n                const result = await target;\n                const isSetRoleSql = result?.command === 'SET';\n\n                // rls SET ROLE 不上报日志\n                if (!isSetRoleSql) {\n                  const duration = Date.now() - start;\n\n                  const parsedParams = JSON.stringify({\n                    operation: result?.command,\n                    sql: result?.statement?.string,\n                    duration_ms: duration,\n                    result_count: result.length ?? 0,\n                    error_message: ''\n                  });\n\n                  nestLogger.log(parsedParams, logContext);\n                } else {\n                  observable.dropTrace(span);\n                }\n\n                span.end();\n                return onfulfilled ? onfulfilled(result) : result;\n              } catch (error) {\n                const duration = Date.now() - start;\n                const sqlStr = (query as unknown as Record<string, unknown>).strings instanceof Array ? (query as unknown as { strings: string[] }).strings[0] : '';\n                const parsedParams = JSON.stringify({\n                  operation: '',\n                  sql: sqlStr,\n                  duration_ms: duration,\n                  result_count: 0,\n                  error_message: error instanceof Error ? error.message : String(error)\n                });\n                nestLogger.log(parsedParams, logContext);\n\n                span.end();\n                return onrejected ? onrejected(error) : Promise.reject(error);\n              }\n            };\n          }\n          const value = Reflect.get(target, prop, receiver);\n          return typeof value === 'function' ? value.bind(receiver) : value;\n        },\n      }) as postgres.PendingQuery<T>;\n    }\n    return query;\n  };\n\n  return new Proxy(sql, {\n    apply(target, thisArg, argArray) {\n      const query = Reflect.apply(target, thisArg, argArray);\n      return wrapQuery(query);\n    },\n    get(target, prop, receiver) {\n      const value = Reflect.get(target, prop, receiver);\n      if (prop === 'begin' && typeof value === 'function') {\n        return <T>(arg: string | ((tx: postgres.TransactionSql) => T | Promise<T>), cb?: (tx: postgres.TransactionSql) => T | Promise<T>) => {\n          if (typeof arg === 'function') {\n            return value.call(target, (tx: postgres.TransactionSql) => arg(genPostgresProxy(tx as unknown as postgres.Sql, observable, requestContext) as unknown as postgres.TransactionSql));\n          }\n          if (typeof cb === 'function') {\n            return value.call(target, arg, (tx: postgres.TransactionSql) => cb(genPostgresProxy(tx as unknown as postgres.Sql, observable, requestContext) as unknown as postgres.TransactionSql));\n          }\n          return value.call(target, arg);\n        };\n      }\n      if (prop === 'unsafe' && typeof value === 'function') {\n        return <T extends readonly (postgres.Row | undefined)[] = postgres.Row[]>(queryStr: string, args?: unknown[], options?: postgres.UnsafeQueryOptions): postgres.PendingQuery<T> => {\n          const query = value.call(target, queryStr, args, options) as postgres.PendingQuery<T>;\n          return wrapQuery(query);\n        };\n      }\n      return typeof value === 'function' ? value.bind(receiver) : value;\n    },\n  }) as postgres.Sql;\n}","import {\n  Module,\n  DynamicModule,\n  Provider,\n  MiddlewareConsumer,\n  InjectionToken,\n  ModuleMetadata,\n} from '@nestjs/common';\nimport { DrizzleDatabaseManager } from './drizzle-manager';\nimport { DataPaasDatabaseService } from './database.service';\nimport { SqlExecutionContextMiddleware } from './sql-exection-context.middleware';\nimport { type DataPaasConfig } from '@/types';\nimport { DATAPAAS_CONFIG, DRIZZLE_DATABASE, DATAPAAS_LOGGER_SERVICE } from './const';\nexport { type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { RequestContextService, OBSERVABLE_SERVICE, ObservableService } from '@lark-apaas/nestjs-common';\nimport { Observable } from '@lark-apaas/nestjs-observable';\n\nexport interface DataPaasModuleOptions extends DataPaasConfig {}\n\n/**\n * 异步配置选项\n */\nexport interface DataPaasModuleAsyncOptions {\n  /**\n   * 需要导入的模块（例如 ConfigModule）\n   */\n  imports?: ModuleMetadata['imports'];\n  /**\n   * 使用工厂函数创建配置\n   */\n  useFactory?: (...args: unknown[]) => Promise<DataPaasConfig> | DataPaasConfig;\n  /**\n   * 工厂函数需要注入的依赖\n   */\n  inject?: InjectionToken[];\n}\n\nfunction verifyDrizzlePatched(): void {\n  try {\n    // 直接加载被 patch 的 CJS 模块并检查导出是否存在\n    // 注意：此路径与补丁文件一致：drizzle-orm/postgres-js/session.cjs\n    // 若未来包结构变化，请同步更新补丁与此校验\n    // eslint-disable-next-line @typescript-eslint/no-var-requires\n    const postgres = require('drizzle-orm/postgres-js');\n    const ok =\n      postgres &&\n      typeof postgres.getAuthContextStore === 'function' &&\n      typeof postgres.runWithAuthContext === 'function';\n    if (!ok) {\n      throw new Error(\n        'drizzle-orm 未检测到补丁导出：getAuthContextStore/runWithAuthContext',\n      );\n    }\n  } catch (err) {\n    const reason = err instanceof Error ? err.message : String(err);\n    throw new Error(`drizzle-orm 补丁未生效或版本不匹配：${reason}`);\n  }\n}\n\n// 在模块初始化阶段进行一次 drizzle-orm 补丁自检\nverifyDrizzlePatched();\n\n@Module({\n  providers: [{\n    provide: DATAPAAS_LOGGER_SERVICE,\n    useFactory: (observable: ObservableService | null, rtx: RequestContextService) => {\n      return observable ?? new Observable(rtx);\n    },\n    inject: [{\n      token: OBSERVABLE_SERVICE,\n      optional: true,\n    }, RequestContextService],\n  }],\n})\nexport class DataPaasModule {\n  configure(_consumer: MiddlewareConsumer) {}\n  /**\n   * 注册数据平台模块（同步方式）\n   * @param options 模块配置选项\n   * @returns 动态模块\n   */\n  static forRoot(options: DataPaasConfig): DynamicModule {\n    let roleSchema = '';\n    let sslModeRequired = false;\n    if (options.connectionString) {\n      try {\n        const url = new URL(options.connectionString);\n\n        roleSchema = url.searchParams.get(\n          'schema'\n        ) as string;\n        sslModeRequired = url.searchParams.get('sslmode') === \"require\";\n      } catch (err) {\n        roleSchema = '';\n      }\n    }\n    const config: DataPaasConfig = {\n      connectionString: options.connectionString,\n      schema: options.schema,\n      logger: options.logger || false,\n      timeout: options.timeout || 10000,\n      maxConnections: options.maxConnections || 1,\n      idleTimeout: options.idleTimeout || 20,\n      connectionTimeout: options.connectionTimeout || 10,\n      ssl: options.ssl || (sslModeRequired ? 'require' : false),\n      autoContext: options.autoContext == null ? true : !!options.autoContext,\n      roleSchema,\n    };\n\n    const providers: Provider[] = [\n      // 配置提供者 - 使用 useValue\n      {\n        provide: DATAPAAS_CONFIG,\n        useValue: config,\n      },\n      // 数据库管理器 - 使用 useClass，依赖注入连接服务和配置服务\n      {\n        provide: DrizzleDatabaseManager,\n        useClass: DrizzleDatabaseManager,\n      },\n      // 数据库服务 - 使用 useClass，依赖注入管理器\n      {\n        provide: DataPaasDatabaseService,\n        useClass: DataPaasDatabaseService,\n      },\n      {\n        provide: SqlExecutionContextMiddleware,\n        useClass: SqlExecutionContextMiddleware,\n      },\n      // Drizzle 数据库实例 - 使用 useFactory 获取管理器中的数据库实例\n      {\n        provide: DRIZZLE_DATABASE,\n        useFactory: async (manager: DrizzleDatabaseManager) => {\n          await manager.initialize();\n          return manager.getDatabase();\n        },\n        inject: [DrizzleDatabaseManager],\n      },\n    ];\n\n    return {\n      module: DataPaasModule,\n      providers,\n      exports: [\n        DrizzleDatabaseManager,\n        DataPaasDatabaseService,\n        SqlExecutionContextMiddleware,\n        DRIZZLE_DATABASE,\n        DATAPAAS_CONFIG,\n      ],\n      global: true,\n    };\n  }\n\n  /**\n   * 注册数据平台模块（异步方式）\n   * 支持从 ConfigModule 等异步源获取配置\n   * @param options 异步配置选项\n   * @returns 动态模块\n   * @example\n   * ```typescript\n   * DataPaasModule.forRootAsync({\n   *   imports: [ConfigModule],\n   *   useFactory: async (configService: ConfigService) => ({\n   *     connectionString: configService.get('DATABASE_URL'),\n   *   }),\n   *   inject: [ConfigService],\n   * })\n   * ```\n   */\n  static forRootAsync(options: DataPaasModuleAsyncOptions): DynamicModule {\n    const asyncProviders = this.createAsyncProviders(options);\n\n    const providers: Provider[] = [\n      ...asyncProviders,\n      // 数据库管理器 - 使用 useClass，依赖注入连接服务和配置服务\n      {\n        provide: DrizzleDatabaseManager,\n        useClass: DrizzleDatabaseManager,\n      },\n      // 数据库服务 - 使用 useClass，依赖注入管理器\n      {\n        provide: DataPaasDatabaseService,\n        useClass: DataPaasDatabaseService,\n      },\n      {\n        provide: SqlExecutionContextMiddleware,\n        useClass: SqlExecutionContextMiddleware,\n      },\n      // Drizzle 数据库实例 - 使用 useFactory 获取管理器中的数据库实例\n      {\n        provide: DRIZZLE_DATABASE,\n        useFactory: async (manager: DrizzleDatabaseManager) => {\n          await manager.initialize();\n          return manager.getDatabase();\n        },\n        inject: [DrizzleDatabaseManager],\n      },\n    ];\n\n    return {\n      module: DataPaasModule,\n      imports: options.imports || [],\n      providers,\n      exports: [\n        DrizzleDatabaseManager,\n        DataPaasDatabaseService,\n        SqlExecutionContextMiddleware,\n        DRIZZLE_DATABASE,\n        DATAPAAS_CONFIG,\n      ],\n      global: true,\n    };\n  }\n\n  /**\n   * 创建异步配置 providers\n   */\n  private static createAsyncProviders(\n    options: DataPaasModuleAsyncOptions\n  ): Provider[] {\n    if (options.useFactory) {\n      return [this.createAsyncConfigProvider(options)];\n    }\n\n    throw new Error('DataPaasModule.forRootAsync 需要指定 useFactory');\n  }\n\n  /**\n   * 创建异步配置 provider\n   */\n  private static createAsyncConfigProvider(\n    options: DataPaasModuleAsyncOptions\n  ): Provider {\n    if (options.useFactory) {\n      return {\n        provide: DATAPAAS_CONFIG,\n        useFactory: async (...args: unknown[]) => {\n          const config = await options.useFactory!(...args);\n          return this.normalizeConfig(config);\n        },\n        inject: options.inject || [],\n      };\n    }\n\n    throw new Error('Invalid async options');\n  }\n\n  /**\n   * 标准化配置，设置默认值\n   */\n  private static normalizeConfig(config: DataPaasConfig): DataPaasConfig {\n    let roleSchema = '';\n    let sslModeRequired = false;\n    if (config.connectionString) {\n      try {\n        const url = new URL(config.connectionString);\n        roleSchema = url.searchParams.get(\n          'schema'\n        ) as string;\n        sslModeRequired = url.searchParams.get('sslmode') === \"require\";\n      } catch (err) {\n        roleSchema = '';\n      }\n    }\n    return {\n      connectionString: config.connectionString,\n      schema: config.schema,\n      logger: config.logger || false,\n      timeout: config.timeout || 10000,\n      maxConnections: config.maxConnections || 1,\n      idleTimeout: config.idleTimeout || 20,\n      connectionTimeout: config.connectionTimeout || 10,\n      ssl: config.ssl || (sslModeRequired ? 'require' : false),\n      autoContext: config.autoContext == null ? true : !!config.autoContext,\n      roleSchema,\n    };\n  }\n}\n","import { Injectable, OnModuleDestroy } from '@nestjs/common';\nimport { type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { DrizzleDatabaseManager } from './drizzle-manager';\nimport { DatabaseConnectionInfo } from '../types';\n\n@Injectable()\nexport class DataPaasDatabaseService implements OnModuleDestroy {\n  constructor(\n    private readonly databaseManager: DrizzleDatabaseManager,\n  ) {}\n\n  /**\n   * 获取 Drizzle 数据库实例\n   * @returns Drizzle 数据库实例\n   */\n  getDatabase(): PostgresJsDatabase {\n    return this.databaseManager.getDatabase();\n  }\n\n  /**\n   * 获取数据库管理器\n   * @returns 数据库管理器实例\n   */\n  getDatabaseManager(): DrizzleDatabaseManager {\n    return this.databaseManager;\n  }\n\n\n  /**\n   * 测试数据库连接\n   * @returns 连接是否正常\n   */\n  async testConnection(): Promise<boolean> {\n    return this.databaseManager.testConnection();\n  }\n\n  /**\n   * 重新连接数据库\n   * @param forceRefresh 是否强制刷新连接信息\n   * @returns 新的数据库实例\n   */\n  async reconnect(forceRefresh: boolean = false): Promise<PostgresJsDatabase> {\n    return this.databaseManager.reconnect(forceRefresh);\n  }\n\n  /**\n   * 获取当前连接信息\n   * @returns 当前连接信息\n   */\n  getConnectionInfo(): DatabaseConnectionInfo | null {\n    return this.databaseManager.getConnectionInfo();\n  }\n\n  /**\n   * 检查连接状态\n   * @returns 是否已连接\n   */\n  isConnected(): boolean {\n    return this.databaseManager.isDatabaseConnected();\n  }\n\n  /**\n   * 开始事务\n   * @param callback 事务回调函数\n   * @returns 事务结果\n   */\n  async transaction<T>(callback: (tx: PostgresJsDatabase) => Promise<T>): Promise<T> {\n    return this.databaseManager.transaction(callback);\n  }\n\n  /**\n   * 获取连接池状态\n   * @returns 连接池状态信息\n   */\n  getConnectionPoolStatus(): { isConnected: boolean; connectionInfo: DatabaseConnectionInfo | null } {\n    return this.databaseManager.getConnectionPoolStatus();\n  }\n\n\n  /**\n   * 模块销毁时断开数据库连接\n   */\n  async onModuleDestroy(): Promise<void> {\n    await this.databaseManager.disconnect();\n  }\n}\n","import { Injectable, NestMiddleware, Inject } from '@nestjs/common';\nimport { Request, Response, NextFunction } from 'express';\nimport * as session from 'drizzle-orm/postgres-js';\nimport { DATAPAAS_CONFIG } from './const';\nimport { type DataPaasConfig } from '../types';\n\ninterface UserContext {\n  userId?: string;\n  tenantId?: number;\n  appId?: string;\n  // 用户角色（如果有）\n  userRoles?: string[];\n  // 是否为系统账号\n  isSystemAccount: boolean;\n  // 用户类型\n  userType?: string;\n}\n\n@Injectable()\nexport class SqlExecutionContextMiddleware implements NestMiddleware {\n  constructor(\n    @Inject(DATAPAAS_CONFIG)\n    private readonly config: DataPaasConfig,\n  ) {}\n\n  use(req: Request, _res: Response, next: NextFunction) {\n    const roleSchema = this.config.roleSchema ?? '';\n    const userContext =\n      (req as unknown as { userContext: UserContext }).userContext || {};\n    const roles = (userContext.userRoles || []).join(',');\n    const userId = userContext.userId;\n    const isSystemAccount = userContext.isSystemAccount;\n    const userType = userContext.userType;\n    const sqls = [\n      // 设置用户 ID, 匿名用户时设置为空字符串\n      userId\n        ? `SET LOCAL app.user_id = '${userId}'`\n        : `SET LOCAL app.user_id = ''`,\n      // 系统用户设置为service_role, 登录用户设置为 authenticated_xxx 角色，匿名用户设置为 anon_xxx 角色\n      isSystemAccount ? `SET LOCAL ROLE 'service_role_${roleSchema}'` :  (\n        userId\n          ? `SET LOCAL ROLE 'authenticated_${roleSchema}'`\n          : `SET LOCAL ROLE 'anon_${roleSchema}'`),\n      // 自定义角色\n      roles && `SET LOCAL app.role_ids = '${roles}'`,\n      // 用户类型\n      userType\n        ? `SET LOCAL app.user_type = '${userType}'`\n        : `SET LOCAL app.user_type = ''`,\n    ]\n      .filter(Boolean)\n      .join(';');\n\n    // 需要确认角色设置是否对后续 SQL 执行有影响，预期是只影响当前会话\n    session.runWithAuthContext({ preSql: sqls }, () => {\n      next();\n    });\n  }\n}\n","import { type PostgresJsDatabase } from 'drizzle-orm/postgres-js';\nimport { sql } from 'drizzle-orm';\n\nimport { DataPaasDatabaseService } from './database.service';\n\n/**\n * 数据库工具函数集合\n */\nexport class DatabaseUtils {\n  /**\n   * 安全执行数据库操作\n   * @param databaseService 数据库服务实例\n   * @param operation 数据库操作函数\n   * @param fallback 失败时的回退值\n   * @returns 操作结果或回退值\n   */\n  static async safeExecute<T>(\n    databaseService: DataPaasDatabaseService,\n    operation: (db: PostgresJsDatabase) => Promise<T>,\n    fallback?: T\n  ): Promise<T | undefined> {\n    try {\n      if (!databaseService.isConnected()) {\n        throw new Error('Database not connected');\n      }\n      \n      const db = databaseService.getDatabase();\n      return await operation(db);\n    } catch (error) {\n      console.error('Database operation failed:', error);\n      return fallback;\n    }\n  }\n\n  /**\n   * 批量执行数据库操作\n   * @param databaseService 数据库服务实例\n   * @param operations 操作数组\n   * @param batchSize 批处理大小\n   * @returns 所有操作的结果\n   */\n  static async batchExecute<T>(\n    databaseService: DataPaasDatabaseService,\n    operations: Array<(db: PostgresJsDatabase) => Promise<T>>,\n    batchSize: number = 10\n  ): Promise<T[]> {\n    const results: T[] = [];\n    const db = databaseService.getDatabase();\n\n    for (let i = 0; i < operations.length; i += batchSize) {\n      const batch = operations.slice(i, i + batchSize);\n      const batchResults = await Promise.all(\n        batch.map(operation => operation(db))\n      );\n      results.push(...batchResults);\n    }\n\n    return results;\n  }\n\n  /**\n   * 执行带重试的数据库操作\n   * @param databaseService 数据库服务实例\n   * @param operation 数据库操作函数\n   * @param maxRetries 最大重试次数\n   * @param retryDelay 重试延迟（毫秒）\n   * @returns 操作结果\n   */\n  static async executeWithRetry<T>(\n    databaseService: DataPaasDatabaseService,\n    operation: (db: PostgresJsDatabase) => Promise<T>,\n    maxRetries: number = 3,\n    retryDelay: number = 1000\n  ): Promise<T> {\n    let lastError: Error;\n\n    for (let attempt = 0; attempt <= maxRetries; attempt++) {\n      try {\n        const db = databaseService.getDatabase();\n        return await operation(db);\n      } catch (error) {\n        lastError = error as Error;\n        \n        if (attempt === maxRetries) {\n          break;\n        }\n\n        console.warn(`Database operation failed (attempt ${attempt + 1}/${maxRetries + 1}):`, error);\n        \n        // 等待后重试\n        await new Promise(resolve => setTimeout(resolve, retryDelay * (attempt + 1)));\n      }\n    }\n\n    throw lastError!;\n  }\n\n  /**\n   * 检查表是否存在\n   * @param databaseService 数据库服务实例\n   * @param tableName 表名\n   * @returns 表是否存在\n   */\n  static async tableExists(\n    db: PostgresJsDatabase,\n    tableName: string\n  ): Promise<boolean> {\n    try {\n      const result = await db.execute(\n        sql`SELECT EXISTS (\n          SELECT FROM information_schema.tables \n          WHERE table_schema = 'public' \n          AND table_name = ${tableName}\n        )`\n      );\n      \n      return result[0]?.exists as boolean || false;\n    } catch (error) {\n      console.error('Failed to check table existence:', error);\n      return false;\n    }\n  }\n\n  /**\n   * 获取表结构信息\n   * @param databaseService 数据库服务实例\n   * @param tableName 表名\n   * @returns 表结构信息\n   */\n  static async getTableStructure(\n    db: PostgresJsDatabase,\n    tableName: string\n  ): Promise<any[]> {\n    try {\n      const result = await db.execute(\n        sql`SELECT \n          column_name,\n          data_type,\n          is_nullable,\n          column_default,\n          character_maximum_length\n        FROM information_schema.columns \n        WHERE table_schema = 'public' \n        AND table_name = ${tableName}\n        ORDER BY ordinal_position`,\n      );\n      \n      return result;\n    } catch (error) {\n      console.error('Failed to get table structure:', error);\n      return [];\n    }\n  }\n\n  /**\n   * 获取数据库版本信息\n   * @param databaseService 数据库服务实例\n   * @returns 数据库版本\n   */\n  static async getDatabaseVersion(\n    db: PostgresJsDatabase\n  ): Promise<string> {\n    try {\n      const result = await db.execute(sql`SELECT version()`);\n      return result[0]?.version as string || 'Unknown';\n    } catch (error) {\n      console.error('Failed to get database version:', error);\n      return 'Unknown';\n    }\n  }\n\n  /**\n   * 获取连接统计信息\n   * @param databaseService 数据库服务实例\n   * @returns 连接统计信息\n   */\n  static async getConnectionStats(\n    db: PostgresJsDatabase\n  ): Promise<any> {\n    try {\n      const result = await db.execute(sql`\n        SELECT \n          count(*) as total_connections,\n          count(*) FILTER (WHERE state = 'active') as active_connections,\n          count(*) FILTER (WHERE state = 'idle') as idle_connections\n        FROM pg_stat_activity \n        WHERE datname = current_database()\n      `);\n      \n      return result[0] || {};\n    } catch (error) {\n      console.error('Failed to get connection stats:', error);\n      return {};\n    }\n  }\n\n  /**\n   * 健康检查\n   * @param databaseService 数据库服务实例\n   * @returns 健康状态\n   */\n  static async healthCheck(\n    databaseService: DataPaasDatabaseService\n  ): Promise<{\n    isHealthy: boolean;\n    isConnected: boolean;\n    version?: string;\n    connectionInfo?: any;\n    error?: string;\n  }> {\n    try {\n      const isConnected = databaseService.isConnected();\n      const connectionInfo = databaseService.getConnectionInfo();\n\n      const db = databaseService.getDatabase();\n      if (!isConnected) {\n        return {\n          isHealthy: false,\n          isConnected: false,\n          error: 'Database not connected'\n        };\n      }\n\n      const version = await this.getDatabaseVersion(db);\n      \n      return {\n        isHealthy: true,\n        isConnected: true,\n        version,\n        connectionInfo\n      };\n    } catch (error) {\n      return {\n        isHealthy: false,\n        isConnected: false,\n        error: error instanceof Error ? error.message : 'Unknown error'\n      };\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;AAAA,SAASA,YAA2CC,cAAc;AAClE,SAASC,eAAwC;AACjD,SAASC,WAAW;AACpB,OAAOC,cAAc;;;ACHd,IAAMC,mBAAmB;AACzB,IAAMC,kBAAkB;AACxB,IAAMC,0BAA0BC,uBAAO,yBAAA;;;ADG9C,SAASC,kBAAkB;AAC3B,SAASC,6BAA6B;;;AEK/B,SAASC,oBAAoBC,kBAA0BC,QAAyB;AACrF,QAAM,EAAEC,gBAAgBC,aAAaC,mBAAmBC,KAAKC,OAAM,IAAKL,UAAU,CAAC;AAEnF,MAAIM;AACJ,MAAI;AACFA,0BAAsB,IAAIC,IAAIR,gBAAAA;AAC9B,UAAMS,OAAO,IAAID,IAAIR,gBAAAA;AACrBS,SAAKC,aAAaC,OAAO,QAAA;AACzBX,uBAAmBS,KAAKG,SAAQ;EAClC,SAASC,OAAO;AACd,UAAM,IAAIC,MAAM,iCAAA;EAClB;AACA,QAAM,EAAEC,UAAUC,UAAUC,UAAUC,MAAMC,SAAQ,IAAKZ;AACzD,QAAMa,mBAAmB;IACvBL,UAAUM,mBAAmBN,QAAAA;IAC7BC,UAAUK,mBAAmBL,QAAAA;IAC7BM,MAAM;MAACL;;IACPC,MAAM;MAACK,SAASL,IAAAA,KAAS;;IACzBC,UAAUE,mBAAmBF,QAAAA;IAC7BK,KAAKtB,kBAAkB;IACvBuB,cAActB,eAAe;IAC7BuB,iBAAiBtB,qBAAqB;IACtCC;IACAsB,UAAUrB,SAASsB,QAAQC,MAAMC;EACnC;AACA,SAAO;IAAE9B;IAAkBoB;EAAiB;AAC9C;AA1BgBrB;;;ACVhB,SAASgC,cAAc;AAIvB,IAAMC,aAAa,IAAIC,OAAO,UAAA;AAKvB,SAASC,iBAAiBC,MAAmBC,YAAwBC,gBAAqC;AAC/G,QAAMC,YAAY,wBAAkDC,UAAAA;AAClE,QAAIA,SAAS,OAAOA,MAAMC,SAAS,YAAY;AAC7C,aAAO,IAAIC,MAAMF,OAAO;QACtBG,IAAIC,QAAQC,MAAMC,UAAQ;AACxB,cAAID,SAAS,QAAQ;AACnB,mBAAO,OAAOE,aAAuDC,eAAAA;AACnE,oBAAMC,SAAUT,MAA6CU,mBAAmBC,QAASX,MAA2CU,QAAQ,CAAA,IAAK;AACjJ,oBAAME,OAAOf,WAAWgB,WAAWJ,QAAQX,gBAAgBgB,WAAAA,GAAcC,eAAAA;AACzE,oBAAMC,cAAc;gBAClBC,SAASL,KAAKI,YAAW,EAAGC;gBAC5BC,QAAQN,KAAKI,YAAW,EAAGE;cAC7B;AACA,oBAAMC,aAAa;gBACjBC,aAAa;gBACbC,wBAAwB;gBACxBC,0BAA0BN;cAC5B;AACAJ,mBAAKW,aAAa,UAAU,MAAA;AAC5BX,mBAAKW,aAAa,eAAe,UAAA;AAEjC,oBAAMC,QAAQC,KAAKC,IAAG;AACtB,kBAAI;AACF,sBAAMC,SAAS,MAAMvB;AACrB,sBAAMwB,eAAeD,QAAQE,YAAY;AAGzC,oBAAI,CAACD,cAAc;AACjB,wBAAME,WAAWL,KAAKC,IAAG,IAAKF;AAE9B,wBAAMO,eAAeC,KAAKC,UAAU;oBAClCC,WAAWP,QAAQE;oBACnBjC,KAAK+B,QAAQQ,WAAWC;oBACxBC,aAAaP;oBACbQ,cAAcX,OAAOY,UAAU;oBAC/BC,eAAe;kBACjB,CAAA;AAEA/C,6BAAWgD,IAAIV,cAAcZ,UAAAA;gBAC/B,OAAO;AACLtB,6BAAW6C,UAAU9B,IAAAA;gBACvB;AAEAA,qBAAK+B,IAAG;AACR,uBAAOpC,cAAcA,YAAYoB,MAAAA,IAAUA;cAC7C,SAASiB,OAAO;AACd,sBAAMd,WAAWL,KAAKC,IAAG,IAAKF;AAC9B,sBAAMf,UAAUT,MAA6CU,mBAAmBC,QAASX,MAA2CU,QAAQ,CAAA,IAAK;AACjJ,sBAAMqB,eAAeC,KAAKC,UAAU;kBAClCC,WAAW;kBACXtC,KAAKa;kBACL4B,aAAaP;kBACbQ,cAAc;kBACdE,eAAeI,iBAAiBC,QAAQD,MAAME,UAAUC,OAAOH,KAAAA;gBACjE,CAAA;AACAnD,2BAAWgD,IAAIV,cAAcZ,UAAAA;AAE7BP,qBAAK+B,IAAG;AACR,uBAAOnC,aAAaA,WAAWoC,KAAAA,IAASI,QAAQC,OAAOL,KAAAA;cACzD;YACF;UACF;AACA,gBAAMM,QAAQC,QAAQhD,IAAIC,QAAQC,MAAMC,QAAAA;AACxC,iBAAO,OAAO4C,UAAU,aAAaA,MAAME,KAAK9C,QAAAA,IAAY4C;QAC9D;MACF,CAAA;IACF;AACA,WAAOlD;EACT,GAnEkB;AAqElB,SAAO,IAAIE,MAAMN,MAAK;IACpByD,MAAMjD,QAAQkD,SAASC,UAAQ;AAC7B,YAAMvD,QAAQmD,QAAQE,MAAMjD,QAAQkD,SAASC,QAAAA;AAC7C,aAAOxD,UAAUC,KAAAA;IACnB;IACAG,IAAIC,QAAQC,MAAMC,UAAQ;AACxB,YAAM4C,QAAQC,QAAQhD,IAAIC,QAAQC,MAAMC,QAAAA;AACxC,UAAID,SAAS,WAAW,OAAO6C,UAAU,YAAY;AACnD,eAAO,CAAIM,KAAiEC,OAAAA;AAC1E,cAAI,OAAOD,QAAQ,YAAY;AAC7B,mBAAON,MAAMQ,KAAKtD,QAAQ,CAACuD,OAAgCH,IAAI7D,iBAAiBgE,IAA+B9D,YAAYC,cAAAA,CAAAA,CAAAA;UAC7H;AACA,cAAI,OAAO2D,OAAO,YAAY;AAC5B,mBAAOP,MAAMQ,KAAKtD,QAAQoD,KAAK,CAACG,OAAgCF,GAAG9D,iBAAiBgE,IAA+B9D,YAAYC,cAAAA,CAAAA,CAAAA;UACjI;AACA,iBAAOoD,MAAMQ,KAAKtD,QAAQoD,GAAAA;QAC5B;MACF;AACA,UAAInD,SAAS,YAAY,OAAO6C,UAAU,YAAY;AACpD,eAAO,CAAmEU,UAAkBC,MAAkBC,YAAAA;AAC5G,gBAAM9D,QAAQkD,MAAMQ,KAAKtD,QAAQwD,UAAUC,MAAMC,OAAAA;AACjD,iBAAO/D,UAAUC,KAAAA;QACnB;MACF;AACA,aAAO,OAAOkD,UAAU,aAAaA,MAAME,KAAK9C,QAAAA,IAAY4C;IAC9D;EACF,CAAA;AACF;AAjGgBvD;;;;;;;;;;;;;;;;;;;;AHET,IAAMoE,yBAAN,MAAMA;SAAAA;;;;;;EACHC,KAAgC;EAChCC,wBAAuD;EACvDC,cAAuB;EACvBC,uBAAgC;EACxC,YAEmBC,QAEAC,YACAC,gBACjB;SAJiBF,SAAAA;SAEAC,aAAAA;SACAC,iBAAAA;AAEjB,SAAKH,uBAAuB,CAAC,CAACI,QAAQC,IAAIC;EAC5C;EAEA,MAAMC,eAAe;AACnB,UAAM,KAAKC,WAAU;EACvB;EAEA,MAAMC,kBAAkB;AACtB,UAAM,KAAKC,WAAU;EACvB;;;;;EAMA,MAAMF,aAA0C;AAC9C,QAAI,KAAKR,sBAAsB;AAC7B,WAAKD,cAAc;AACnB,WAAKF,KAAK;QAACc,MAAM;MAAsB;AACvC,aAAO,KAAKd;IACd;AACA,QAAI;AACF,YAAM,EAAEe,kBAAkBC,iBAAgB,IAAKC,oBAAoB,KAAKb,OAAOW,kBAAkB,KAAKX,MAAM;AAC5G,YAAMc,UAAUC,SAASJ,kBAAkBC,gBAAAA;AAC3C,YAAMI,WAAWC,iBAAiBH,SAAS,KAAKb,YAAY,KAAKC,cAAc;AAE/E,WAAKN,KAAKsB,QAAQF,UAAU;QAAEG,QAAQ,KAAKnB,OAAOmB;QAAQC,QAAQ;MAAM,CAAA;AACxE,WAAKvB,wBAAwB;QAC3Bc;QACAC,kBAAkB;UAChBS,KAAKT,iBAAiBS;UACtBC,mBAAmBV,iBAAiBW;UACpCC,aAAaZ,iBAAiBa;UAC9BC,gBAAgBd,iBAAiBe;QACnC;MACF;AACA,WAAK7B,cAAc;AACnB,aAAO,KAAKF;IACd,SAASgC,OAAO;AACd,WAAK9B,cAAc;AACnB,YAAM,IAAI+B,MAAM,6CAA6CD,iBAAiBC,QAAQD,MAAME,UAAU,eAAA,EAAiB;IACzH;EACF;;;;;EAMAC,cAAkC;AAChC,QAAI,CAAC,KAAKnC,MAAM,CAAC,KAAKE,aAAa;AACjC,YAAM,IAAI+B,MAAM,oDAAA;IAClB;AACA,WAAO,KAAKjC;EACd;;;;;EAMA,MAAMoC,iBAAmC;AACvC,QAAI,CAAC,KAAKpC,IAAI;AACZ,aAAO;IACT;AAEA,QAAI;AACF,YAAM,KAAKA,GAAGqC,QAAQC,aAAa;AACnC,aAAO;IACT,SAASN,OAAO;AACdO,cAAQP,MAAM,oCAAoCA,KAAAA;AAClD,WAAK9B,cAAc;AACnB,aAAO;IACT;EACF;;;;;;EAOA,MAAMsC,UAAUC,gBAAyB,OAAoC;AAC3E,UAAM,KAAK5B,WAAU;AACrB,WAAO,KAAKF,WAAU;EACxB;;;;EAKA,MAAME,aAA4B;AAChC,QAAI,KAAKb,IAAI;AACX,UAAI;AAEF,aAAKA,KAAK;AACV,aAAKE,cAAc;AACnB,aAAKD,wBAAwB;MAC/B,SAAS+B,OAAO;AACdO,gBAAQP,MAAM,sCAAsCA,KAAAA;MACtD;IACF;EACF;;;;;EAMAU,oBAAmD;AACjD,WAAO,KAAKzC;EACd;;;;;EAMA0C,sBAA+B;AAC7B,WAAO,KAAKzC;EACd;;;;;;EAQA,MAAM0C,YAAeC,UAA8D;AACjF,UAAM7C,KAAK,KAAKmC,YAAW;AAC3B,WAAOnC,GAAG4C,YAAYC,QAAAA;EACxB;;;;;EAMAC,0BAAmG;AACjG,WAAO;MACL5C,aAAa,KAAKA;MAClB6C,gBAAgB,KAAK9C;IACvB;EACF;AACF;;;;;;;;;;;;;;AIjKA,SACE+C,cAMK;;;ACPP,SAASC,cAAAA,mBAAmC;;;;;;;;;;;;AAMrC,IAAMC,0BAAN,MAAMA;SAAAA;;;;EACX,YACmBC,iBACjB;SADiBA,kBAAAA;EAChB;;;;;EAMHC,cAAkC;AAChC,WAAO,KAAKD,gBAAgBC,YAAW;EACzC;;;;;EAMAC,qBAA6C;AAC3C,WAAO,KAAKF;EACd;;;;;EAOA,MAAMG,iBAAmC;AACvC,WAAO,KAAKH,gBAAgBG,eAAc;EAC5C;;;;;;EAOA,MAAMC,UAAUC,eAAwB,OAAoC;AAC1E,WAAO,KAAKL,gBAAgBI,UAAUC,YAAAA;EACxC;;;;;EAMAC,oBAAmD;AACjD,WAAO,KAAKN,gBAAgBM,kBAAiB;EAC/C;;;;;EAMAC,cAAuB;AACrB,WAAO,KAAKP,gBAAgBQ,oBAAmB;EACjD;;;;;;EAOA,MAAMC,YAAeC,UAA8D;AACjF,WAAO,KAAKV,gBAAgBS,YAAYC,QAAAA;EAC1C;;;;;EAMAC,0BAAmG;AACjG,WAAO,KAAKX,gBAAgBW,wBAAuB;EACrD;;;;EAMA,MAAMC,kBAAiC;AACrC,UAAM,KAAKZ,gBAAgBa,WAAU;EACvC;AACF;;;;;;;;;;ACrFA,SAASC,cAAAA,aAA4BC,UAAAA,eAAc;AAEnD,YAAYC,aAAa;;;;;;;;;;;;;;;;;;AAiBlB,IAAMC,gCAAN,MAAMA;SAAAA;;;;EACX,YAEmBC,QACjB;SADiBA,SAAAA;EAChB;EAEHC,IAAIC,KAAcC,MAAgBC,MAAoB;AACpD,UAAMC,aAAa,KAAKL,OAAOK,cAAc;AAC7C,UAAMC,cACHJ,IAAgDI,eAAe,CAAC;AACnE,UAAMC,SAASD,YAAYE,aAAa,CAAA,GAAIC,KAAK,GAAA;AACjD,UAAMC,SAASJ,YAAYI;AAC3B,UAAMC,kBAAkBL,YAAYK;AACpC,UAAMC,WAAWN,YAAYM;AAC7B,UAAMC,OAAO;;MAEXH,SACI,4BAA4BA,MAAAA,MAC5B;;MAEJC,kBAAkB,gCAAgCN,UAAAA,MAChDK,SACI,iCAAiCL,UAAAA,MACjC,wBAAwBA,UAAAA;;MAE9BE,SAAS,6BAA6BA,KAAAA;;MAEtCK,WACI,8BAA8BA,QAAAA,MAC9B;MAEHE,OAAOC,OAAAA,EACPN,KAAK,GAAA;AAGRO,IAAQC,2BAAmB;MAAEC,QAAQL;IAAK,GAAG,MAAA;AAC3CT,WAAAA;IACF,CAAA;EACF;AACF;;;;;;;;;;;AF5CA,SAASe,yBAAAA,wBAAuBC,0BAA6C;AAC7E,SAASC,cAAAA,mBAAkB;;;;;;;;AAsB3B,SAASC,uBAAAA;AACP,MAAI;AAKF,UAAMC,YAAWC,UAAQ,yBAAA;AACzB,UAAMC,KACJF,aACA,OAAOA,UAASG,wBAAwB,cACxC,OAAOH,UAASI,uBAAuB;AACzC,QAAI,CAACF,IAAI;AACP,YAAM,IAAIG,MACR,0GAAA;IAEJ;EACF,SAASC,KAAK;AACZ,UAAMC,SAASD,eAAeD,QAAQC,IAAIE,UAAUC,OAAOH,GAAAA;AAC3D,UAAM,IAAID,MAAM,uFAA2BE,MAAAA,EAAQ;EACrD;AACF;AApBSR;AAuBTA,qBAAAA;AAcO,IAAMW,iBAAN,MAAMA,gBAAAA;SAAAA;;;EACXC,UAAUC,WAA+B;EAAC;;;;;;EAM1C,OAAOC,QAAQC,SAAwC;AACrD,QAAIC,aAAa;AACjB,QAAIC,kBAAkB;AACtB,QAAIF,QAAQG,kBAAkB;AAC5B,UAAI;AACF,cAAMC,MAAM,IAAIC,IAAIL,QAAQG,gBAAgB;AAE5CF,qBAAaG,IAAIE,aAAaC,IAC5B,QAAA;AAEFL,0BAAkBE,IAAIE,aAAaC,IAAI,SAAA,MAAe;MACxD,SAASf,KAAK;AACZS,qBAAa;MACf;IACF;AACA,UAAMO,SAAyB;MAC7BL,kBAAkBH,QAAQG;MAC1BM,QAAQT,QAAQS;MAChBC,QAAQV,QAAQU,UAAU;MAC1BC,SAASX,QAAQW,WAAW;MAC5BC,gBAAgBZ,QAAQY,kBAAkB;MAC1CC,aAAab,QAAQa,eAAe;MACpCC,mBAAmBd,QAAQc,qBAAqB;MAChDC,KAAKf,QAAQe,QAAQb,kBAAkB,YAAY;MACnDc,aAAahB,QAAQgB,eAAe,OAAO,OAAO,CAAC,CAAChB,QAAQgB;MAC5Df;IACF;AAEA,UAAMgB,YAAwB;;MAE5B;QACEC,SAASC;QACTC,UAAUZ;MACZ;;MAEA;QACEU,SAASG;QACTC,UAAUD;MACZ;;MAEA;QACEH,SAASK;QACTD,UAAUC;MACZ;MACA;QACEL,SAASM;QACTF,UAAUE;MACZ;;MAEA;QACEN,SAASO;QACTC,YAAY,8BAAOC,YAAAA;AACjB,gBAAMA,QAAQC,WAAU;AACxB,iBAAOD,QAAQE,YAAW;QAC5B,GAHY;QAIZC,QAAQ;UAACT;;MACX;;AAGF,WAAO;MACLU,QAAQnC;MACRqB;MACAe,SAAS;QACPX;QACAE;QACAC;QACAC;QACAN;;MAEFc,QAAQ;IACV;EACF;;;;;;;;;;;;;;;;;EAkBA,OAAOC,aAAalC,SAAoD;AACtE,UAAMmC,iBAAiB,KAAKC,qBAAqBpC,OAAAA;AAEjD,UAAMiB,YAAwB;SACzBkB;;MAEH;QACEjB,SAASG;QACTC,UAAUD;MACZ;;MAEA;QACEH,SAASK;QACTD,UAAUC;MACZ;MACA;QACEL,SAASM;QACTF,UAAUE;MACZ;;MAEA;QACEN,SAASO;QACTC,YAAY,8BAAOC,YAAAA;AACjB,gBAAMA,QAAQC,WAAU;AACxB,iBAAOD,QAAQE,YAAW;QAC5B,GAHY;QAIZC,QAAQ;UAACT;;MACX;;AAGF,WAAO;MACLU,QAAQnC;MACRyC,SAASrC,QAAQqC,WAAW,CAAA;MAC5BpB;MACAe,SAAS;QACPX;QACAE;QACAC;QACAC;QACAN;;MAEFc,QAAQ;IACV;EACF;;;;EAKA,OAAeG,qBACbpC,SACY;AACZ,QAAIA,QAAQ0B,YAAY;AACtB,aAAO;QAAC,KAAKY,0BAA0BtC,OAAAA;;IACzC;AAEA,UAAM,IAAIT,MAAM,iEAAA;EAClB;;;;EAKA,OAAe+C,0BACbtC,SACU;AACV,QAAIA,QAAQ0B,YAAY;AACtB,aAAO;QACLR,SAASC;QACTO,YAAY,iCAAUa,SAAAA;AACpB,gBAAM/B,SAAS,MAAMR,QAAQ0B,WAAU,GAAKa,IAAAA;AAC5C,iBAAO,KAAKC,gBAAgBhC,MAAAA;QAC9B,GAHY;QAIZsB,QAAQ9B,QAAQ8B,UAAU,CAAA;MAC5B;IACF;AAEA,UAAM,IAAIvC,MAAM,uBAAA;EAClB;;;;EAKA,OAAeiD,gBAAgBhC,QAAwC;AACrE,QAAIP,aAAa;AACjB,QAAIC,kBAAkB;AACtB,QAAIM,OAAOL,kBAAkB;AAC3B,UAAI;AACF,cAAMC,MAAM,IAAIC,IAAIG,OAAOL,gBAAgB;AAC3CF,qBAAaG,IAAIE,aAAaC,IAC5B,QAAA;AAEFL,0BAAkBE,IAAIE,aAAaC,IAAI,SAAA,MAAe;MACxD,SAASf,KAAK;AACZS,qBAAa;MACf;IACF;AACA,WAAO;MACLE,kBAAkBK,OAAOL;MACzBM,QAAQD,OAAOC;MACfC,QAAQF,OAAOE,UAAU;MACzBC,SAASH,OAAOG,WAAW;MAC3BC,gBAAgBJ,OAAOI,kBAAkB;MACzCC,aAAaL,OAAOK,eAAe;MACnCC,mBAAmBN,OAAOM,qBAAqB;MAC/CC,KAAKP,OAAOO,QAAQb,kBAAkB,YAAY;MAClDc,aAAaR,OAAOQ,eAAe,OAAO,OAAO,CAAC,CAACR,OAAOQ;MAC1Df;IACF;EACF;AACF;;;IAvNEgB,WAAW;MAAC;QACVC,SAASuB;QACTf,YAAY,wBAACgB,YAAsCC,QAAAA;AACjD,iBAAOD,cAAc,IAAIE,YAAWD,GAAAA;QACtC,GAFY;QAGZb,QAAQ;UAAC;YACPe,OAAOC;YACPC,UAAU;UACZ;UAAGC;;MACL;;;;;;AGvEF,SAASC,OAAAA,YAAW;AAOb,IAAMC,gBAAN,MAAMA;EAPb,OAOaA;;;;;;;;;;EAQX,aAAaC,YACXC,iBACAC,WACAC,UACwB;AACxB,QAAI;AACF,UAAI,CAACF,gBAAgBG,YAAW,GAAI;AAClC,cAAM,IAAIC,MAAM,wBAAA;MAClB;AAEA,YAAMC,KAAKL,gBAAgBM,YAAW;AACtC,aAAO,MAAML,UAAUI,EAAAA;IACzB,SAASE,OAAO;AACdC,cAAQD,MAAM,8BAA8BA,KAAAA;AAC5C,aAAOL;IACT;EACF;;;;;;;;EASA,aAAaO,aACXT,iBACAU,YACAC,YAAoB,IACN;AACd,UAAMC,UAAe,CAAA;AACrB,UAAMP,KAAKL,gBAAgBM,YAAW;AAEtC,aAASO,IAAI,GAAGA,IAAIH,WAAWI,QAAQD,KAAKF,WAAW;AACrD,YAAMI,QAAQL,WAAWM,MAAMH,GAAGA,IAAIF,SAAAA;AACtC,YAAMM,eAAe,MAAMC,QAAQC,IACjCJ,MAAMK,IAAInB,CAAAA,cAAaA,UAAUI,EAAAA,CAAAA,CAAAA;AAEnCO,cAAQS,KAAI,GAAIJ,YAAAA;IAClB;AAEA,WAAOL;EACT;;;;;;;;;EAUA,aAAaU,iBACXtB,iBACAC,WACAsB,aAAqB,GACrBC,aAAqB,KACT;AACZ,QAAIC;AAEJ,aAASC,UAAU,GAAGA,WAAWH,YAAYG,WAAW;AACtD,UAAI;AACF,cAAMrB,KAAKL,gBAAgBM,YAAW;AACtC,eAAO,MAAML,UAAUI,EAAAA;MACzB,SAASE,OAAO;AACdkB,oBAAYlB;AAEZ,YAAImB,YAAYH,YAAY;AAC1B;QACF;AAEAf,gBAAQmB,KAAK,sCAAsCD,UAAU,CAAA,IAAKH,aAAa,CAAA,MAAOhB,KAAAA;AAGtF,cAAM,IAAIW,QAAQU,CAAAA,YAAWC,WAAWD,SAASJ,cAAcE,UAAU,EAAA,CAAA;MAC3E;IACF;AAEA,UAAMD;EACR;;;;;;;EAQA,aAAaK,YACXzB,IACA0B,WACkB;AAClB,QAAI;AACF,YAAMC,SAAS,MAAM3B,GAAG4B,QACtBC;;;6BAGqBH,SAAAA;UACnB;AAGJ,aAAOC,OAAO,CAAA,GAAIG,UAAqB;IACzC,SAAS5B,OAAO;AACdC,cAAQD,MAAM,oCAAoCA,KAAAA;AAClD,aAAO;IACT;EACF;;;;;;;EAQA,aAAa6B,kBACX/B,IACA0B,WACgB;AAChB,QAAI;AACF,YAAMC,SAAS,MAAM3B,GAAG4B,QACtBC;;;;;;;;2BAQmBH,SAAAA;kCACO;AAG5B,aAAOC;IACT,SAASzB,OAAO;AACdC,cAAQD,MAAM,kCAAkCA,KAAAA;AAChD,aAAO,CAAA;IACT;EACF;;;;;;EAOA,aAAa8B,mBACXhC,IACiB;AACjB,QAAI;AACF,YAAM2B,SAAS,MAAM3B,GAAG4B,QAAQC,sBAAqB;AACrD,aAAOF,OAAO,CAAA,GAAIM,WAAqB;IACzC,SAAS/B,OAAO;AACdC,cAAQD,MAAM,mCAAmCA,KAAAA;AACjD,aAAO;IACT;EACF;;;;;;EAOA,aAAagC,mBACXlC,IACc;AACd,QAAI;AACF,YAAM2B,SAAS,MAAM3B,GAAG4B,QAAQC;;;;;;;OAO/B;AAED,aAAOF,OAAO,CAAA,KAAM,CAAC;IACvB,SAASzB,OAAO;AACdC,cAAQD,MAAM,mCAAmCA,KAAAA;AACjD,aAAO,CAAC;IACV;EACF;;;;;;EAOA,aAAaiC,YACXxC,iBAOC;AACD,QAAI;AACF,YAAMG,cAAcH,gBAAgBG,YAAW;AAC/C,YAAMsC,iBAAiBzC,gBAAgB0C,kBAAiB;AAExD,YAAMrC,KAAKL,gBAAgBM,YAAW;AACtC,UAAI,CAACH,aAAa;AAChB,eAAO;UACLwC,WAAW;UACXxC,aAAa;UACbI,OAAO;QACT;MACF;AAEA,YAAM+B,UAAU,MAAM,KAAKD,mBAAmBhC,EAAAA;AAE9C,aAAO;QACLsC,WAAW;QACXxC,aAAa;QACbmC;QACAG;MACF;IACF,SAASlC,OAAO;AACd,aAAO;QACLoC,WAAW;QACXxC,aAAa;QACbI,OAAOA,iBAAiBH,QAAQG,MAAMqC,UAAU;MAClD;IACF;EACF;AACF;","names":["Injectable","Inject","drizzle","sql","postgres","DRIZZLE_DATABASE","DATAPAAS_CONFIG","DATAPAAS_LOGGER_SERVICE","Symbol","Observable","RequestContextService","parseConnectionInfo","connectionString","config","maxConnections","idleTimeout","connectionTimeout","ssl","logger","parsedConfigFromUrl","URL","temp","searchParams","delete","toString","error","Error","username","password","hostname","port","pathname","connectionConfig","decodeURIComponent","host","parseInt","max","idle_timeout","connect_timeout","onnotice","console","log","undefined","Logger","nestLogger","Logger","genPostgresProxy","sql","observable","requestContext","wrapQuery","query","then","Proxy","get","target","prop","receiver","onfulfilled","onrejected","sqlStr","strings","Array","span","startTrace","getContext","requestRootSpan","spanContext","traceId","spanId","logContext","source_type","paas_attributes_module","paas_parent_span_context","setAttribute","start","Date","now","result","isSetRoleSql","command","duration","parsedParams","JSON","stringify","operation","statement","string","duration_ms","result_count","length","error_message","log","dropTrace","end","error","Error","message","String","Promise","reject","value","Reflect","bind","apply","thisArg","argArray","arg","cb","call","tx","queryStr","args","options","DrizzleDatabaseManager","db","currentConnectionInfo","isConnected","skipInitDbConnection","config","observable","requestContext","process","env","DEPRECATED_SKIP_INIT_DB_CONNECTION","onModuleInit","initialize","onModuleDestroy","disconnect","name","connectionString","connectionConfig","parseConnectionInfo","baseSql","postgres","sqlProxy","genPostgresProxy","drizzle","schema","logger","ssl","connectionTimeout","connect_timeout","idleTimeout","idle_timeout","maxConnections","max","error","Error","message","getDatabase","testConnection","execute","sql","console","reconnect","_forceRefresh","getConnectionInfo","isDatabaseConnected","transaction","callback","getConnectionPoolStatus","connectionInfo","Module","Injectable","DataPaasDatabaseService","databaseManager","getDatabase","getDatabaseManager","testConnection","reconnect","forceRefresh","getConnectionInfo","isConnected","isDatabaseConnected","transaction","callback","getConnectionPoolStatus","onModuleDestroy","disconnect","Injectable","Inject","session","SqlExecutionContextMiddleware","config","use","req","_res","next","roleSchema","userContext","roles","userRoles","join","userId","isSystemAccount","userType","sqls","filter","Boolean","session","runWithAuthContext","preSql","RequestContextService","OBSERVABLE_SERVICE","Observable","verifyDrizzlePatched","postgres","require","ok","getAuthContextStore","runWithAuthContext","Error","err","reason","message","String","DataPaasModule","configure","_consumer","forRoot","options","roleSchema","sslModeRequired","connectionString","url","URL","searchParams","get","config","schema","logger","timeout","maxConnections","idleTimeout","connectionTimeout","ssl","autoContext","providers","provide","DATAPAAS_CONFIG","useValue","DrizzleDatabaseManager","useClass","DataPaasDatabaseService","SqlExecutionContextMiddleware","DRIZZLE_DATABASE","useFactory","manager","initialize","getDatabase","inject","module","exports","global","forRootAsync","asyncProviders","createAsyncProviders","imports","createAsyncConfigProvider","args","normalizeConfig","DATAPAAS_LOGGER_SERVICE","observable","rtx","Observable","token","OBSERVABLE_SERVICE","optional","RequestContextService","sql","DatabaseUtils","safeExecute","databaseService","operation","fallback","isConnected","Error","db","getDatabase","error","console","batchExecute","operations","batchSize","results","i","length","batch","slice","batchResults","Promise","all","map","push","executeWithRetry","maxRetries","retryDelay","lastError","attempt","warn","resolve","setTimeout","tableExists","tableName","result","execute","sql","exists","getTableStructure","getDatabaseVersion","version","getConnectionStats","healthCheck","connectionInfo","getConnectionInfo","isHealthy","message"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lark-apaas/nestjs-datapaas",
-  "version": "1.0.8-alpha.5",
+  "version": "1.0.8-alpha.7",
   "description": "FullStack Nestjs DataPaas",
   "type": "module",
   "main": "./dist/index.js",
@@ -66,7 +66,7 @@
   },
   "peerDependencies": {
     "@lark-apaas/nestjs-common": "0.1.2-alpha.2",
-    "@lark-apaas/nestjs-observable": "0.0.3-alpha.3",
+    "@lark-apaas/nestjs-observable": "0.0.3-alpha.4",
     "@nestjs/common": "^10.0.0",
     "@nestjs/core": "^10.0.0",
     "drizzle-orm": "0.44.6"