更新完善页面

config/database.js

@@ -16,44 +16,109 @@ const config = {
   password: requireEnv('DB_PASSWORD', ''),
   database: requireEnv('DB_NAME', 'miniprogram'),
   waitForConnections: true,
-  connectionLimit: 10,
+  connectionLimit: parseInt(process.env.DB_POOL_LIMIT) || 10,
   queueLimit: 0
 }
 
+const SLOW_QUERY_MS = parseInt(process.env.SLOW_QUERY_MS) || 500
+
 const pool = mysql.createPool(config)
 
+const queryStats = {
+  total: 0,
+  slow: 0,
+  errors: 0,
+  totalDurationMs: 0,
+  lastSlow: []
+}
+
+function recordSlow(sql, durationMs, params) {
+  queryStats.slow += 1
+  const entry = {
+    sql: sql.replace(/\s+/g, ' ').slice(0, 200),
+    durationMs: Math.round(durationMs),
+    paramCount: Array.isArray(params) ? params.length : 0,
+    at: new Date().toISOString()
+  }
+  queryStats.lastSlow.unshift(entry)
+  if (queryStats.lastSlow.length > 20) queryStats.lastSlow.pop()
+  console.warn('[SLOW QUERY]', entry.durationMs + 'ms', entry.sql)
+}
+
 async function query(sql, params = []) {
+  const start = Date.now()
   const connection = await pool.getConnection()
   try {
-    const [rows, fields] = await connection.execute(sql, params)
+    const [rows] = await connection.execute(sql, params)
+    const duration = Date.now() - start
+    queryStats.total += 1
+    queryStats.totalDurationMs += duration
+    if (duration >= SLOW_QUERY_MS) recordSlow(sql, duration, params)
     return rows
+  } catch (err) {
+    queryStats.errors += 1
+    throw err
   } finally {
     connection.release()
   }
 }
 
+function getPoolMetrics() {
+  const p = pool.pool || pool
+  return {
+    connectionLimit: config.connectionLimit,
+    allConnections: (p._allConnections && p._allConnections.length) || 0,
+    freeConnections: (p._freeConnections && p._freeConnections.length) || 0,
+    acquiringConnections: (p._acquiringConnections && p._acquiringConnections.length) || 0,
+    queue: (p._connectionQueue && p._connectionQueue.length) || 0
+  }
+}
+
+function getQueryStats() {
+  const avg = queryStats.total > 0 ? Math.round(queryStats.totalDurationMs / queryStats.total) : 0
+  return {
+    total: queryStats.total,
+    slow: queryStats.slow,
+    errors: queryStats.errors,
+    avgDurationMs: avg,
+    lastSlow: queryStats.lastSlow.slice(0, 10)
+  }
+}
+
+function resetQueryStats() {
+  queryStats.total = 0
+  queryStats.slow = 0
+  queryStats.errors = 0
+  queryStats.totalDurationMs = 0
+  queryStats.lastSlow = []
+}
+
 async function initDatabase() {
   try {
     const connection = await mysql.createConnection({
       host: config.host,
       port: config.port,
       user: config.user,
-      password: config.password,
-      multipleStatements: true
+      password: config.password
     })
-    
+
     await connection.query(`CREATE DATABASE IF NOT EXISTS ${config.database} CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci`)
     await connection.query(`USE ${config.database}`)
-    
+
     const fs = require('fs')
     const path = require('path')
     const schema = fs.readFileSync(path.join(__dirname, 'schema.sql'), 'utf8')
     const statements = schema.split(';').filter(s => s.trim())
-    
+
     for (const statement of statements) {
-      await connection.query(statement)
+      try {
+        await connection.query(statement)
+      } catch (e) {
+        if (/Duplicate column name|already exists|exists/i.test(e.message)) continue
+        throw e
+      }
     }
-    
+
     await connection.end()
     console.log('Database initialized successfully')
   } catch (error) {
@@ -77,10 +142,37 @@ async function transaction(callback) {
   }
 }
 
+let healthCheckTimer = null
+
+function startHealthCheck(intervalMs = 30000) {
+  if (healthCheckTimer) clearInterval(healthCheckTimer)
+  healthCheckTimer = setInterval(async () => {
+    try {
+      const connection = await pool.getConnection()
+      await connection.ping()
+      connection.release()
+    } catch (err) {
+      console.error('Database health check failed:', err.message)
+    }
+  }, intervalMs)
+}
+
+function stopHealthCheck() {
+  if (healthCheckTimer) {
+    clearInterval(healthCheckTimer)
+    healthCheckTimer = null
+  }
+}
+
 module.exports = {
   pool,
   query,
   transaction,
   initDatabase,
+  startHealthCheck,
+  stopHealthCheck,
+  getPoolMetrics,
+  getQueryStats,
+  resetQueryStats,
   config
 }