database.js 710 B

12345678910111213141516171819202122232425
  1. const mysql = require('mysql2/promise');
  2. const pool = mysql.createPool({
  3. host: process.env.DB_HOST || '127.0.0.1',
  4. user: process.env.DB_USER || 'notebot',
  5. password: process.env.DB_PASSWORD || 'NZWdAr4neDPrWm5k',
  6. database: process.env.DB_NAME || 'notebot',
  7. waitForConnections: true,
  8. connectionLimit: 10,
  9. queueLimit: 0
  10. });
  11. // 测试数据库连接
  12. const testConnection = async () => {
  13. try {
  14. const connection = await pool.getConnection();
  15. console.log('数据库连接成功');
  16. connection.release();
  17. } catch (error) {
  18. console.error('数据库连接失败:', error);
  19. process.exit(1);
  20. }
  21. };
  22. module.exports = { pool, testConnection };