services/config/database.js

const mysql = require('mysql2/promise')
require('dotenv').config()

const config = {
  host: process.env.DB_HOST || '110.42.255.239',
  port: parseInt(process.env.DB_PORT || '3306'),
  user: process.env.DB_USER || 'admin',
  password: process.env.DB_PASSWORD || 'Admin@123',
  database: process.env.DB_NAME || 'miniprogram',
  waitForConnections: true,
  connectionLimit: 10,
  queueLimit: 0
}
/*
# 登录服务器，找到 Koa 进程
ssh ubuntu@110.42.255.239
pm2 restart weixin  
*/
const pool = mysql.createPool(config)

async function query(sql, params = []) {
  const connection = await pool.getConnection()
  try {
    const [rows, fields] = await connection.execute(sql, params)
    return rows
  } finally {
    connection.release()
  }
}

async function initDatabase() {
  try {
    const connection = await mysql.createConnection({
      host: config.host,
      port: config.port,
      user: config.user,
      password: config.password,
      multipleStatements: true
    })
    
    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)
    }
    
    await connection.end()
    console.log('Database initialized successfully')
  } catch (error) {
    console.error('Failed to initialize database:', error)
    throw error
  }
}

module.exports = {
  pool,
  query,
  initDatabase,
  config
}