Drizzle ORM 中获取 MySQL 自增 ID 的最佳实践

在数据库操作中，获取新插入记录的自增 ID 是一个常见需求。Drizzle ORM 作为一款现代化的 TypeScript ORM 工具，提供了简洁高效的方式来处理这个问题。

问题背景

当开发者使用 MySQL 数据库并启用了自增 ID 功能时，在插入新记录后通常需要立即获取该记录生成的 ID。这个需求在业务逻辑中非常普遍，比如在创建用户后需要返回用户 ID，或者在创建订单后需要引用订单 ID。

Drizzle ORM 的解决方案

Drizzle ORM 通过 MySQL 驱动返回的结果对象提供了获取自增 ID 的方式。具体实现如下：

const [result] = await db.insert(users)
  .values({
    name: 'John Doe',
    email: '[e-mail]'
  });

const newUserId = result.insertId;

技术细节解析

1. 返回值结构：Drizzle 的 insert 操作返回一个数组，第一个元素是包含操作结果的对象
2. insertId 属性：结果对象中的 insertId 属性包含了最新生成的自增 ID
3. 批量插入处理：当执行批量插入时，insertId 会返回第一个插入记录的 ID

实际应用场景

这种获取自增 ID 的方式特别适用于以下场景：

1. 关联数据创建：先创建主表记录，获取 ID 后再创建关联表记录
2. API 响应：在 RESTful API 中返回新建资源的 ID
3. 日志记录：将新记录的 ID 用于审计日志

性能考量

使用这种方式获取自增 ID 几乎不会带来额外的性能开销，因为：

1. MySQL 服务器在执行插入操作时已经生成了 ID
2. 客户端驱动在返回结果时已经包含了这个信息
3. 不需要额外的查询语句

最佳实践建议

1. 错误处理：始终检查操作是否成功，再访问 insertId
2. 类型安全：在 TypeScript 中，可以显式声明 insertId 的类型为 number
3. 事务处理：在事务中使用时，确保整个操作原子性

通过这种方式，Drizzle ORM 为开发者提供了简洁而强大的自增 ID 获取能力，使得数据库操作更加高效和直观。