添加数据库修复脚本

server/index.js

@@ -4,6 +4,7 @@ const cors = require('cors');
 const path = require('path');
 const db = require('./db');
 const prometheusService = require('./prometheus-service');
+const checkAndFixDatabase = require('./db-integrity-check');
 
 const app = express();
 const PORT = process.env.PORT || 3000;
@@ -157,6 +158,21 @@ app.post('/api/setup/init', async (req, res) => {
       ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
     `);
 
+    await connection.query(`
+      CREATE TABLE IF NOT EXISTS site_settings (
+        id INT PRIMARY KEY DEFAULT 1,
+        page_name VARCHAR(255) DEFAULT '数据可视化展示大屏',
+        title VARCHAR(255) DEFAULT '数据可视化展示大屏',
+        logo_url TEXT,
+        default_theme VARCHAR(20) DEFAULT 'dark',
+        updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP
+      ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
+    `);
+    await connection.query(`
+      INSERT IGNORE INTO site_settings (id, page_name, title, default_theme) 
+      VALUES (1, '数据可视化展示大屏', '数据可视化展示大屏', 'dark')
+    `);
+
     await connection.end();
 
     // Save to .env
@@ -348,6 +364,47 @@ app.post('/api/sources/test', async (req, res) => {
   }
 });
 
+// ==================== Site Settings ====================
+
+// Get site settings
+app.get('/api/settings', async (req, res) => {
+  try {
+    const [rows] = await db.query('SELECT * FROM site_settings WHERE id = 1');
+    if (rows.length === 0) {
+      return res.json({
+        page_name: '数据可视化展示大屏',
+        title: '数据可视化展示大屏',
+        logo_url: null
+      });
+    }
+    res.json(rows[0]);
+  } catch (err) {
+    console.error('Error fetching settings:', err);
+    res.status(500).json({ error: 'Failed to fetch settings' });
+  }
+});
+
+// Update site settings
+app.post('/api/settings', requireAuth, async (req, res) => {
+  const { page_name, title, logo_url, default_theme } = req.body;
+  try {
+    await db.query(
+      `INSERT INTO site_settings (id, page_name, title, logo_url, default_theme) 
+       VALUES (1, ?, ?, ?, ?) 
+       ON DUPLICATE KEY UPDATE 
+         page_name = VALUES(page_name), 
+         title = VALUES(title), 
+         logo_url = VALUES(logo_url),
+         default_theme = VALUES(default_theme)`,
+      [page_name, title, logo_url, default_theme]
+    );
+    res.json({ success: true });
+  } catch (err) {
+    console.error('Error updating settings:', err);
+    res.status(500).json({ error: 'Failed to update settings' });
+  }
+});
+
 // ==================== Metrics Aggregation ====================
 
 // Get all aggregated metrics from all Prometheus sources
@@ -597,29 +654,8 @@ async function recordTrafficStats() {
   }
 }
 
-// Check if need to create traffic_stats table if db already initialized
-async function ensureTrafficTable() {
-  if (!isDbInitialized) return;
-  try {
-    await db.query(`
-      CREATE TABLE IF NOT EXISTS traffic_stats (
-        id INT AUTO_INCREMENT PRIMARY KEY,
-        rx_bytes BIGINT UNSIGNED DEFAULT 0,
-        tx_bytes BIGINT UNSIGNED DEFAULT 0,
-        rx_bandwidth DOUBLE DEFAULT 0,
-        tx_bandwidth DOUBLE DEFAULT 0,
-        timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-        UNIQUE INDEX (timestamp)
-      ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
-    `);
-    // Add columns if missing for existing tables
-    try { await db.query('ALTER TABLE traffic_stats ADD COLUMN rx_bandwidth DOUBLE DEFAULT 0'); } catch(e) {}
-    try { await db.query('ALTER TABLE traffic_stats ADD COLUMN tx_bandwidth DOUBLE DEFAULT 0'); } catch(e) {}
-    try { await db.query('ALTER TABLE traffic_stats ADD UNIQUE INDEX (timestamp)'); } catch(e) {}
-  } catch (err) {}
-}
-
-ensureTrafficTable().then(() => {
+// Check and fix database integrity on startup
+checkAndFixDatabase().then(() => {
   initialPreload();
 });