Node-postgres如何将查询返回的ID二维数组转为一维数组？

好问题！我来给你两种可行的方案，你可以根据自己的场景选择：

1. 直接通过PostgreSQL查询返回一维数组 #

你可以用PostgreSQL内置的array_agg()聚合函数，把所有id值直接聚合为一个一维数组返回。这样查询结果会是单一行，里面包含你需要的一维数组：

const idQuery: pg.QueryArrayConfig = { 
  name: 'get-ids', 
  text: 'SELECT array_agg(id) FROM MySchema.SomeTable', 
  rowMode: "array" 
};
pool.query(idQuery).then((result: pg.QueryResult) => {
  const ids = result.rows[0][0]; // 直接拿到 [1,2]
  console.log(ids);
});

优缺点说明 #

• 优点：数据库层面完成聚合，客户端不需要额外循环处理，逻辑更集中。
• 注意：如果你的表数据量极大，array_agg()会把所有id加载到数据库内存中，可能存在性能隐患，这种情况下更适合用客户端扁平化的方式。

2. 客户端手动扁平化数组 #

这是最直观且通用的方式，利用数组方法提取每个子数组的第一个元素（因为你只查询了id字段，每个行数组只有一个元素）：

const idQuery: pg.QueryArrayConfig = { 
  name: 'get-ids', 
  text: 'SELECT id FROM MySchema.SomeTable', 
  rowMode: "array" 
};
pool.query(idQuery).then((result: pg.QueryResult) => {
  // 两种方式都可以，map更明确（明确取每个行的第一个元素）
  const ids = result.rows.map(row => row[0]); 
  // 或者用flat()，适合确定每个子数组只有一个元素的场景
  // const ids = result.rows.flat();
  console.log(ids); // 得到 [1,2]
});

补充说明 #

如果你移除rowMode: "array"得到对象数组，其实也可以用result.rows.map(row => row.id)得到一维数组，但你提到这种方式不符合需求，所以就不多展开了。

总结 #

• 小数据量场景：两种方法都可以，看你更喜欢逻辑放在数据库还是客户端。
• 大数据量场景：优先选客户端扁平化，避免数据库内存过载。

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