import mysql from 'mysql2/promise';
import dotenv from 'dotenv';

dotenv.config();

async function testConnection() {
  console.log('测试数据库连接...');
  console.log('配置信息:');
  console.log('  Host:', process.env.DB_HOST);
  console.log('  Port:', process.env.DB_PORT);
  console.log('  User:', process.env.DB_USER);
  console.log('  Password:', process.env.DB_PASSWORD ? '***' : '(空)');
  console.log('  Database:', process.env.DB_NAME);
  
  try {
    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 || '',
      database: process.env.DB_NAME || 'employee_performance',
    });
    
    console.log('\n✅ 数据库连接成功！');
    
    // 测试查询
    const [rows] = await connection.query('SELECT COUNT(*) as count FROM user');
    console.log('用户表记录数:', (rows as any)[0].count);
    
    await connection.end();
    process.exit(0);
  } catch (error: any) {
    console.error('\n❌ 数据库连接失败:', error.message);
    console.error('\n请检查:');
    console.error('1. MySQL 服务是否正在运行');
    console.error('2. 用户名和密码是否正确');
    console.error('3. 数据库是否已创建');
    process.exit(1);
  }
}

testConnection();