Node.js模块导出前等待密钥库返回密码创建数据库池问题

Node.js模块加载是同步流程，直接在模块导出阶段创建依赖异步密码的连接池必然会因为密码未就绪导致失败。以下是具体的解决方案：

1. 修改pool.js：异步创建并导出连接池Promise #

首先确保从密钥库获取密码的函数返回Promise（绝大多数密钥库SDK都是异步API），再将连接池的创建逻辑包装为异步函数，最后导出连接池的Promise实例：

const mysql = require('mysql');
// 假设serverCa已正确定义，比如读取自文件或环境变量
const serverCa = '你的CA证书内容';

// 异步从密钥库获取密码，替换为实际密钥库调用逻辑
const getPWrd = async () => {
  // 示例：调用AWS Secrets Manager或其他密钥库的异步API
  // const secret = await secretsManager.getSecretValue({SecretId: 'db-password'});
  // return JSON.parse(secret.SecretString).password;
  return Promise.resolve('从密钥库拿到的实际密码');
};

const getDBPool = async () => {
  const password = await getPWrd();
  const dbPool = mysql.createPool({
    host: '你的数据库主机',
    database: '目标数据库名',
    user: '数据库用户名',
    password,
    connectionLimit: 10,
    ssl: {
      rejectUnauthorized: true,
      ca: serverCa,
    },
  });

  // 可选：验证连接池有效性，避免返回无效实例
  return new Promise((resolve, reject) => {
    dbPool.getConnection((err, conn) => {
      if (err) {
        reject(err);
        return;
      }
      conn.release();
      resolve(dbPool);
    });
  });
};

// 导出连接池Promise，确保只创建一次连接池
module.exports = getDBPool();

2. 修改db.js：等待连接池就绪后再使用 #

由于导出的是Promise，使用时必须等待其解析完成才能调用连接池的方法：

const dbPoolPromise = require('./pool.js');

// 封装异步查询方法
async function dbQuery(sql, params = []) {
  const dbPool = await dbPoolPromise;
  return new Promise((resolve, reject) => {
    dbPool.query(sql, params, (err, results) => {
      if (err) reject(err);
      else resolve(results);
    });
  });
}

// 调用示例
dbQuery('SELECT * FROM your_table')
  .then(results => console.log('查询结果:', results))
  .catch(err => console.error('查询失败:', err));

// 在async函数中使用
async function businessLogic() {
  try {
    const data = await dbQuery('SELECT * FROM your_table WHERE id = ?', [1]);
    // 处理业务逻辑
  } catch (err) {
    console.error('业务逻辑执行失败:', err);
  }
}

核心逻辑说明 #

• 模块加载阶段无法等待异步操作完成，因此必须将连接池的创建转为异步流程，通过Promise导出就绪的连接池实例。
• 导出的Promise只会执行一次getDBPool，确保全局只有一个连接池实例，避免资源浪费。
• 所有数据库操作都需要等待连接池Promise解析完成，保证使用的是已正确配置密码的连接池。

内容的提问来源于stack exchange，提问作者Jelly Donut