mr-sliy 1.0.0

package/src/utils/mysql.js

@@ -0,0 +1,198 @@
+/**
+ * MySQL数据库连接模块
+ * 用于云端知识库同步
+ * 可选配置，未启用时不影响本地SQLite
+ */
+
+const mysql = require('mysql2/promise');
+const { config } = require('../config');
+const { logger } = require('./logger');
+
+let pool = null;
+
+/**
+ * 获取MySQL连接池
+ */
+function getPool() {
+  if (!config.mysql.enabled) {
+    return null;
+  }
+  
+  if (!pool) {
+    try {
+      pool = mysql.createPool({
+        host: config.mysql.host,
+        port: config.mysql.port,
+        user: config.mysql.user,
+        password: config.mysql.password,
+        database: config.mysql.database,
+        connectionLimit: config.mysql.connectionLimit,
+        waitForConnections: true,
+        queueLimit: 0,
+        charset: 'utf8mb4'
+      });
+      
+      logger.info('MySQL连接池创建成功');
+    } catch (error) {
+      logger.warn(`MySQL连接池创建失败: ${error.message}`);
+      pool = null;
+    }
+  }
+  
+  return pool;
+}
+
+/**
+ * 测试MySQL连接
+ */
+async function testConnection() {
+  const pool = getPool();
+  if (!pool) {
+    return { success: false, message: 'MySQL未启用' };
+  }
+  
+  try {
+    const connection = await pool.getConnection();
+    await connection.ping();
+    connection.release();
+    return { success: true, message: 'MySQL连接成功' };
+  } catch (error) {
+    return { success: false, message: error.message };
+  }
+}
+
+/**
+ * 执行查询
+ */
+async function query(sql, params = []) {
+  const pool = getPool();
+  if (!pool) {
+    throw new Error('MySQL未启用');
+  }
+  
+  try {
+    const [rows] = await pool.execute(sql, params);
+    return rows;
+  } catch (error) {
+    logger.debug(`MySQL查询失败: ${error.message}`);
+    throw error;
+  }
+}
+
+/**
+ * 执行插入/更新/删除
+ */
+async function execute(sql, params = []) {
+  const pool = getPool();
+  if (!pool) {
+    throw new Error('MySQL未启用');
+  }
+  
+  try {
+    const [result] = await pool.execute(sql, params);
+    return {
+      success: true,
+      affectedRows: result.affectedRows,
+      insertId: result.insertId
+    };
+  } catch (error) {
+    logger.debug(`MySQL执行失败: ${error.message}`);
+    throw error;
+  }
+}
+
+/**
+ * 初始化MySQL数据库表
+ */
+async function initDatabase() {
+  const pool = getPool();
+  if (!pool) {
+    return false;
+  }
+  
+  try {
+    // 创建知识条目表
+    await query(`
+      CREATE TABLE IF NOT EXISTS knowledge_entries (
+        id VARCHAR(36) PRIMARY KEY,
+        title VARCHAR(255) NOT NULL,
+        category VARCHAR(50) NOT NULL,
+        content TEXT NOT NULL,
+        tags TEXT,
+        language VARCHAR(20) DEFAULT 'general',
+        source VARCHAR(50) DEFAULT 'default',
+        created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+        updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+        INDEX idx_category (category),
+        INDEX idx_language (language)
+      ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
+    `);
+    
+    // 创建优化案例表
+    await query(`
+      CREATE TABLE IF NOT EXISTS optimization_cases (
+        id VARCHAR(36) PRIMARY KEY,
+        title VARCHAR(255) NOT NULL,
+        category VARCHAR(50) NOT NULL,
+        before_code TEXT,
+        after_code TEXT,
+        description TEXT,
+        language VARCHAR(20) DEFAULT 'javascript',
+        complexity VARCHAR(20) DEFAULT 'medium',
+        tags TEXT,
+        created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+        updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+        INDEX idx_category (category),
+        INDEX idx_language (language)
+      ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
+    `);
+    
+    // 创建同步元数据表
+    await query(`
+      CREATE TABLE IF NOT EXISTS sync_metadata (
+        id INT PRIMARY KEY AUTO_INCREMENT,
+        table_name VARCHAR(50) NOT NULL,
+        last_sync_at TIMESTAMP NULL,
+        record_count INT DEFAULT 0,
+        machine_id VARCHAR(32),
+        created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+        updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+        UNIQUE KEY uk_table_machine (table_name, machine_id)
+      ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
+    `);
+    
+    logger.info('MySQL数据库表初始化完成');
+    return true;
+  } catch (error) {
+    logger.warn(`MySQL数据库表初始化失败: ${error.message}`);
+    return false;
+  }
+}
+
+/**
+ * 关闭连接池
+ */
+async function closePool() {
+  if (pool) {
+    await pool.end();
+    pool = null;
+    logger.info('MySQL连接池已关闭');
+  }
+}
+
+/**
+ * 检查MySQL是否可用
+ */
+function isEnabled() {
+  return config.mysql.enabled && getPool() !== null;
+}
+
+module.exports = {
+  getPool,
+  testConnection,
+  query,
+  execute,
+  initDatabase,
+  closePool,
+  isEnabled
+};

package/src/utils/response.js

@@ -0,0 +1,124 @@
+/**
+ * API响应工具模块
+ * 统一API响应格式
+ */
+
+/**
+ * 成功响应
+ */
+function success(data = null, message = '操作成功', code = 200) {
+  return {
+    success: true,
+    code,
+    message,
+    data,
+    timestamp: new Date().toISOString()
+  };
+}
+
+/**
+ * 错误响应
+ */
+function error(message = '操作失败', code = 500, errors = null) {
+  return {
+    success: false,
+    code,
+    message,
+    errors,
+    timestamp: new Date().toISOString()
+  };
+}
+
+/**
+ * 分页响应
+ */
+function paginate(data, total, page, pageSize) {
+  return {
+    success: true,
+    code: 200,
+    message: '查询成功',
+    data: {
+      list: data,
+      pagination: {
+        total,
+        page: parseInt(page),
+        pageSize: parseInt(pageSize),
+        totalPages: Math.ceil(total / parseInt(pageSize))
+      }
+    },
+    timestamp: new Date().toISOString()
+  };
+}
+
+/**
+ * 参数验证错误响应
+ */
+function validationError(errors) {
+  return {
+    success: false,
+    code: 400,
+    message: '参数验证失败',
+    errors,
+    timestamp: new Date().toISOString()
+  };
+}
+
+/**
+ * 未授权响应
+ */
+function unauthorized(message = '未授权访问') {
+  return {
+    success: false,
+    code: 401,
+    message,
+    timestamp: new Date().toISOString()
+  };
+}
+
+/**
+ * 禁止访问响应
+ */
+function forbidden(message = '禁止访问') {
+  return {
+    success: false,
+    code: 403,
+    message,
+    timestamp: new Date().toISOString()
+  };
+}
+
+/**
+ * 资源未找到响应
+ */
+function notFound(message = '资源未找到') {
+  return {
+    success: false,
+    code: 404,
+    message,
+    timestamp: new Date().toISOString()
+  };
+}
+
+/**
+ * 服务端错误响应
+ */
+function serverError(message = '服务器内部错误', errors = null) {
+  return {
+    success: false,
+    code: 500,
+    message,
+    errors,
+    timestamp: new Date().toISOString()
+  };
+}
+
+module.exports = {
+  success,
+  error,
+  paginate,
+  validationError,
+  unauthorized,
+  forbidden,
+  notFound,
+  serverError
+};