Postgres.js 中获取插入记录ID的最佳实践

在使用Postgres.js进行数据库操作时，获取新插入记录的自动生成ID是一个常见需求。本文将探讨几种优雅的实现方式，帮助开发者更好地处理这一场景。

问题背景

当我们向PostgreSQL数据库插入新记录时，经常需要获取数据库自动生成的ID（通常是SERIAL或IDENTITY列）。在Postgres.js中，直接的方式是通过RETURNING子句查询ID，然后将其赋给现有对象。

基本解决方案

最简单的实现方式是直接访问返回结果数组的第一个元素：

gameType.id = (await this.db `INSERT INTO game_types ${this.db(gameType)} RETURNING id`)[0].id;

这种方式虽然有效，但代码可读性较差，且容易出错。

改进方案：解构赋值

更优雅的方式是使用ES6的解构赋值语法：

const [{id: gtId}] = await this.db`INSERT INTO game_types ${this.db(gameType)} RETURNING id`;
gameType.id = gtId;

这种方法：

1. 直接从返回数组中解构出第一个对象
2. 将其id属性重命名为gtId变量
3. 将gtId赋值给gameType.id

完整对象返回方案

如果需要获取插入记录的完整信息，可以RETURNING *：

const [insertedGameType] = await this.db`INSERT INTO game_types ${this.db(gameType)} RETURNING *`;
Object.assign(gameType, insertedGameType);

这种方式一次性获取所有字段，并使用Object.assign合并到原有对象中。

最佳实践建议

1. 对于简单场景，使用解构赋值获取特定字段（如ID）
2. 需要完整记录时，RETURNING *并合并对象更高效
3. 考虑在业务层封装这些操作，避免重复代码
4. 注意错误处理，确保插入操作成功后再访问结果

这些方法不仅提高了代码可读性，也减少了潜在的错误风险，是Postgres.js开发中的推荐实践。