如何获取PostgreSQL中刚插入的行详情？（使用pg模块）

嘿，我来帮你搞定这个问题！用pg模块在循环里插入数据，同时拿到每一行刚插入的完整详情（包括自动生成的主键id）完全可行——核心就是利用PostgreSQL的RETURNING子句，再配合pg的查询方法就行。下面给你详细的实现方案：

PostgreSQL的RETURNING子句可以在INSERT（以及UPDATE/DELETE）操作完成后，返回受影响行的指定字段。如果写RETURNING *，就能直接拿到插入后的整行数据，其中就包含数据库自动生成的主键id（比如SERIAL或IDENTITY类型的主键）。

如果你的循环插入需要保证数据一致性（要么所有数据都插入成功，要么全部回滚），用事务包裹串行插入是最佳选择。用async/await可以让逻辑更清晰，避免回调地狱：

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

// 初始化连接池，替换成你的数据库配置
const pool = new Pool({
  user: '你的用户名',
  host: 'localhost',
  database: '你的数据库名',
  password: '你的密码',
  port: 5432,
});

async function batchInsert(dataList) {
  const client = await pool.connect();
  try {
    // 开启事务
    await client.query('BEGIN');

    const insertedRows = [];
    for (const item of dataList) {
      // 参数化查询：用$1、$2占位符，防止SQL注入，pg会自动处理类型转换
      const result = await client.query(
        `INSERT INTO 你的表名 (字段1, 字段2) VALUES ($1, $2) RETURNING *`,
        [item.val1, item.val2]
      );
      // result.rows[0]就是刚插入的完整行数据，包含主键id
      insertedRows.push(result.rows[0]);
    }

    // 提交事务，确认所有插入生效
    await client.query('COMMIT');
    return insertedRows;
  } catch (error) {
    // 出错则回滚，撤销所有已执行的插入
    await client.query('ROLLBACK');
    throw error; // 抛出错误让上层处理
  } finally {
    // 无论成功失败，都要释放数据库连接
    client.release();
  }
}

// 调用示例
const testData = [
  { val1: '测试值1', val2: '测试内容1' },
  { val1: '测试值2', val2: '测试内容2' }
];

batchInsert(testData)
  .then(rows => console.log('插入成功的行详情：', rows))
  .catch(err => console.error('插入失败：', err));

如果你的插入操作之间没有依赖，且允许个别插入失败不影响整体，可以用Promise.all实现并行插入，速度会更快：

async function parallelInsert(dataList) {
  const pool = new Pool(); // 用连接池自动管理连接

  // 把每个插入操作包装成Promise
  const insertPromises = dataList.map(item => 
    pool.query(
      `INSERT INTO 你的表名 (字段1, 字段2) VALUES ($1, $2) RETURNING *`,
      [item.val1, item.val2]
    ).then(res => res.rows[0])
  );

  // 等待所有Promise完成，拿到所有插入后的行数据
  const insertedRows = await Promise.all(insertPromises);
  return insertedRows;
}

// 调用示例
parallelInsert(testData)
  .then(rows => console.log('并行插入的行详情：', rows))
  .catch(err => console.error('并行插入失败：', err));

• 必须用参数化查询：绝对不要直接把变量拼接到SQL字符串里，这会导致严重的SQL注入风险，而且pg的参数化查询会自动处理数据类型转换，避免很多类型错误。
• 事务的使用场景：如果你的循环插入是一个业务整体（比如批量导入订单，不能只导入一半），一定要用事务包裹，否则部分插入成功部分失败会导致数据不一致。
• 连接池配置：并行插入时要注意连接池的最大连接数设置，避免同时发起过多请求导致连接耗尽。

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