feat(log): 增强日志系统功能与追踪能力

- 新增模块级别日志控制和采样配置 - 为操作日志添加requestId字段支持全链路追踪 - 优化请求日志中间件，增强结构化记录能力 - 实现日志降级机制，数据库不可用时写入文件 - 添加相关数据库迁移脚本和模型索引
2026-05-07 17:14:54 +08:00
parent 54b0a8e7bb
commit 03e320af9e
8 changed files with 2809 additions and 58 deletions
@@ -128,6 +128,11 @@ const migrations = [
    description: '为 racks 表添加 rowPos、colPos、facing 字段，支持机柜位置定位',
    migrate: migrateRackPositionFields,
  },
+  {
+    name: '操作日志请求追踪',
+    description: '为 operation_logs 表添加 requestId 字段和复合索引，支持请求追踪',
+    migrate: migrateOperationLogRequestId,
+  },
 ];

 async function runMigrations() {
@@ -900,6 +905,52 @@ async function migrateRackPositionFields() {
  console.log('    机柜位置字段迁移完成');
 }

+async function addIndexIfNotExists(tableName, indexName, fields) {
+  const dialect = sequelize.getDialect();
+  try {
+    if (dialect === 'sqlite') {
+      await sequelize.query(
+        `CREATE INDEX IF NOT EXISTS ${indexName} ON ${tableName}(${fields.join(', ')})`
+      );
+    } else {
+      const [indexes] = await sequelize.query(
+        `SELECT INDEX_NAME FROM INFORMATION_SCHEMA.STATISTICS WHERE TABLE_NAME = ? AND INDEX_NAME = ? AND TABLE_SCHEMA = DATABASE()`,
+        { replacements: [tableName, indexName], type: sequelize.QueryTypes.SELECT }
+      );
+      if (!indexes || indexes.length === 0) {
+        await sequelize.query(
+          `CREATE INDEX \`${indexName}\` ON \`${tableName}\`(\`${fields.join('`, `')}\`)`
+        );
+      } else {
+        console.log(`    索引 ${indexName} 已存在，跳过`);
+        return;
+      }
+    }
+    console.log(`    索引 ${indexName} 创建成功`);
+  } catch (error) {
+    if (error.message.includes('already exists') || error.message.includes('Duplicate key name')) {
+      console.log(`    索引 ${indexName} 已存在，跳过`);
+    } else {
+      throw error;
+    }
+  }
+}
+
+async function migrateOperationLogRequestId() {
+  const tableName = 'operation_logs';
+
+  if (!(await tableExists(tableName))) {
+    console.log(`    ${tableName} 表不存在，跳过`);
+    return;
+  }
+
+  await addColumnIfNotExists(tableName, 'requestId', 'VARCHAR(255)');
+  await addIndexIfNotExists(tableName, 'operation_logs_requestId', ['requestId']);
+  await addIndexIfNotExists(tableName, 'operation_logs_module_createdAt', ['module', 'createdAt']);
+
+  console.log('    操作日志requestId字段和索引迁移完成');
+}
+
 // 执行迁移
 runMigrations().catch(error => {
  console.error('迁移执行失败:', error);
@@ -0,0 +1,115 @@
+/**
+ * 操作日志表添加 requestId 字段和索引
+ * 支持请求追踪ID关联，优化查询性能
+ * 支持 SQLite 和 MySQL，幂等执行
+ */
+
+require('dotenv').config({ path: require('path').join(__dirname, '../.env') });
+
+const { sequelize, dbDialect } = require('../db');
+
+async function getTableColumns(tableName) {
+  const dialect = sequelize.getDialect();
+  if (dialect === 'sqlite') {
+    const tableInfo = await sequelize.query(`PRAGMA table_info(${tableName})`, {
+      type: sequelize.QueryTypes.SELECT,
+    });
+    return tableInfo.map(col => col.name);
+  }
+  const tableInfo = await sequelize.query(`SHOW COLUMNS FROM ${tableName}`, {
+    type: sequelize.QueryTypes.SELECT,
+  });
+  return tableInfo.map(col => col.Field);
+}
+
+async function tableExists(tableName) {
+  const dialect = sequelize.getDialect();
+  if (dialect === 'sqlite') {
+    const tables = await sequelize.query(
+      "SELECT name FROM sqlite_master WHERE type='table' AND name=?",
+      { replacements: [tableName], type: sequelize.QueryTypes.SELECT }
+    );
+    return tables.length > 0;
+  }
+  const tables = await sequelize.query('SHOW TABLES LIKE ?', {
+    replacements: [tableName],
+    type: sequelize.QueryTypes.SELECT,
+  });
+  return tables.length > 0;
+}
+
+async function addColumnIfNotExists(tableName, columnName, columnDef) {
+  const columns = await getTableColumns(tableName);
+  if (!columns.includes(columnName)) {
+    await sequelize.query(`ALTER TABLE ${tableName} ADD COLUMN ${columnName} ${columnDef}`);
+    console.log(`    ${tableName} 表添加 ${columnName} 字段成功`);
+  } else {
+    console.log(`    ${tableName} 表 ${columnName} 字段已存在，跳过`);
+  }
+}
+
+async function addIndexIfNotExists(tableName, indexName, fields) {
+  const dialect = sequelize.getDialect();
+  try {
+    if (dialect === 'sqlite') {
+      await sequelize.query(
+        `CREATE INDEX IF NOT EXISTS ${indexName} ON ${tableName}(${fields.join(', ')})`
+      );
+    } else {
+      const [indexes] = await sequelize.query(
+        `SELECT INDEX_NAME FROM INFORMATION_SCHEMA.STATISTICS WHERE TABLE_NAME = ? AND INDEX_NAME = ? AND TABLE_SCHEMA = DATABASE()`,
+        { replacements: [tableName, indexName], type: sequelize.QueryTypes.SELECT }
+      );
+      if (!indexes || indexes.length === 0) {
+        await sequelize.query(
+          `CREATE INDEX \`${indexName}\` ON \`${tableName}\`(\`${fields.join('`, `')}\`)`
+        );
+      } else {
+        console.log(`    索引 ${indexName} 已存在，跳过`);
+        return;
+      }
+    }
+    console.log(`    索引 ${indexName} 创建成功`);
+  } catch (error) {
+    if (error.message.includes('already exists') || error.message.includes('Duplicate key name')) {
+      console.log(`    索引 ${indexName} 已存在，跳过`);
+    } else {
+      throw error;
+    }
+  }
+}
+
+async function migrateOperationLogRequestId() {
+  const tableName = 'operation_logs';
+
+  if (!(await tableExists(tableName))) {
+    console.log(`    ${tableName} 表不存在，跳过`);
+    return;
+  }
+
+  await addColumnIfNotExists(tableName, 'requestId', 'VARCHAR(255)');
+
+  await addIndexIfNotExists(tableName, 'operation_logs_requestId', ['requestId']);
+  await addIndexIfNotExists(tableName, 'operation_logs_module_createdAt', ['module', 'createdAt']);
+
+  console.log('    操作日志requestId字段和索引迁移完成');
+}
+
+async function run() {
+  console.log('========================================');
+  console.log('  操作日志表添加requestId字段迁移脚本  ');
+  console.log('========================================');
+  console.log(`数据库类型: ${dbDialect}`);
+
+  try {
+    await migrateOperationLogRequestId();
+    console.log('\n迁移完成');
+  } catch (error) {
+    console.error('迁移失败:', error.message);
+    process.exit(1);
+  } finally {
+    await sequelize.close();
+  }
+}
+
+run();