Express中如何为并发请求的Mongoose查询绑定对应RequestId？

问题根源 #

之前的实现每次请求都会覆盖全局的mongoose.set("debug")回调，并发请求时，后进入的请求会替换掉前一个请求的回调逻辑，导致前一个请求的异步查询执行时，使用的是后一个请求的req.requestId。

解决方案：利用Node.js AsyncLocalStorage 维护请求上下文 #

AsyncLocalStorage 能为每个请求的异步调用链提供独立的存储空间，确保并发请求的上下文不会互相干扰，是解决这类异步上下文跟踪问题的标准方案。

步骤1：创建上下文存储模块 #

新建context.js，封装AsyncLocalStorage实例：

import { AsyncLocalStorage } from 'async_hooks';

export const asyncLocalStorage = new AsyncLocalStorage();

步骤2：修改请求中间件，存入requestId到上下文 #

不再在中间件中修改Mongoose的全局debug配置，而是将requestId存入异步上下文：

import { asyncLocalStorage } from './context.js';

export default (req, res, next) => {
  asyncLocalStorage.run(new Map(), () => {
    // 将当前请求的requestId存入上下文
    asyncLocalStorage.getStore().set('requestId', req.requestId);
    next();
  });
};

步骤3：全局配置Mongoose debug函数，从上下文取requestId #

在项目启动时（比如app.js或数据库初始化文件）全局设置一次Mongoose的debug回调，从异步上下文中获取当前请求的requestId：

import mongoose from 'mongoose';
import { asyncLocalStorage } from './context.js';
import Logger from './your-logger-path.js';

mongoose.set("debug", (collectionName, methodName, ...methodArgs) => {
  const store = asyncLocalStorage.getStore();
  // 若不存在上下文（比如非请求触发的查询），用默认值标识
  const requestId = store ? store.get('requestId') : 'non-request';
  
  Logger.info({
    label: "Mongoose",
    requestId: requestId,
    message: `${collectionName}.${methodName}(${methodArgs.join(', ')})`,
  });
});

替代方案：手动为查询附加requestId（不推荐） #

如果无法使用AsyncLocalStorage（比如Node.js版本过低），可以在每个Mongoose查询时手动传递requestId，并通过Mongoose的查询钩子提取：

• 在请求中间件中，将req.requestId挂到res.locals或自定义对象
• 执行查询时，通过mongoose.Query.prototype.setOptions或自定义方法附加requestId
• 在Mongoose的pre钩子中获取该requestId并记录

但这种方式需要侵入所有查询代码，维护成本高，优先推荐AsyncLocalStorage方案。

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