fdb2 1.0.0

package/server/model/connection.entity.ts

@@ -0,0 +1,66 @@
+
+/**
+ * 数据库连接配置实体
+ * 注意：此实体仅用于代码结构，实际数据保存在本地JSON文件中
+ */
+export class ConnectionEntity {
+  /**
+   * 连接ID
+   */
+  id: string;
+
+  /**
+   * 连接名称
+   */
+  name: string;
+
+  /**
+   * 数据库类型 (mysql, postgresql, sqlite, etc.)
+   */
+  type: string;
+
+  /**
+   * 主机地址
+   */
+  host: string;
+
+  /**
+   * 端口号
+   */
+  port: number;
+
+  /**
+   * 数据库名称
+   */
+  database: string;
+
+  /**
+   * 用户名
+   */
+  username: string;
+
+  /**
+   * 密码 (加密存储)
+   */
+  password: string;
+
+  /**
+   * 连接参数 (SSL等)
+   */
+  options: Record<string, any>;
+
+  /**
+   * 是否启用
+   */
+  enabled: boolean;
+
+  /**
+   * 创建时间
+   */
+  createdAt: Date;
+
+  /**
+   * 更新时间
+   */
+  updatedAt: Date;
+}

package/server/model/database.entity.ts

@@ -0,0 +1,246 @@
+/**
+ * 数据库信息实体
+ * 用于存储数据库的基本信息和结构
+ */
+
+export class DatabaseEntity {
+  /**
+   * 数据库名称
+   */
+  name: string;
+
+  /**
+   * 数据库字符集
+   */
+  charset?: string;
+
+  /**
+   * 数据库排序规则
+   */
+  collation?: string;
+
+  /**
+   * 表数量
+   */
+  tableCount: number;
+
+  /**
+   * 数据库大小 (字节)
+   */
+  size: number;
+
+  /**
+   * 创建时间
+   */
+  createdAt?: Date;
+
+  /**
+   * 表信息列表
+   */
+  tables?: TableEntity[];
+}
+
+/**
+ * 数据库表实体
+ */
+export class TableEntity {
+  /**
+   * 表名
+   */
+  name: string;
+
+  /**
+   * 表类型
+   */
+  type?: string;
+
+  /**
+   * 引擎 (MySQL)
+   */
+  engine?: string;
+
+  /**
+   * 行数
+   */
+  rowCount: number;
+
+  /**
+   * 数据大小 (字节)
+   */
+  dataSize: number;
+
+  /**
+   * 索引大小 (字节)
+   */
+  indexSize: number;
+
+  /**
+   * 字符集
+   */
+  charset?: string;
+
+  /**
+   * 排序规则
+   */
+  collation?: string;
+
+  /**
+   * 创建时间
+   */
+  createdAt?: Date;
+
+  /**
+   * 更新时间
+   */
+  updatedAt?: Date;
+
+  /**
+   * 注释
+   */
+  comment?: string;
+
+  /**
+   * 列信息
+   */
+  columns?: ColumnEntity[];
+
+  /**
+   * 索引信息
+   */
+  indexes?: IndexEntity[];
+
+  /**
+   * 外键信息
+   */
+  foreignKeys?: ForeignKeyEntity[];
+}
+
+/**
+ * 数据库列实体
+ */
+export class ColumnEntity {
+  /**
+   * 列名
+   */
+  name: string;
+
+  /**
+   * 数据类型
+   */
+  type: string;
+
+  /**
+   * 是否允许NULL
+   */
+  nullable: boolean;
+
+  /**
+   * 默认值
+   */
+  defaultValue?: any;
+
+  /**
+   * 是否主键
+   */
+  isPrimary: boolean;
+
+  /**
+   * 是否自增
+   */
+  isAutoIncrement: boolean;
+
+  /**
+   * 字符长度
+   */
+  length?: number;
+
+  /**
+   * 小数位数
+   */
+  precision?: number;
+
+  /**
+   * 小数点后位数
+   */
+  scale?: number;
+
+  /**
+   * 字符集
+   */
+  charset?: string;
+
+  /**
+   * 排序规则
+   */
+  collation?: string;
+
+  /**
+   * 注释
+   */
+  comment?: string;
+}
+
+/**
+ * 数据库索引实体
+ */
+export class IndexEntity {
+  /**
+   * 索引名
+   */
+  name: string;
+
+  /**
+   * 索引类型 (PRIMARY, UNIQUE, INDEX, FULLTEXT)
+   */
+  type: string;
+
+  /**
+   * 索引列
+   */
+  columns: string[];
+
+  /**
+   * 是否唯一
+   */
+  unique: boolean;
+
+  /**
+   * 注释
+   */
+  comment?: string;
+}
+
+/**
+ * 外键实体
+ */
+export class ForeignKeyEntity {
+  /**
+   * 约束名
+   */
+  name: string;
+
+  /**
+   * 本地表列
+   */
+  column: string;
+
+  /**
+   * 目标表
+   */
+  referencedTable: string;
+
+  /**
+   * 目标表列
+   */
+  referencedColumn: string;
+
+  /**
+   * 删除规则 (CASCADE, SET NULL, RESTRICT, NO ACTION)
+   */
+  onDelete: string;
+
+  /**
+   * 更新规则 (CASCADE, SET NULL, RESTRICT, NO ACTION)
+   */
+  onUpdate: string;
+}

