Mongoose 中事务与 populate 操作的并发问题解析

事务环境下 populate 操作的陷阱

在使用 Mongoose 操作 MongoDB 数据库时，开发者经常会遇到需要在事务中执行查询并填充关联文档的场景。然而，当我们在事务中先查询文档再附加会话(session)进行 populate 操作时，可能会遇到一个隐蔽的问题——"TransientTransactionError"错误。

问题现象

典型的表现是：在一个事务中，首先不使用会话查询文档，然后通过$session()方法为文档附加会话，最后调用 populate 方法填充多个关联字段时，系统有约30%的概率会抛出以下错误：

MongoServerError: Given transaction number 5 does not match any in-progress transactions. The active transaction number is 4

这个错误表明 MongoDB 服务器端的事务编号与客户端请求不匹配，导致事务无法继续执行。

问题根源

深入分析这个问题，我们可以发现几个关键点：

1. Mongoose 的 populate 实现机制：默认情况下，Mongoose 会并行执行所有 populate 路径的查询操作，以提高性能。

2. MongoDB 事务限制：MongoDB 的事务模型在同一时间只允许执行一个操作，不支持并发操作。

3. 会话附加时机：当文档在初始查询时未使用会话，后续附加会话进行 populate 时，并行查询会违反事务的单操作限制。

解决方案

针对这个问题，目前有以下几种解决方案：

1. 串行化 populate 操作：将并行的 populate 改为串行执行，逐个填充关联字段：

await docD.populate('refA');
await docD.populate('refB');
await docD.populate('refC');

2. 初始查询时附加会话：在第一次查询时就使用会话，确保所有操作都在同一事务上下文中：

const docD = await modelD.findOne().session(session);

3. 等待官方修复：Mongoose 团队已经意识到这个问题，正在开发支持事务环境下串行 populate 的功能。

最佳实践建议

在事务中使用 populate 时，建议开发者：

1. 始终在查询链的最开始附加会话，而不是后续操作中才添加。

2. 对于复杂的关联查询，考虑使用聚合管道(aggregation pipeline)替代 populate，可以更好地控制事务行为。

3. 在事务中避免同时填充过多关联字段，必要时可以分步执行。

4. 关注 Mongoose 的版本更新，及时获取对事务支持的改进。

总结

这个问题揭示了 MongoDB 事务模型与 Mongoose ORM 便捷功能之间的微妙冲突。理解这些底层机制有助于开发者编写更健壮的数据库操作代码，特别是在需要事务保证的场景下。随着 Mongoose 的持续改进，这类问题将得到更好的解决，但目前开发者需要了解这些限制并采取适当的应对措施。