yunrui_asset/backend/db.js

const { Sequelize } = require('sequelize');
const logger = require('./utils/logger').module('Sequelize');

// 获取数据库配置
const DB_TYPE = process.env.DB_TYPE || 'sqlite'; // 数据库类型：sqlite 或 mysql

// 根据数据库类型创建连接
let sequelize;
let dbDialect = 'sqlite'; // 实际使用的数据库类型

/**
 * Sequelize SQL 日志函数
 * 仅输出 debug 级别日志，生产环境自动过滤
 */
const sqlLogger = (msg) => logger.debug(msg);

if (DB_TYPE === 'mysql') {
  // MySQL 配置
  sequelize = new Sequelize(
    process.env.MYSQL_DATABASE || 'idc_management',
    process.env.MYSQL_USERNAME || 'root',
    process.env.MYSQL_PASSWORD || '',
    {
      host: process.env.MYSQL_HOST || 'localhost',
      port: parseInt(process.env.MYSQL_PORT) || 3306,
      dialect: 'mysql',
      logging: sqlLogger,
      // 连接池配置 - 提升并发处理能力
      pool: {
        max: 10, // 最大连接数
        min: 2, // 最小连接数
        acquire: 30000, // 获取连接超时时间(ms)
        idle: 10000, // 连接空闲时间(ms)
      },
    }
  );
  dbDialect = 'mysql';
} else {
  sequelize = new Sequelize({
    dialect: 'sqlite',
    storage: process.env.DB_PATH || './idc_management.db',
    logging: sqlLogger,
    // SQLite 连接池配置
    pool: {
      max: 5,
      min: 1,
      acquire: 30000,
      idle: 10000,
    },
  });
  dbDialect = 'sqlite';

  // 启用 SQLite 外键约束支持
  sequelize.query('PRAGMA foreign_keys = ON');
}

module.exports = { sequelize, DB_TYPE, dbDialect };