利用 Prometheus 自身的数据持久化能力保证统计的准确性

server/index.js

@@ -535,19 +535,7 @@ app.get('/api/metrics/overview', async (req, res) => {
       allServers = allServers.concat(m.servers);
     }
 
-    // --- 24h Traffic from DB (Integrating Bandwidth) ---
-    try {
-      // Each record represents a 5-second interval
-      const [sumRows] = await db.query('SELECT SUM(rx_bandwidth) as sumRx, SUM(tx_bandwidth) as sumTx FROM traffic_stats WHERE timestamp >= NOW() - INTERVAL 1 DAY');
-      
-      if (sumRows.length > 0 && sumRows[0].sumRx !== null) {
-        // Total bytes = Sum of (bytes/sec) * 5 seconds
-        traffic24hRx = sumRows[0].sumRx * 5;
-        traffic24hTx = sumRows[0].sumTx * 5;
-      }
-    } catch (err) {
-      console.error('Error calculating 24h traffic from DB integration:', err);
-    }
+
 
     const overview = {
       totalServers,
@@ -616,23 +604,31 @@ app.get('/api/metrics/overview', async (req, res) => {
   }
 });
 
-// Get network traffic history from DB (past 24h)
+// Get network traffic history (past 24h) from Prometheus
 app.get('/api/metrics/network-history', async (req, res) => {
   try {
-    const [rows] = await db.query('SELECT rx_bandwidth, tx_bandwidth, UNIX_TIMESTAMP(timestamp) as ts FROM traffic_stats WHERE timestamp >= NOW() - INTERVAL 1 DAY ORDER BY ts ASC');
-    
-    if (rows.length === 0) {
+    const [sources] = await db.query('SELECT * FROM prometheus_sources');
+    if (sources.length === 0) {
       return res.json({ timestamps: [], rx: [], tx: [] });
     }
 
-    res.json({
-      timestamps: rows.map(r => r.ts * 1000),
-      rx: rows.map(r => r.rx_bandwidth),
-      tx: rows.map(r => r.tx_bandwidth)
-    });
+    const histories = await Promise.all(sources.map(source => 
+      prometheusService.getNetworkHistory(source.url).catch(err => {
+        console.error(`Error fetching network history from ${source.name}:`, err.message);
+        return null;
+      })
+    ));
+
+    const validHistories = histories.filter(h => h !== null);
+    if (validHistories.length === 0) {
+      return res.json({ timestamps: [], rx: [], tx: [] });
+    }
+
+    const merged = prometheusService.mergeNetworkHistories(validHistories);
+    res.json(merged);
   } catch (err) {
-    console.error('Error fetching network history from DB:', err);
-    res.status(500).json({ error: 'Failed to fetch network history' });
+    console.error('Error fetching network history history:', err);
+    res.status(500).json({ error: 'Failed to fetch network history history' });
   }
 });
 
@@ -716,66 +712,9 @@ app.get('*', (req, res, next) => {
 });
 
 
-async function recordTrafficStats() {
-  if (!isDbInitialized) return;
-  try {
-    const [sources] = await db.query('SELECT * FROM prometheus_sources');
-    if (sources.length === 0) return;
-
-    let totalRxBytes = 0;
-    let totalTxBytes = 0;
-    let totalRxBandwidth = 0;
-    let totalTxBandwidth = 0;
-
-    const results = await Promise.all(sources.map(async source => {
-      try {
-        const [rxBytesRes, txBytesRes, rxBWRes, txBWRes] = await Promise.all([
-          prometheusService.query(source.url, 'sum(node_network_receive_bytes_total{device!~"lo|veth.*|docker.*|br-.*"})'),
-          prometheusService.query(source.url, 'sum(node_network_transmit_bytes_total{device!~"lo|veth.*|docker.*|br-.*"})'),
-          prometheusService.query(source.url, 'sum(rate(node_network_receive_bytes_total{device!~"lo|veth.*|docker.*|br-.*"}[1m]))'),
-          prometheusService.query(source.url, 'sum(rate(node_network_transmit_bytes_total{device!~"lo|veth.*|docker.*|br-.*"}[1m]))')
-        ]);
-        
-        return {
-          rxBytes: (rxBytesRes.length > 0) ? parseFloat(rxBytesRes[0].value[1]) : 0,
-          txBytes: (txBytesRes.length > 0) ? parseFloat(txBytesRes[0].value[1]) : 0,
-          rxBW: (rxBWRes.length > 0) ? parseFloat(rxBWRes[0].value[1]) : 0,
-          txBW: (txBWRes.length > 0) ? parseFloat(txBWRes[0].value[1]) : 0
-        };
-      } catch (e) {
-        return { rxBytes: 0, txBytes: 0, rxBW: 0, txBW: 0 };
-      }
-    }));
-
-    for (const r of results) {
-      totalRxBytes += r.rxBytes;
-      totalTxBytes += r.txBytes;
-      totalRxBandwidth += r.rxBW;
-      totalTxBandwidth += r.txBW;
-    }
-
-    // Always insert a record if we have sources, so the timeline advances
-    // Even if traffic is 0, we want to see 0 on the chart
-    await db.query('INSERT INTO traffic_stats (rx_bytes, tx_bytes, rx_bandwidth, tx_bandwidth) VALUES (?, ?, ?, ?)', [
-      Math.round(totalRxBytes), 
-      Math.round(totalTxBytes),
-      totalRxBandwidth,
-      totalTxBandwidth
-    ]);
-    console.log(`[Traffic Recorder] Saved stats: BW_RX=${totalRxBandwidth.toFixed(2)}, BW_TX=${totalTxBandwidth.toFixed(2)}`);
-  } catch (err) {
-    console.error('[Traffic Recorder] Error recording stats:', err);
-  }
-}
-
 // Check and fix database integrity on startup
 checkAndFixDatabase();
 
-// Record traffic every 5 seconds (17,280 points/day)
-setInterval(recordTrafficStats, 5 * 1000);
-// Initial record after a short delay
-setTimeout(recordTrafficStats, 10000);
-
 app.listen(PORT, HOST, () => {
   console.log(`\n  🚀 Data Visualization Display Wall`);
   console.log(`  📊 Server running at http://${HOST === '0.0.0.0' ? 'localhost' : HOST}:${PORT}`);