opcode-pg-memory 2.2.8 → 2.3.0

package/dist/src/db/init-db.js

@@ -0,0 +1,545 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DatabaseInitializer = exports.DEFAULT_DB_CONFIG = void 0;
+exports.getDatabaseInitializer = getDatabaseInitializer;
+exports.initializeDatabase = initializeDatabase;
+exports.closeDatabase = closeDatabase;
+const pg_1 = require("pg");
+const pgvector_1 = __importDefault(require("pgvector"));
+const logger_1 = require("../services/logger");
+exports.DEFAULT_DB_CONFIG = {
+    host: process.env.PG_HOST || 'localhost',
+    port: parseInt(process.env.PG_PORT || '5432', 10),
+    database: process.env.PG_DATABASE || 'opencode_memory',
+    user: process.env.PG_USER || 'opencode',
+    password: process.env.PG_PASSWORD || '',
+    ssl: process.env.PG_SSL === 'true',
+    maxConnections: parseInt(process.env.PG_MAX_CONNECTIONS || '20', 10)
+};
+class DatabaseInitializer {
+    pool = null;
+    config;
+    logger = (0, logger_1.createLogger)('init-db');
+    constructor(config = {}) {
+        this.config = { ...exports.DEFAULT_DB_CONFIG, ...config };
+    }
+    async initialize() {
+        if (this.pool) {
+            return this.pool;
+        }
+        this.pool = new pg_1.Pool({
+            host: this.config.host,
+            port: this.config.port,
+            database: this.config.database,
+            user: this.config.user,
+            password: this.config.password,
+            ssl: this.config.ssl,
+            max: this.config.maxConnections
+        });
+        // 注册 pgvector 类型
+        // @ts-ignore - pgvector types may vary
+        pgvector_1.default.registerTypes?.(this.pool);
+        // 测试连接
+        await this.testConnection();
+        // 初始化数据库结构
+        await this.setupDatabase();
+        return this.pool;
+    }
+    async testConnection() {
+        try {
+            const client = await this.pool.connect();
+            const result = await client.query('SELECT NOW() as now');
+            client.release();
+            this.logger.info('Database connected:', result.rows[0].now);
+        }
+        catch (error) {
+            this.logger.error('Database connection failed:', error);
+            throw new Error(`Failed to connect to PostgreSQL: ${error}`);
+        }
+    }
+    async setupDatabase() {
+        const client = await this.pool.connect();
+        try {
+            await client.query('BEGIN');
+            // 1. 启用 pgvector 扩展
+            await this.createExtensions(client);
+            // 2. 创建枚举类型
+            await this.createEnums(client);
+            // 3. 创建表
+            await this.createTables(client);
+            // 4. 迁移旧列名（session_id → session_map_id）
+            await this.migrateLegacyColumnNames(client);
+            // 5. 创建索引
+            await this.createIndexes(client);
+            // 6. 迁移旧 sessions 数据到 session_map
+            await this.migrateSessionsData(client);
+            // 6. 初始化 OmO 适配 Schema（如果启用）
+            await this.initializeOmOSchema(client);
+            await client.query('COMMIT');
+            this.logger.info('Database schema initialized successfully');
+        }
+        catch (error) {
+            await client.query('ROLLBACK');
+            this.logger.error('Database setup failed:', error);
+            throw error;
+        }
+        finally {
+            client.release();
+        }
+    }
+    async createExtensions(client) {
+        await client.query(`
+      CREATE EXTENSION IF NOT EXISTS vector;
+      CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
+    `);
+        this.logger.info('Extensions created');
+    }
+    async createEnums(client) {
+        await client.query(`
+      DO $$
+      BEGIN
+        IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'entity_tier') THEN
+          CREATE TYPE entity_tier AS ENUM ('permanent', 'project', 'session');
+        END IF;
+        
+        IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'relation_type') THEN
+          CREATE TYPE relation_type AS ENUM ('belongs_to', 'depends_on', 'references', 'implements', 'uses', 'custom');
+        END IF;
+      END $$;
+    `);
+        this.logger.info('Enums created');
+    }
+    async createTables(client) {
+        // ── session_map 表（替代旧 sessions 表，旧表保留为历史数据） ──
+        await client.query(`
+      CREATE TABLE IF NOT EXISTS session_map (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        opencode_session_id VARCHAR(255) UNIQUE NOT NULL,
+        omo_task_id VARCHAR(255),
+        project_id VARCHAR(255),
+        model_context_limit INTEGER NOT NULL DEFAULT 128000,
+        created_at TIMESTAMPTZ DEFAULT NOW(),
+        last_active_at TIMESTAMPTZ DEFAULT NOW(),
+        metadata JSONB DEFAULT '{}'
+      );
+    `);
+        // ── topic_segments 表（NEW：会话内话题分段） ──
+        await client.query(`
+      CREATE TABLE IF NOT EXISTS topic_segments (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        session_map_id UUID NOT NULL REFERENCES session_map(id) ON DELETE CASCADE,
+        segment_index INTEGER NOT NULL,
+        summary TEXT,
+        embedding vector(1536),
+        start_message_external_id VARCHAR(255),
+        end_message_external_id VARCHAR(255),
+        created_at TIMESTAMPTZ DEFAULT NOW(),
+        closed_at TIMESTAMPTZ,
+        observation_count INTEGER DEFAULT 0,
+        metadata JSONB DEFAULT '{}',
+        UNIQUE(session_map_id, segment_index)
+      );
+    `);
+        // ── entities 表 ──
+        await client.query(`
+      CREATE TABLE IF NOT EXISTS entities (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        session_map_id UUID REFERENCES session_map(id) ON DELETE CASCADE,
+        topic_segment_id UUID REFERENCES topic_segments(id) ON DELETE SET NULL,
+        name VARCHAR(500) NOT NULL,
+        type VARCHAR(100) NOT NULL,
+        tier entity_tier DEFAULT 'session',
+        weight FLOAT DEFAULT 1.0 CHECK (weight >= 0 AND weight <= 10),
+        description TEXT,
+        embedding vector(1536),
+        first_seen_at TIMESTAMPTZ DEFAULT NOW(),
+        last_seen_at TIMESTAMPTZ DEFAULT NOW(),
+        confidence FLOAT DEFAULT 0.8 CHECK (confidence >= 0 AND confidence <= 1),
+        metadata JSONB DEFAULT '{}'
+      );
+    `);
+        // ── relations 表 ──
+        await client.query(`
+      CREATE TABLE IF NOT EXISTS relations (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        source_entity_id UUID REFERENCES entities(id) ON DELETE CASCADE,
+        target_entity_id UUID REFERENCES entities(id) ON DELETE CASCADE,
+        relation_type relation_type NOT NULL,
+        confidence FLOAT DEFAULT 0.8 CHECK (confidence >= 0 AND confidence <= 1),
+        description TEXT,
+        created_at TIMESTAMPTZ DEFAULT NOW(),
+        session_map_id UUID REFERENCES session_map(id) ON DELETE CASCADE,
+        topic_segment_id UUID REFERENCES topic_segments(id) ON DELETE SET NULL,
+        CONSTRAINT chk_different_entities CHECK (source_entity_id != target_entity_id)
+      );
+    `);
+        // ── observations 表 ──
+        await client.query(`
+      CREATE TABLE IF NOT EXISTS observations (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        session_map_id UUID REFERENCES session_map(id) ON DELETE CASCADE,
+        topic_segment_id UUID REFERENCES topic_segments(id) ON DELETE SET NULL,
+        tool_name VARCHAR(255),
+        tool_input_summary TEXT,
+        tool_output_summary TEXT,
+        embedding vector(1536),
+        importance INTEGER DEFAULT 3 CHECK (importance >= 1 AND importance <= 5),
+        created_at TIMESTAMPTZ DEFAULT NOW(),
+        message_id VARCHAR(255),
+        metadata JSONB DEFAULT '{}'
+      );
+    `);
+        // ── reflections 表 ──
+        await client.query(`
+      CREATE TABLE IF NOT EXISTS reflections (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        session_map_id UUID REFERENCES session_map(id) ON DELETE CASCADE,
+        topic_segment_id UUID REFERENCES topic_segments(id) ON DELETE SET NULL,
+        summary TEXT NOT NULL,
+        source_observation_ids UUID[],
+        confidence FLOAT DEFAULT 0.8 CHECK (confidence >= 0 AND confidence <= 1),
+        pattern_type VARCHAR(100),
+        created_at TIMESTAMPTZ DEFAULT NOW(),
+        embedding vector(1536),
+        metadata JSONB DEFAULT '{}'
+      );
+    `);
+        // ── reflection_errors 表 ──
+        await client.query(`
+      CREATE TABLE IF NOT EXISTS reflection_errors (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        session_map_id UUID REFERENCES session_map(id) ON DELETE CASCADE,
+        error_message TEXT NOT NULL,
+        error_stack TEXT,
+        observation_count INTEGER,
+        created_at TIMESTAMPTZ DEFAULT NOW(),
+        retry_count INTEGER DEFAULT 0
+      );
+    `);
+        // ── semantic_cache 表 ──
+        await client.query(`
+      CREATE TABLE IF NOT EXISTS semantic_cache (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        query_hash VARCHAR(64) UNIQUE NOT NULL,
+        query_text TEXT NOT NULL,
+        query_embedding vector(1536) NOT NULL,
+        response_text TEXT NOT NULL,
+        hit_count INTEGER DEFAULT 1,
+        last_hit_at TIMESTAMPTZ DEFAULT NOW(),
+        created_at TIMESTAMPTZ DEFAULT NOW(),
+        similarity_threshold FLOAT DEFAULT 0.92,
+        is_pruned BOOLEAN DEFAULT FALSE,
+        session_map_id UUID REFERENCES session_map(id) ON DELETE SET NULL,
+        topic_segment_id UUID REFERENCES topic_segments(id) ON DELETE SET NULL
+      );
+    `);
+        // ── token_usage_log 表 ──
+        await client.query(`
+      CREATE TABLE IF NOT EXISTS token_usage_log (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        session_map_id UUID REFERENCES session_map(id) ON DELETE CASCADE,
+        operation_type VARCHAR(100) NOT NULL,
+        tokens_used INTEGER NOT NULL,
+        model_name VARCHAR(100),
+        created_at TIMESTAMPTZ DEFAULT NOW(),
+        metadata JSONB DEFAULT '{}'
+      );
+    `);
+        // ── cache_threshold_log 表 ──
+        await client.query(`
+      CREATE TABLE IF NOT EXISTS cache_threshold_log (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        threshold_value FLOAT NOT NULL,
+        hit_rate FLOAT,
+        query_count INTEGER,
+        adjustment_reason VARCHAR(50),
+        created_at TIMESTAMPTZ DEFAULT NOW()
+      );
+    `);
+        this.logger.info('Tables created');
+    }
+    async createIndexes(client) {
+        // ── session_map 表索引 ──
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_session_map_opencode_id ON session_map(opencode_session_id);
+      CREATE INDEX IF NOT EXISTS idx_session_map_omo_task_id ON session_map(omo_task_id);
+      CREATE INDEX IF NOT EXISTS idx_session_map_project_id ON session_map(project_id);
+    `);
+        // ── topic_segments 表索引 ──
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_topic_segments_session ON topic_segments(session_map_id);
+    `);
+        // HNSW 索引 for topic_segments
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_topic_segments_embedding ON topic_segments 
+      USING hnsw (embedding vector_cosine_ops)
+      WITH (m = 16, ef_construction = 64);
+    `);
+        // ── entities 表索引 ──
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_entities_session_map_type ON entities(session_map_id, type);
+      CREATE INDEX IF NOT EXISTS idx_entities_name ON entities(name);
+      CREATE INDEX IF NOT EXISTS idx_entities_tier_weight ON entities(tier, weight DESC) WHERE weight >= 0.3;
+      CREATE INDEX IF NOT EXISTS idx_entities_topic_segment ON entities(topic_segment_id);
+    `);
+        // HNSW 索引 for entities
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_entities_embedding ON entities 
+      USING hnsw (embedding vector_cosine_ops)
+      WITH (m = 16, ef_construction = 64);
+    `);
+        // ── relations 表索引 ──
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_relations_source ON relations(source_entity_id);
+      CREATE INDEX IF NOT EXISTS idx_relations_target ON relations(target_entity_id);
+      CREATE INDEX IF NOT EXISTS idx_relations_session_map ON relations(session_map_id);
+      CREATE INDEX IF NOT EXISTS idx_relations_confidence ON relations(confidence) WHERE confidence >= 0.5;
+      CREATE INDEX IF NOT EXISTS idx_relations_topic_segment ON relations(topic_segment_id);
+    `);
+        // ── observations 表索引 ──
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_observations_session_map ON observations(session_map_id);
+      CREATE INDEX IF NOT EXISTS idx_observations_importance ON observations(importance DESC);
+      CREATE INDEX IF NOT EXISTS idx_observations_created_at ON observations(created_at DESC);
+      CREATE INDEX IF NOT EXISTS idx_observations_topic_segment ON observations(topic_segment_id);
+    `);
+        // HNSW 索引 for observations
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_observations_embedding ON observations 
+      USING hnsw (embedding vector_cosine_ops)
+      WITH (m = 16, ef_construction = 64);
+    `);
+        // ── reflections 表索引 ──
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_reflections_session_map ON reflections(session_map_id);
+      CREATE INDEX IF NOT EXISTS idx_reflections_pattern ON reflections(pattern_type);
+      CREATE INDEX IF NOT EXISTS idx_reflections_confidence ON reflections(confidence) WHERE confidence >= 0.6;
+      CREATE INDEX IF NOT EXISTS idx_reflections_topic_segment ON reflections(topic_segment_id);
+    `);
+        // HNSW 索引 for reflections
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_reflections_embedding ON reflections 
+      USING hnsw (embedding vector_cosine_ops)
+      WITH (m = 16, ef_construction = 64);
+    `);
+        // ── reflection_errors 表索引 ──
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_reflection_errors_session_map ON reflection_errors(session_map_id);
+      CREATE INDEX IF NOT EXISTS idx_reflection_errors_created_at ON reflection_errors(created_at DESC);
+    `);
+        // ── semantic_cache 表索引 ──
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_semantic_cache_hash ON semantic_cache(query_hash);
+      CREATE INDEX IF NOT EXISTS idx_semantic_cache_hit_count ON semantic_cache(hit_count DESC);
+      CREATE INDEX IF NOT EXISTS idx_semantic_cache_last_hit ON semantic_cache(last_hit_at DESC);
+      CREATE INDEX IF NOT EXISTS idx_semantic_cache_pruned ON semantic_cache(is_pruned) WHERE is_pruned = FALSE;
+      CREATE INDEX IF NOT EXISTS idx_semantic_cache_session_map ON semantic_cache(session_map_id);
+      CREATE INDEX IF NOT EXISTS idx_semantic_cache_topic_segment ON semantic_cache(topic_segment_id);
+    `);
+        // HNSW 索引 for semantic_cache
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_semantic_cache_embedding ON semantic_cache 
+      USING hnsw (query_embedding vector_cosine_ops)
+      WITH (m = 16, ef_construction = 64);
+    `);
+        // ── token_usage_log 表索引 ──
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_token_usage_session_map ON token_usage_log(session_map_id);
+      CREATE INDEX IF NOT EXISTS idx_token_usage_operation ON token_usage_log(operation_type);
+      CREATE INDEX IF NOT EXISTS idx_token_usage_created_at ON token_usage_log(created_at DESC);
+    `);
+        // ── cache_threshold_log 表索引 ──
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_cache_threshold_created_at ON cache_threshold_log(created_at DESC);
+    `);
+        // ── 移除旧 sessions 表索引（如果存在） ──
+        // 旧 sessions 表保留为历史数据；新系统使用 session_map
+        await client.query(`
+      DROP INDEX IF EXISTS idx_sessions_external_id;
+      DROP INDEX IF EXISTS idx_sessions_project_id;
+      DROP INDEX IF EXISTS idx_sessions_reflection_last_at;
+    `);
+        // 移除旧子表 session_id 索引（如果存在）
+        await client.query(`
+      DROP INDEX IF EXISTS idx_entities_session_type;
+      DROP INDEX IF EXISTS idx_relations_session;
+      DROP INDEX IF EXISTS idx_observations_session;
+      DROP INDEX IF EXISTS idx_reflections_session;
+      DROP INDEX IF EXISTS idx_reflection_errors_session;
+      DROP INDEX IF EXISTS idx_token_usage_session;
+    `);
+        this.logger.info('Indexes created');
+    }
+    /**
+     * 迁移旧列名 session_id → session_map_id
+     *
+     * 兼容从 v1.x 升级的场景：旧 schema 中 entities/observations 等子表
+     * 使用 session_id 列名，但新 schema 使用 session_map_id。
+     * 重命名是 PostgreSQL 元数据操作，不重写数据行，FK 关联自动继承新列名。
+     */
+    async migrateLegacyColumnNames(client) {
+        const tables = ['entities', 'observations', 'relations', 'reflections',
+            'semantic_cache', 'token_usage_log', 'reflection_errors'];
+        for (const table of tables) {
+            try {
+                // 检查表是否存在且包含旧列名 session_id 但不含 session_map_id
+                const hasOld = await client.query(`
+          SELECT 1 FROM information_schema.columns
+          WHERE table_schema = 'public' AND table_name = $1 AND column_name = 'session_id'
+        `, [table]);
+                const hasNew = await client.query(`
+          SELECT 1 FROM information_schema.columns
+          WHERE table_schema = 'public' AND table_name = $1 AND column_name = 'session_map_id'
+        `, [table]);
+                if (hasOld.rows.length > 0 && hasNew.rows.length === 0) {
+                    await client.query(`
+            ALTER TABLE "${table}" RENAME COLUMN session_id TO session_map_id
+          `);
+                    this.logger.info(`Renamed ${table}.session_id → session_map_id`);
+                }
+            }
+            catch (err) {
+                // 某些表可能没有旧列名（新安装），静默跳过
+                this.logger.info(`Skipped column rename for ${table}: ${err}`);
+            }
+        }
+        this.logger.info('Legacy column migration complete');
+    }
+    /**
+     * 将旧 sessions 表的数据迁移到新的 session_map 表。
+     * 旧 sessions 表保留不删除（_legacy），仅复制元数据映射。
+     * 注意：子表（entities/observations 等）的 session_map_id 不会自动填充，
+     * 需要单独的迁移脚本处理 FK 重映射。
+     */
+    async migrateSessionsData(client) {
+        try {
+            const exists = await client.query(`
+        SELECT 1 FROM information_schema.tables 
+        WHERE table_schema = 'public' AND table_name = 'sessions'
+      `);
+            if (exists.rows.length === 0) {
+                this.logger.info('No legacy sessions table found, skipping migration');
+                return;
+            }
+            // 注释旧表为 legacy 标记
+            await client.query(`
+        COMMENT ON TABLE sessions IS '_legacy: replaced by session_map. Retained for data safety.';
+      `);
+            await client.query(`
+        INSERT INTO session_map (opencode_session_id, project_id, model_context_limit, created_at, last_active_at, metadata)
+        SELECT external_id, project_id, model_context_limit, created_at, updated_at, metadata
+        FROM sessions
+        ON CONFLICT (opencode_session_id) DO NOTHING;
+      `);
+            this.logger.info('Legacy sessions data migrated to session_map');
+        }
+        catch (error) {
+            this.logger.warn('Sessions migration warning (non-fatal):', error);
+        }
+    }
+    async initializeOmOSchema(client) {
+        // 检查是否需要 OmO 支持
+        const omOEnabled = process.env.OMO_ENABLED === 'true' ||
+            process.env.OMO_INTEGRATION === 'enabled';
+        if (!omOEnabled) {
+            this.logger.info('OmO integration not enabled, skipping OmO schema');
+            return;
+        }
+        this.logger.info('Initializing OmO schema...');
+        // 1. 添加 source_agent 字段到相关表
+        const tablesWithAgent = ['observations', 'semantic_cache', 'entities', 'reflections'];
+        for (const table of tablesWithAgent) {
+            try {
+                await client.query(`
+          ALTER TABLE ${table}
+          ADD COLUMN IF NOT EXISTS source_agent VARCHAR(255),
+          ADD COLUMN IF NOT EXISTS agent_task_id VARCHAR(255)
+        `);
+                // 创建索引
+                await client.query(`
+          CREATE INDEX IF NOT EXISTS idx_${table}_source_agent ON ${table}(source_agent)
+        `);
+                await client.query(`
+          CREATE INDEX IF NOT EXISTS idx_${table}_agent_task ON ${table}(agent_task_id)
+        `);
+            }
+            catch (error) {
+                this.logger.warn(`Schema update warning for ${table}:`, error);
+            }
+        }
+        // 2. 创建 OmO 协调表（使用 session_map_id 引用）
+        await client.query(`
+      CREATE TABLE IF NOT EXISTS omo_coordination (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        session_map_id UUID REFERENCES session_map(id) ON DELETE CASCADE,
+        agent_id VARCHAR(255),
+        coordination_type VARCHAR(100) NOT NULL,
+        coordination_data JSONB DEFAULT '{}',
+        created_at TIMESTAMPTZ DEFAULT NOW()
+      )
+    `);
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_omo_coordination_session_map ON omo_coordination(session_map_id)
+    `);
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_omo_coordination_agent ON omo_coordination(agent_id)
+    `);
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_omo_coordination_type ON omo_coordination(coordination_type)
+    `);
+        // 3. 创建 OmO Wisdom 表（用于与 OmO Wisdom Accumulation 同步）
+        await client.query(`
+      CREATE TABLE IF NOT EXISTS omo_wisdom (
+        id VARCHAR(255) PRIMARY KEY,
+        type VARCHAR(100) NOT NULL,
+        content TEXT NOT NULL,
+        metadata JSONB DEFAULT '{}',
+        agent_id VARCHAR(255),
+        created_at TIMESTAMPTZ DEFAULT NOW()
+      )
+    `);
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_omo_wisdom_agent ON omo_wisdom(agent_id)
+    `);
+        await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_omo_wisdom_type ON omo_wisdom(type)
+    `);
+        this.logger.info('OmO schema initialized');
+    }
+    async close() {
+        if (this.pool) {
+            await this.pool.end();
+            this.pool = null;
+            this.logger.info('Database connection closed');
+        }
+    }
+    getPool() {
+        if (!this.pool) {
+            throw new Error('Database not initialized. Call initialize() first.');
+        }
+        return this.pool;
+    }
+}
+exports.DatabaseInitializer = DatabaseInitializer;
+// 单例导出
+let initializer = null;
+function getDatabaseInitializer(config) {
+    if (!initializer) {
+        initializer = new DatabaseInitializer(config);
+    }
+    return initializer;
+}
+async function initializeDatabase(config) {
+    const init = getDatabaseInitializer(config);
+    return init.initialize();
+}
+async function closeDatabase() {
+    if (initializer) {
+        await initializer.close();
+        initializer = null;
+    }
+}
+//# sourceMappingURL=init-db.js.map

