如何在node-postgres的pool.query回调外部获取可遍历操作的查询结果对象

你现在遇到的是JavaScript典型的异步执行时序不匹配问题：

• pool.query 是异步API，执行时不会阻塞后续代码运行。实际代码运行顺序和直觉里的线性顺序完全不一样：
  1. 初始化空的sqlOut数组
  2. 发起数据库查询请求，把回调函数暂存到事件循环队列，不等待查询结果返回
  3. 直接向下走到console.log(sqlOut)代码行——这时候数据库还在完成连接、查询、返回数据的流程，回调根本没执行，自然拿到的就是初始状态的空数组
  4. 等所有数据库操作完成、查询结果返回后，才会运行回调内部的赋值逻辑，这时候你的打印操作早就执行完了

node-postgres 从8.x版本开始原生支持Promise接口，不需要硬写嵌套回调。最贴合从Python转过来的线性逻辑习惯的写法是搭配async/await，写出来的代码和同步执行逻辑几乎没有区别：

const { Pool } = require('pg');

// 把数据库查询逻辑封装成async函数
async function getTableColumns(tableName) {
  const pool = new Pool({
    user: 'postgres',
    host: 'localhost',
    database: 'pps',
    password: 'XXXX',
    port: 5432,
  });

  try {
    // 用await挂起当前函数执行，直接拿到查询结果，无需写回调
    const queryResult = await pool.query({
      text: `Select column_name, data_type, character_maximum_length from information_schema.columns where table_name = $1`,
      rowMode: 'array',
      values: [tableName]
    });
    // 这里的queryResult.rows就是你之前想存在sqlOut里的内容
    return queryResult.rows;
  } catch (err) {
    console.error('执行查询出错', err.stack);
    throw err;
  } finally {
    // 无论查询成功/失败，最后主动关闭连接池，避免进程一直挂住
    await pool.end();
  }
}

// 拿到结果后再执行后续分析逻辑
getTableColumns('profiling_nws6')
  .then(sqlOut => {
    // 这里可以任意对sqlOut做遍历、生成报告、执行后续列分析的操作
    console.log('拿到的表列信息：', sqlOut);
    // 比如遍历每一列执行9项分析查询的逻辑，就可以写在这
    sqlOut.forEach(([columnName]) => {
      console.log('准备对列', columnName, '执行分析查询');
    })
  })

注意事项 #

• 不要依赖回调外部的变量赋值来拿到异步结果：这是JS异步执行环境和Python同步逻辑最核心的区别，Python中调用数据库查询会阻塞代码直到结果返回，JS中如果不主动等待结果，代码会直接向下执行
• 示例里用了参数化占位符$1代替直接拼接表名到SQL字符串，既可以避免SQL注入风险，也能适配不同的表名输入
• 所有数据库操作完成后，一定要调用pool.end()释放连接池资源，不然会导致Node进程无法正常退出

如果一定要保留原来的回调写法，所有依赖sqlOut的后续逻辑，都必须移动到pool.query的回调函数内部执行，不能写到回调外部：

pool.query({ text: sql, rowMode: 'array' }, (err, result) => {
  if (err) {
    return console.error('Error executing query', err.stack);
  }
  const sqlOut = result.rows.slice(0);
  // 所有需要用到sqlOut的操作，全部写在这下面
  console.log('查询结果：', sqlOut);
  sqlOut.forEach(/* 编写列分析逻辑 */);
  // 所有操作完成后再关闭连接池
  pool.end();
});
// 这里的代码永远在查询返回前执行，不要写依赖sqlOut的逻辑

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