package/server/service/connection.service.ts

@@ -0,0 +1,334 @@
+import * as fs from 'fs';
+import * as path from 'path';
+import { ConnectionEntity } from '../model/connection.entity';
+import { DataSource, type DataSourceOptions } from 'typeorm';
+
+/**
+ * 数据库连接管理服务
+ * 负责管理数据库连接配置和连接实例
+ */
+export class ConnectionService {
+
+  /**
+   * 连接配置文件路径
+   */
+  private readonly configPath = path.join(process.cwd(), 'data', 'connections.json');
+
+  /**
+   * 活跃的数据库连接实例
+   */
+  private activeConnections: Map<string, DataSource> = new Map();
+
+  /**
+   * 初始化服务，创建配置目录
+   */
+  async init() {
+    const configDir = path.dirname(this.configPath);
+    if (!fs.existsSync(configDir)) {
+      fs.mkdirSync(configDir, { recursive: true });
+    }
+  }
+
+  /**
+   * 获取所有数据库连接配置
+   */
+  async getAllConnections(): Promise<ConnectionEntity[]> {
+    try {
+      if (!fs.existsSync(this.configPath)) {
+        return [];
+      }
+      const data = fs.readFileSync(this.configPath, 'utf8');
+      return JSON.parse(data) as ConnectionEntity[];
+    } catch (error) {
+      console.error('读取连接配置失败:', error);
+      return [];
+    }
+  }
+
+  /**
+   * 根据ID获取数据库连接配置
+   */
+  async getConnectionById(id: string): Promise<ConnectionEntity | null> {
+    const connections = await this.getAllConnections();
+    return connections.find(conn => conn.id === id) || null;
+  }
+
+  /**
+   * 添加数据库连接配置
+   */
+  async addConnection(connection: ConnectionEntity): Promise<ConnectionEntity> {
+    const connections = await this.getAllConnections();
+    
+    // 检查名称是否重复
+    if (connections.find(conn => conn.name === connection.name)) {
+      throw new Error('连接名称已存在');
+    }
+
+    // 生成ID并设置时间戳
+    connection.id = this.generateId();
+    connection.createdAt = new Date();
+    connection.updatedAt = new Date();
+    connection.enabled = connection.enabled !== undefined ? connection.enabled : true;
+
+    connections.push(connection);
+    await this.saveConnections(connections);
+    
+    return connection;
+  }
+
+  /**
+   * 更新数据库连接配置
+   */
+  async updateConnection(id: string, updates: Partial<ConnectionEntity>): Promise<ConnectionEntity> {
+    const connections = await this.getAllConnections();
+    const index = connections.findIndex(conn => conn.id === id);
+    
+    if (index === -1) {
+      throw new Error('连接配置不存在');
+    }
+
+    // 检查名称重复
+    if (updates.name && connections.find((conn, idx) => conn.name === updates.name && idx !== index)) {
+      throw new Error('连接名称已存在');
+    }
+    // @ts-ignore
+    connections[index] = { ...connections[index], ...updates, updatedAt: new Date() };
+    await this.saveConnections(connections);
+    // @ts-ignore
+    return connections[index];
+  }
+
+  /**
+   * 删除数据库连接配置
+   */
+  async deleteConnection(id: string): Promise<void> {
+    const connections = await this.getAllConnections();
+    const filteredConnections = connections.filter(conn => conn.id !== id);
+    
+    if (filteredConnections.length === connections.length) {
+      throw new Error('连接配置不存在');
+    }
+
+    // 关闭活跃连接
+    if (this.activeConnections.has(id)) {
+      await this.activeConnections.get(id)?.destroy();
+      this.activeConnections.delete(id);
+    }
+
+    await this.saveConnections(filteredConnections);
+  }
+
+  /**
+   * 测试数据库连接
+   */
+  async testConnection(connection: ConnectionEntity): Promise<boolean> {
+    try {
+      console.log('test', connection);
+      const tempDataSource = await this.createTypeORMDataSource(connection);
+      await tempDataSource.query('SELECT 1');
+      await tempDataSource.destroy();
+      return true;
+    } catch (error) {
+      console.error(error);
+      return false;
+    }
+  }
+
+  /**
+   * 获取活跃的数据库连接
+   */
+  async getActiveConnection(id: string, database?: string): Promise<DataSource> {
+    const key = database ? `${id}_${database}` : id;
+    if (this.activeConnections.has(key)) {
+      const db = this.activeConnections.get(key);
+      // 检查连接是否仍然有效
+      if (db?.isInitialized) {
+        return db;
+      } else {
+        // 连接已关闭，从缓存中移除
+        this.activeConnections.delete(key);
+      }
+    }
+
+    // 连接池大小限制，防止连接数无限增长
+    if (this.activeConnections.size >= 10) {
+      // 关闭最旧的连接
+      const oldestKey = this.activeConnections.keys().next().value || '';
+      const oldestConnection = this.activeConnections.get(oldestKey);
+      try {
+        await oldestConnection?.destroy();
+      } catch (error) {
+        console.error(`关闭旧连接 ${oldestKey} 失败:`, error);
+      }
+      this.activeConnections.delete(oldestKey);
+    }
+
+    const connectionConfig = await this.getConnectionById(id);
+    if (!connectionConfig) {
+      throw new Error('连接配置不存在');
+    }
+
+    // 创建一个新的连接配置，使用指定的数据库
+    const updatedConnectionConfig: ConnectionEntity = {      
+      ...connectionConfig,
+      database: database || connectionConfig.database
+    };
+
+    const dataSource = await this.createTypeORMDataSource(updatedConnectionConfig);
+    this.activeConnections.set(key, dataSource);
+    
+    return dataSource;
+  }
+
+  /**
+   * 关闭数据库连接
+   */
+  async closeConnection(id: string, database?: string): Promise<void> {
+    const key = database ? `${id}_${database}` : id;
+    if (this.activeConnections.has(key)) {
+      await this.activeConnections.get(key)?.destroy();
+      this.activeConnections.delete(key);
+    }
+    // 也关闭默认连接（如果存在）
+    if (database && this.activeConnections.has(id)) {
+      await this.activeConnections.get(id)?.destroy();
+      this.activeConnections.delete(id);
+    }
+  }
+
+  /**
+   * 关闭特定连接的所有数据库连接
+   */
+  async closeAllConnectionsForId(connectionId: string): Promise<void> {
+    const keysToDelete: string[] = [];
+    for (const [key, dataSource] of this.activeConnections) {
+      if (key.startsWith(connectionId + '_') || key === connectionId) {
+        try {
+          await dataSource.destroy();
+          keysToDelete.push(key);
+        } catch (error) {
+          console.error(`关闭连接 ${key} 失败:`, error);
+        }
+      }
+    }
+    keysToDelete.forEach(key => this.activeConnections.delete(key));
+  }
+
+  /**
+   * 关闭所有数据库连接
+   */
+  async closeAllConnections(): Promise<void> {
+    for (const [id, dataSource] of this.activeConnections) {
+      try {
+        await dataSource.destroy();
+      } catch (error) {
+        console.error(`关闭连接 ${id} 失败:`, error);
+      }
+    }
+    this.activeConnections.clear();
+  }
+
+  /**
+   * 创建TypeORM数据源
+   */
+  private async createTypeORMDataSource(connectionConfig: ConnectionEntity): Promise<DataSource> {
+    const connectionOptions = this.getTypeORMOptions(connectionConfig);
+    return new DataSource(connectionOptions).initialize();
+  }
+
+  /**
+   * 获取TypeORM连接配置
+   */
+  private getTypeORMOptions(connectionConfig: ConnectionEntity): DataSourceOptions {
+    const baseOptions = {
+      type: connectionConfig.type as any,
+      host: connectionConfig.host,
+      port: connectionConfig.port,
+      username: connectionConfig.username,
+      password: connectionConfig.password,
+      database: connectionConfig.database,
+      synchronize: false,
+      logging: false,
+      // 关键配置：开启多语句执行
+      extra: {
+        multipleStatements: true
+      },
+      ...connectionConfig.options
+    };
+
+    // 根据数据库类型调整配置
+    switch (connectionConfig.type.toLowerCase()) {
+      case 'sqlite':
+        return {
+          ...baseOptions,
+          type: 'sqlite' as any,
+          database: connectionConfig.database,
+          host: undefined,
+          port: undefined,
+          username: undefined,
+          password: undefined
+        };
+      case 'postgres':
+      case 'postgresql':
+        return {
+          ...baseOptions,
+          type: 'postgres' as any,
+          ssl: connectionConfig.options?.ssl || false
+        };
+      case 'oracle':
+        return {
+          ...baseOptions,
+          type: 'oracle' as any,
+          connectString: `${connectionConfig.host}:${connectionConfig.port}/${connectionConfig.database}`,
+          host: undefined,
+          port: undefined,
+          database: undefined,
+          extra: {
+            connectionTimeout: 60000,
+            poolMax: 10,
+            poolMin: 1,
+            poolIncrement: 1
+          }
+        };
+      case 'mssql':
+      case 'sqlserver':
+        return {
+          ...baseOptions,
+          type: 'mssql' as any,
+          options: {
+            encrypt: connectionConfig.options?.encrypt || false,
+            trustServerCertificate: true
+          },
+          extra: {
+            connectionTimeout: 60000,
+            requestTimeout: 15000
+          }
+        };
+      default:
+        return baseOptions;
+    }
+  }
+
+  /**
+   * 保存连接配置到文件
+   */
+  private async saveConnections(connections: ConnectionEntity[]): Promise<void> {
+    try {
+      const dir = path.dirname(this.configPath);
+      if (!fs.existsSync(dir)) {
+        fs.mkdirSync(dir, { recursive: true });
+      }
+      fs.writeFileSync(this.configPath, JSON.stringify(connections, null, 2), 'utf8');
+    } catch (error) {
+      console.error('保存连接配置失败:', error);
+      throw error;
+    }
+  }
+
+  /**
+   * 生成唯一ID
+   */
+  private generateId(): string {
+    return Date.now().toString(36) + Math.random().toString(36).substr(2);
+  }
+}