fix: 更新数据面板的驱动方式

server/index.js

@@ -40,18 +40,22 @@ if (fs.existsSync(distPath)) {
 
 const server = http.createServer(app)
 
+const { getStats } = require('./stats')
+
 const wss = new WebSocketServer({ server, path: '/ws' })
 wss.on('connection', (ws) => {
-  ws.send(JSON.stringify({ type: 'situation', data: getSituation() }))
+  ws.send(JSON.stringify({ type: 'situation', data: getSituation(), stats: getStats() }))
 })
+
 function broadcastSituation() {
   try {
-    const data = JSON.stringify({ type: 'situation', data: getSituation() })
+    const data = JSON.stringify({ type: 'situation', data: getSituation(), stats: getStats() })
     wss.clients.forEach((c) => {
       if (c.readyState === 1) c.send(data)
     })
   } catch (_) {}
 }
+app.set('broadcastSituation', broadcastSituation)
 setInterval(broadcastSituation, 3000)
 
 // 供爬虫调用：更新 situation.updated_at 并立即广播

server/routes.js

@@ -1,5 +1,6 @@
 const express = require('express')
 const { getSituation } = require('./situationData')
+const { getStats } = require('./stats')
 const db = require('./db')
 
 const router = express.Router()
@@ -84,16 +85,6 @@ function getClientIp(req) {
   return req.ip || req.socket?.remoteAddress || 'unknown'
 }
 
-function getStats() {
-  const viewers = db.prepare(
-    "SELECT COUNT(*) as n FROM visits WHERE last_seen > datetime('now', '-2 minutes')"
-  ).get().n
-  const cumulative = db.prepare('SELECT total FROM visitor_count WHERE id = 1').get()?.total ?? 0
-  const feedbackCount = db.prepare('SELECT COUNT(*) as n FROM feedback').get().n ?? 0
-  const shareCount = db.prepare('SELECT total FROM share_count WHERE id = 1').get()?.total ?? 0
-  return { viewers, cumulative, feedbackCount, shareCount }
-}
-
 router.post('/visit', (req, res) => {
   try {
     const ip = getClientIp(req)
@@ -103,6 +94,8 @@ router.post('/visit', (req, res) => {
     db.prepare(
       'INSERT INTO visitor_count (id, total) VALUES (1, 1) ON CONFLICT(id) DO UPDATE SET total = total + 1'
     ).run()
+    const broadcast = req.app?.get?.('broadcastSituation')
+    if (typeof broadcast === 'function') broadcast()
     res.json(getStats())
   } catch (err) {
     console.error(err)

server/stats.js

@@ -0,0 +1,13 @@
+const db = require('./db')
+
+function getStats() {
+  const viewers = db.prepare(
+    "SELECT COUNT(*) as n FROM visits WHERE last_seen > datetime('now', '-2 minutes')"
+  ).get().n
+  const cumulative = db.prepare('SELECT total FROM visitor_count WHERE id = 1').get()?.total ?? 0
+  const feedbackCount = db.prepare('SELECT COUNT(*) as n FROM feedback').get().n ?? 0
+  const shareCount = db.prepare('SELECT total FROM share_count WHERE id = 1').get()?.total ?? 0
+  return { viewers, cumulative, feedbackCount, shareCount }
+}
+
+module.exports = { getStats }