87 lines
2.0 KiB
JavaScript
87 lines
2.0 KiB
JavaScript
const { Sequelize } = require('sequelize');
|
||
const mysql = require('mysql2/promise');
|
||
|
||
// 环境配置
|
||
const NODE_ENV = 'production';
|
||
|
||
// 数据库配置
|
||
const dbConfig = {
|
||
development: {
|
||
host: 'localhost',
|
||
user: 'root',
|
||
password: '123456',
|
||
database: 'rentease'
|
||
},
|
||
production: {
|
||
host: '8.152.207.41',
|
||
user: 'rentease',
|
||
password: 'Wxx@123!',
|
||
database: 'rentease'
|
||
}
|
||
};
|
||
|
||
// 获取当前环境的配置
|
||
const currentConfig = dbConfig[NODE_ENV];
|
||
|
||
// 先创建数据库
|
||
const createDatabase = async () => {
|
||
try {
|
||
// 连接到MySQL服务器
|
||
const connection = await mysql.createConnection({
|
||
host: currentConfig.host,
|
||
user: currentConfig.user,
|
||
password: currentConfig.password
|
||
});
|
||
|
||
// 创建数据库
|
||
await connection.query(`CREATE DATABASE IF NOT EXISTS ${currentConfig.database}`);
|
||
console.log('数据库创建成功');
|
||
await connection.end();
|
||
return true;
|
||
} catch (error) {
|
||
console.error('创建数据库失败:', error);
|
||
return false;
|
||
}
|
||
};
|
||
|
||
// 创建数据库连接 - 使用MySQL
|
||
const sequelize = new Sequelize(
|
||
currentConfig.database, // 数据库名称
|
||
currentConfig.user, // 用户名
|
||
currentConfig.password, // 密码
|
||
{
|
||
host: currentConfig.host,
|
||
dialect: 'mysql',
|
||
logging: false,
|
||
timezone: '+08:00'
|
||
}
|
||
);
|
||
|
||
// 测试数据库连接
|
||
const testConnection = async () => {
|
||
try {
|
||
await createDatabase();
|
||
// 重新创建sequelize实例，确保连接管理器是活跃的
|
||
const newSequelize = new Sequelize(
|
||
currentConfig.database, // 数据库名称
|
||
currentConfig.user, // 用户名
|
||
currentConfig.password, // 密码
|
||
{
|
||
host: currentConfig.host,
|
||
dialect: 'mysql',
|
||
logging: false,
|
||
timezone: '+08:00'
|
||
}
|
||
);
|
||
await newSequelize.authenticate();
|
||
console.log('数据库连接成功');
|
||
return newSequelize;
|
||
} catch (error) {
|
||
console.error('数据库连接失败:', error);
|
||
return sequelize;
|
||
}
|
||
};
|
||
|
||
// 导出sequelize实例
|
||
module.exports = sequelize;
|