103 lines
2.5 KiB
JavaScript
103 lines
2.5 KiB
JavaScript
|
const mysql = require('mysql2');
|
|||
|
|
|
|||
|
|
// 数据库配置
|
|||
|
|
const configs = [
|
|||
|
|
{
|
|||
|
|
name: '测试环境',
|
|||
|
|
host: '192.168.0.240',
|
|||
|
|
port: 3306,
|
|||
|
|
user: 'root',
|
|||
|
|
password: 'aiot$Aiot123'
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: '生产环境',
|
|||
|
|
host: '129.211.213.226',
|
|||
|
|
port: 9527,
|
|||
|
|
user: 'root',
|
|||
|
|
password: 'aiotAiot123!'
|
|||
|
|
}
|
|||
|
|
];
|
|||
|
|
|
|||
|
|
// 简单的SQL语句（避免编码问题）
|
|||
|
|
const sqlStatements = [
|
|||
|
|
"CREATE DATABASE IF NOT EXISTS jiebandata CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci",
|
|||
|
|
"USE jiebandata",
|
|||
|
|
`CREATE TABLE IF NOT EXISTS users (
|
|||
|
|
id INT AUTO_INCREMENT PRIMARY KEY,
|
|||
|
|
openid VARCHAR(64) UNIQUE NOT NULL,
|
|||
|
|
nickname VARCHAR(50) NOT NULL,
|
|||
|
|
avatar VARCHAR(255),
|
|||
|
|
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
|||
|
|
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP
|
|||
|
|
) ENGINE=InnoDB`
|
|||
|
|
];
|
|||
|
|
|
|||
|
|
function executeSQL(connection, sql, description) {
|
|||
|
|
return new Promise((resolve, reject) => {
|
|||
|
|
console.log(description);
|
|||
|
|
connection.query(sql, (err, results) => {
|
|||
|
|
if (err) {
|
|||
|
|
console.error('失败:', err.message);
|
|||
|
|
reject(err);
|
|||
|
|
} else {
|
|||
|
|
console.log('成功');
|
|||
|
|
resolve(results);
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async function setupDatabase(config) {
|
|||
|
|
console.log(`\n开始设置 ${config.name} 数据库...`);
|
|||
|
|
|
|||
|
|
const connection = mysql.createConnection({
|
|||
|
|
host: config.host,
|
|||
|
|
port: config.port,
|
|||
|
|
user: config.user,
|
|||
|
|
password: config.password
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
await new Promise((resolve, reject) => {
|
|||
|
|
connection.connect((err) => {
|
|||
|
|
if (err) reject(err);
|
|||
|
|
else resolve();
|
|||
|
|
});
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
for (let i = 0; i < sqlStatements.length; i++) {
|
|||
|
|
await executeSQL(connection, sqlStatements[i], `执行SQL ${i + 1}/${sqlStatements.length}`);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
console.log('✅ 数据库设置完成');
|
|||
|
|
return true;
|
|||
|
|
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('❌ 数据库设置失败:', error.message);
|
|||
|
|
return false;
|
|||
|
|
} finally {
|
|||
|
|
connection.end();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async function main() {
|
|||
|
|
console.log('🎯 结伴客数据库初始化');
|
|||
|
|
console.log('='.repeat(50));
|
|||
|
|
|
|||
|
|
let successCount = 0;
|
|||
|
|
|
|||
|
|
for (const config of configs) {
|
|||
|
|
const success = await setupDatabase(config);
|
|||
|
|
if (success) successCount++;
|
|||
|
|
console.log('\n' + '='.repeat(50));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
console.log(`📊 完成: ${successCount}/${configs.length} 个环境成功`);
|
|||
|
|
|
|||
|
|
if (successCount > 0) {
|
|||
|
|
console.log('\n🎉 数据库初始化完成！');
|
|||
|
|
console.log('现在可以运行测试脚本来验证数据库结构。');
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
main().catch(console.error);
|