first commit

backend/init-db.ts

@@ -0,0 +1,74 @@
+import mysql from 'mysql2/promise';
+import fs from 'fs';
+import path from 'path';
+import dotenv from 'dotenv';
+
+dotenv.config();
+
+async function initDatabase() {
+  try {
+    console.log('开始初始化数据库...\n');
+    
+    // 先连接到 MySQL（不指定数据库）
+    const connection = await mysql.createConnection({
+      host: process.env.DB_HOST || 'localhost',
+      port: Number(process.env.DB_PORT) || 3306,
+      user: process.env.DB_USER || 'root',
+      password: process.env.DB_PASSWORD || '',
+    });
+    
+    console.log('✓ 已连接到 MySQL 服务器');
+    
+    // 创建数据库
+    await connection.query(`CREATE DATABASE IF NOT EXISTS ${process.env.DB_NAME || 'employee_performance'} DEFAULT CHARACTER SET utf8mb4 DEFAULT COLLATE utf8mb4_unicode_ci`);
+    console.log('✓ 数据库已创建');
+    
+    // 切换到目标数据库
+    await connection.query(`USE ${process.env.DB_NAME || 'employee_performance'}`);
+    
+    // 读取并执行初始化脚本
+    const initSql = fs.readFileSync(path.join(__dirname, 'src/db/init.sql'), 'utf-8');
+    const statements = initSql
+      .split(';')
+      .map(s => s.trim())
+      .filter(s => s.length > 0 && !s.startsWith('--') && !s.startsWith('CREATE DATABASE') && !s.startsWith('USE'));
+    
+    for (const statement of statements) {
+      if (statement.trim()) {
+        await connection.query(statement);
+      }
+    }
+    console.log('✓ 表结构已创建');
+    
+    // 读取并执行种子数据脚本
+    const seedSql = fs.readFileSync(path.join(__dirname, 'src/db/seed.sql'), 'utf-8');
+    const seedStatements = seedSql
+      .split(';')
+      .map(s => s.trim())
+      .filter(s => s.length > 0 && !s.startsWith('--') && !s.startsWith('USE'));
+    
+    for (const statement of seedStatements) {
+      if (statement.trim()) {
+        await connection.query(statement);
+      }
+    }
+    console.log('✓ 测试数据已插入');
+    
+    await connection.end();
+    
+    console.log('\n✅ 数据库初始化完成！');
+    console.log('\n测试账号信息：');
+    console.log('总经理: gm001 / 123456');
+    console.log('技术部经理: mgr001 / 123456');
+    console.log('销售部经理: mgr002 / 123456');
+    console.log('技术部员工: emp001, emp002, emp003 / 123456');
+    console.log('销售部员工: emp004, emp005 / 123456');
+    
+    process.exit(0);
+  } catch (error: any) {
+    console.error('\n❌ 数据库初始化失败:', error.message);
+    process.exit(1);
+  }
+}
+
+initDatabase();