70 lines
1.9 KiB
JavaScript
70 lines
1.9 KiB
JavaScript
|
/**
|
|||
|
|
* 数据库初始化脚本
|
|||
|
|
* 运行: npm run init-db
|
|||
|
|
*/
|
|||
|
|
const fs = require('fs');
|
|||
|
|
const path = require('path');
|
|||
|
|
const mysql = require('mysql2/promise');
|
|||
|
|
require('dotenv').config();
|
|||
|
|
|
|||
|
|
async function initDatabase() {
|
|||
|
|
console.log('开始初始化数据库...');
|
|||
|
|
|
|||
|
|
// 先连接到MySQL（不指定数据库）
|
|||
|
|
const connection = await mysql.createConnection({
|
|||
|
|
host: process.env.DB_HOST || 'localhost',
|
|||
|
|
port: process.env.DB_PORT || 3306,
|
|||
|
|
user: process.env.DB_USER || 'root',
|
|||
|
|
password: process.env.DB_PASSWORD || '',
|
|||
|
|
multipleStatements: true
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
// 读取并执行schema.sql
|
|||
|
|
console.log('创建数据库表结构...');
|
|||
|
|
const schemaSQL = fs.readFileSync(
|
|||
|
|
path.join(__dirname, 'schema.sql'),
|
|||
|
|
'utf8'
|
|||
|
|
);
|
|||
|
|
await connection.query(schemaSQL);
|
|||
|
|
console.log('表结构创建成功！');
|
|||
|
|
|
|||
|
|
// 读取并执行种子数据
|
|||
|
|
console.log('导入种子故事数据（第1部分）...');
|
|||
|
|
const seedSQL1 = fs.readFileSync(
|
|||
|
|
path.join(__dirname, 'seed_stories_part1.sql'),
|
|||
|
|
'utf8'
|
|||
|
|
);
|
|||
|
|
await connection.query(seedSQL1);
|
|||
|
|
console.log('种子数据第1部分导入成功！');
|
|||
|
|
|
|||
|
|
console.log('导入种子故事数据（第2部分）...');
|
|||
|
|
const seedSQL2 = fs.readFileSync(
|
|||
|
|
path.join(__dirname, 'seed_stories_part2.sql'),
|
|||
|
|
'utf8'
|
|||
|
|
);
|
|||
|
|
await connection.query(seedSQL2);
|
|||
|
|
console.log('种子数据第2部分导入成功！');
|
|||
|
|
|
|||
|
|
console.log('\n数据库初始化完成！');
|
|||
|
|
console.log('共创建10个种子故事，包含66个剧情节点和多个结局分支。');
|
|||
|
|
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('数据库初始化失败:', error.message);
|
|||
|
|
throw error;
|
|||
|
|
} finally {
|
|||
|
|
await connection.end();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 运行初始化
|
|||
|
|
initDatabase()
|
|||
|
|
.then(() => {
|
|||
|
|
console.log('\n可以启动服务器了: npm start');
|
|||
|
|
process.exit(0);
|
|||
|
|
})
|
|||
|
|
.catch((err) => {
|
|||
|
|
console.error('初始化过程中出现错误:', err);
|
|||
|
|
process.exit(1);
|
|||
|
|
});
|