package/dist/src/db/init-db.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"init-db.js","sourceRoot":"","sources":["../../../src/db/init-db.ts"],"names":[],"mappings":";;;;;;AAulBA,wDAKC;AAED,gDAGC;AAED,sCAKC;AAxmBD,2BAAsC;AACtC,wDAAgC;AAChC,+CAAkD;AAYrC,QAAA,iBAAiB,GAAmB;IAC/C,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,WAAW;IACxC,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,MAAM,EAAE,EAAE,CAAC;IACjD,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,iBAAiB;IACtD,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,UAAU;IACvC,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,EAAE;IACvC,GAAG,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM;IAClC,cAAc,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,IAAI,EAAE,EAAE,CAAC;CACrE,CAAC;AAEF,MAAa,mBAAmB;IACtB,IAAI,GAAgB,IAAI,CAAC;IACzB,MAAM,CAAiB;IACvB,MAAM,GAAG,IAAA,qBAAY,EAAC,SAAS,CAAC,CAAC;IAEzC,YAAY,SAAkC,EAAE;QAC9C,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,yBAAiB,EAAE,GAAG,MAAM,EAAE,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,OAAO,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,IAAI,SAAI,CAAC;YACnB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YACtB,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YACtB,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG;YACpB,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;SAChC,CAAC,CAAC;QAEH,iBAAiB;QACjB,uCAAuC;QACvC,kBAAQ,CAAC,aAAa,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEpC,OAAO;QACP,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAE5B,WAAW;QACX,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAE3B,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,cAAc;QAC1B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAK,CAAC,OAAO,EAAE,CAAC;YAC1C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzD,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAC9D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;YACxD,MAAM,IAAI,KAAK,CAAC,oCAAoC,KAAK,EAAE,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa;QACzB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAK,CAAC,OAAO,EAAE,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAE5B,oBAAoB;YACpB,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAEpC,YAAY;YACZ,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAE/B,SAAS;YACT,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAEhC,wCAAwC;YACxC,MAAM,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;YAE5C,UAAU;YACV,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;YAEjC,kCAAkC;YAClC,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;YAEvC,6BAA6B;YAC7B,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;YAEvC,MAAM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC7B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC/B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;YACnD,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,MAAkB;QAC/C,MAAM,MAAM,CAAC,KAAK,CAAC;;;KAGlB,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IACzC,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,MAAkB;QAC1C,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;;;KAWlB,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACpC,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,MAAkB;QAC3C,gDAAgD;QAChD,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;;;KAWlB,CAAC,CAAC;QAEH,sCAAsC;QACtC,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;;;;;;;KAelB,CAAC,CAAC;QAEH,mBAAmB;QACnB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;;;;;;;;KAgBlB,CAAC,CAAC;QAEH,oBAAoB;QACpB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;;;;;KAalB,CAAC,CAAC;QAEH,uBAAuB;QACvB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;;;;;;KAclB,CAAC,CAAC;QAEH,sBAAsB;QACtB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;;;;;KAalB,CAAC,CAAC;QAEH,4BAA4B;QAC5B,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;;KAUlB,CAAC,CAAC;QAEH,yBAAyB;QACzB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;;;;;;;KAelB,CAAC,CAAC;QAEH,0BAA0B;QAC1B,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;;KAUlB,CAAC,CAAC;QAEH,8BAA8B;QAC9B,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;KASlB,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACrC,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,MAAkB;QAC5C,wBAAwB;QACxB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;KAIlB,CAAC,CAAC;QAEH,2BAA2B;QAC3B,MAAM,MAAM,CAAC,KAAK,CAAC;;KAElB,CAAC,CAAC;QAEH,6BAA6B;QAC7B,MAAM,MAAM,CAAC,KAAK,CAAC;;;;KAIlB,CAAC,CAAC;QAEH,qBAAqB;QACrB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;KAKlB,CAAC,CAAC;QAEH,uBAAuB;QACvB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;KAIlB,CAAC,CAAC;QAEH,sBAAsB;QACtB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;KAMlB,CAAC,CAAC;QAEH,yBAAyB;QACzB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;KAKlB,CAAC,CAAC;QAEH,2BAA2B;QAC3B,MAAM,MAAM,CAAC,KAAK,CAAC;;;;KAIlB,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;KAKlB,CAAC,CAAC;QAEH,0BAA0B;QAC1B,MAAM,MAAM,CAAC,KAAK,CAAC;;;;KAIlB,CAAC,CAAC;QAEH,8BAA8B;QAC9B,MAAM,MAAM,CAAC,KAAK,CAAC;;;KAGlB,CAAC,CAAC;QAEH,2BAA2B;QAC3B,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;KAOlB,CAAC,CAAC;QAEH,6BAA6B;QAC7B,MAAM,MAAM,CAAC,KAAK,CAAC;;;;KAIlB,CAAC,CAAC;QAEH,4BAA4B;QAC5B,MAAM,MAAM,CAAC,KAAK,CAAC;;;;KAIlB,CAAC,CAAC;QAEH,gCAAgC;QAChC,MAAM,MAAM,CAAC,KAAK,CAAC;;KAElB,CAAC,CAAC;QAEH,+BAA+B;QAC/B,wCAAwC;QACxC,MAAM,MAAM,CAAC,KAAK,CAAC;;;;KAIlB,CAAC,CAAC;QAEH,4BAA4B;QAC5B,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;KAOlB,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACtC,CAAC;IAED;;;;;;OAMG;IACK,KAAK,CAAC,wBAAwB,CAAC,MAAkB;QACvD,MAAM,MAAM,GAAG,CAAC,UAAU,EAAE,cAAc,EAAE,WAAW,EAAE,aAAa;YACpE,gBAAgB,EAAE,iBAAiB,EAAE,mBAAmB,CAAC,CAAC;QAE5D,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,CAAC;gBACH,8CAA8C;gBAC9C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;;;SAGjC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;gBACZ,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;;;SAGjC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;gBAEZ,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACvD,MAAM,MAAM,CAAC,KAAK,CAAC;2BACF,KAAK;WACrB,CAAC,CAAC;oBACH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,KAAK,8BAA8B,CAAC,CAAC;gBACnE,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,uBAAuB;gBACvB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,6BAA6B,KAAK,KAAK,GAAG,EAAE,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;IACvD,CAAC;IAED;;;;;OAKG;IACK,KAAK,CAAC,mBAAmB,CAAC,MAAkB;QAClD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;;;OAGjC,CAAC,CAAC;YAEH,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,oDAAoD,CAAC,CAAC;gBACvE,OAAO;YACT,CAAC;YAED,kBAAkB;YAClB,MAAM,MAAM,CAAC,KAAK,CAAC;;OAElB,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;OAKlB,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;QACnE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yCAAyC,EAAE,KAAK,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,MAAkB;QAClD,gBAAgB;QAChB,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,KAAK,MAAM;YAClC,OAAO,CAAC,GAAG,CAAC,eAAe,KAAK,SAAS,CAAC;QAE7D,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;YACrE,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAE/C,4BAA4B;QAC5B,MAAM,eAAe,GAAG,CAAC,cAAc,EAAE,gBAAgB,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC;QAEtF,KAAK,MAAM,KAAK,IAAI,eAAe,EAAE,CAAC;YACpC,IAAI,CAAC;gBACH,MAAM,MAAM,CAAC,KAAK,CAAC;wBACH,KAAK;;;SAGpB,CAAC,CAAC;gBAEH,OAAO;gBACP,MAAM,MAAM,CAAC,KAAK,CAAC;2CACgB,KAAK,oBAAoB,KAAK;SAChE,CAAC,CAAC;gBAEH,MAAM,MAAM,CAAC,KAAK,CAAC;2CACgB,KAAK,kBAAkB,KAAK;SAC9D,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,6BAA6B,KAAK,GAAG,EAAE,KAAK,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QAED,sCAAsC;QACtC,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;KASlB,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,KAAK,CAAC;;KAElB,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,KAAK,CAAC;;KAElB,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,KAAK,CAAC;;KAElB,CAAC,CAAC;QAEH,qDAAqD;QACrD,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;KASlB,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,KAAK,CAAC;;KAElB,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,KAAK,CAAC;;KAElB,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;YACjB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAED,OAAO;QACL,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;QACxE,CAAC;QACD,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;CACF;AA1jBD,kDA0jBC;AAED,OAAO;AACP,IAAI,WAAW,GAA+B,IAAI,CAAC;AAEnD,SAAgB,sBAAsB,CAAC,MAAgC;IACrE,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,WAAW,GAAG,IAAI,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAChD,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,CAAC;AAEM,KAAK,UAAU,kBAAkB,CAAC,MAAgC;IACvE,MAAM,IAAI,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;IAC5C,OAAO,IAAI,CAAC,UAAU,EAAE,CAAC;AAC3B,CAAC;AAEM,KAAK,UAAU,aAAa;IACjC,IAAI,WAAW,EAAE,CAAC;QAChB,MAAM,WAAW,CAAC,KAAK,EAAE,CAAC;QAC1B,WAAW,GAAG,IAAI,CAAC;IACrB,CAAC;AACH,CAAC"}