Drizzle ORM 事务中处理外键关联插入的注意事项

在使用 Drizzle ORM 进行数据库操作时，事务处理是保证数据一致性的重要机制。特别是在处理具有外键关联的多表插入场景时，开发者需要特别注意事务的正确使用方式。

典型场景分析

考虑一个典型的数据库设计场景：我们有两个相关联的表，dbtMergeConflicts 和 dbtMergeConflictFiles，它们通过外键建立关联。dbtMergeConflictFiles 表中的 dbtMergeConflictId 字段引用了 dbtMergeConflicts 表的主键。

这种设计模式在实际开发中非常常见，特别是在需要记录主从关系的业务场景中。例如，一个订单系统可能包含订单主表和订单明细表，或者一个博客系统可能包含文章主表和评论表。

常见错误模式

许多开发者在使用 Drizzle ORM 进行此类操作时，会编写类似下面的代码：

await db.transaction(async tx => {
    const res = await tx.insert(dbtMergeConflicts).values({
        dbtProjectId: projectId
    }).returning({id: dbtMergeConflicts.id})
    
    await db.insert(dbtMergeConflictFiles).values({
        dbtMergeConflictId: res[0]!.id,
        filePath: 'some_path'
    })
})

这段代码看似合理，但实际上存在一个关键问题：在事务回调函数内部，第二个插入操作没有使用事务对象 tx，而是直接使用了全局的 db 对象。这会导致第二个插入操作在事务外部执行，从而引发外键约束冲突。

问题本质

这个问题的本质在于事务隔离性。当使用 db 而非 tx 执行第二个插入时：

1. 第一个插入操作虽然已经提交到事务中，但事务尚未最终提交
2. 第二个插入操作在事务外执行，无法"看到"事务内未提交的数据
3. 数据库引擎会检查外键约束，发现引用的主键记录不存在

正确实现方式

正确的做法是在事务回调函数内统一使用事务对象 tx 执行所有操作：

await db.transaction(async tx => {
    const res = await tx.insert(dbtMergeConflicts).values({
        dbtProjectId: projectId
    }).returning({id: dbtMergeConflicts.id})
    
    await tx.insert(dbtMergeConflictFiles).values({
        dbtMergeConflictId: res[0]!.id,
        filePath: 'some_path'
    })
})

这种写法确保了：

• 所有操作都在同一个事务上下文中执行
• 第二个插入操作能够正确引用第一个插入操作生成的主键
• 整个操作具有原子性，要么全部成功，要么全部回滚

扩展思考

在实际开发中，处理关联表插入时还需要考虑以下几点：

1. 批量插入性能：当需要插入大量关联记录时，可以考虑使用批量插入优化性能
2. 错误处理：应该在事务中添加适当的错误处理逻辑，确保异常时能够正确回滚
3. 事务隔离级别：了解不同隔离级别对并发操作的影响，根据业务需求选择合适的级别
4. 返回结果处理：合理设计 returning 子句，获取必要的插入结果用于后续操作

总结

Drizzle ORM 提供了强大的事务支持，但在使用时必须确保所有相关操作都在同一事务上下文中执行。特别是在处理外键关联的插入操作时，统一使用事务对象是避免约束冲突的关键。理解事务的隔离特性，遵循一致的事务使用模式，才能确保数据库操作的完整性和一致性。