Mongoose中replyTo关联多集合时填充失败问题排查

场景描述 #

通过messages = await ChatMessage.find()获取消息数组，每条消息的replyTo字段定义为mongoose.Schema.Types.ObjectId，原本仅关联chattingMessage文档，但实际数据库中该字段同时存储了chattingMessage（对应模型ChatMessage）和groupMessage（对应模型Message）两种类型的引用。

实现的填充函数能正确填充ChatMessage类型的引用，但Message类型的引用填充后，replyTo字段仍保留ObjectId值，未被替换为查询到的文档对象——尽管终端日志显示已成功找到对应的replyMessage。

填充函数代码 #

// chattingMessage -> ChatMessage , groupMessage -> Message
async function populateReplyTo(messages) {
  const populatedMessages = await Promise.all(messages.map(async (message) => {
    if (message.replyTo) {
      let replyMessage = await ChatMessage.findById(message.replyTo);
      if (!replyMessage) replyMessage = await Message.findById(message.replyTo);

      if (replyMessage) {
        console.log('Found replyMessage:', replyMessage); // 日志能正确打印对象
        message.replyTo = replyMessage; // 赋值不生效？
      } else {
        console.log('No replyMessage found, assigning null'); 
        message.replyTo = null;
      }
    }

    if (message.replyTo && message.replyTo.sender) {
      const replySender = await User.findById(message.replyTo.sender).select('_id username name');
      if (replySender) {
        message.replyTo.sender = replySender; 
      }
    }
    
    return message;
  }));
  console.log(populatedMessages)
  return populatedMessages;
}

终端日志 #

Found replyMessage: {
  _id: new ObjectId("6714e824c212c22e48b96c89"),
  sender: new ObjectId("6714e1acc212c22e48b92089"),
  content: 'hii',
  chat: new ObjectId("66604d0079ebab20eec4da73"),
  replyTo: null,
  status: 'Read',
  messageType: 'ChatMessage',
  deletedBy: [],
  reports: [],
  isDeleted: false,
  attachments: [],
  createdAt: 2024-10-20T11:23:16.998Z,
  updatedAt: 2024-10-23T13:45:07.054Z,
  __v: 0
} but populated one is  not assigned , instead i only find reference in replyTo

[
 {
    _id: new ObjectId("671681fada25dcde60b63526"),
    sender: new ObjectId("66dafffec77d3018b2b0ddb3"),
    receiver: new ObjectId("6714e1acc212c22e48b92089"),
    messageType: 'ChatMessage',
    subscriptionRequestMessageAction: 'none',
    message: 'where to start',
    replyTo: new ObjectId("9023e824c212c22e48b96c89"), // 仍为ObjectId引用，未替换为对象
    status: 'Read',
    deletedBy: [],
    reports: [],
    createdAt: 2024-10-21T16:31:54.514Z,
    updatedAt: 2024-10-23T15:47:24.551Z,
    __v: 0
  }
]

原因分析 #

核心问题是Mongoose返回的message是Document实例，而非普通JavaScript对象：

1. Schema中replyTo被定义为ObjectId类型，直接赋值文档对象会触发Mongoose的字段类型校验，导致修改被拦截，无法生效。
2. 日志中显示的replyMessage ID与最终message的replyTo ID不一致，需确认是否存在处理消息不匹配的情况（如循环或数据混淆）。

解决方案 #

方案1：转换为普通JS对象后修改 #

将Mongoose文档转换为普通对象，脱离Schema约束后即可自由修改字段：

async function populateReplyTo(messages) {
  const populatedMessages = await Promise.all(messages.map(async (messageDoc) => {
    // 转换为普通JS对象
    const message = messageDoc.toObject();
    if (message.replyTo) {
      let replyMessage = await ChatMessage.findById(message.replyTo);
      if (!replyMessage) replyMessage = await Message.findById(message.replyTo);

      if (replyMessage) {
        // 同样将replyMessage转为普通对象
        message.replyTo = replyMessage.toObject();
      } else {
        message.replyTo = null;
      }
    }

    if (message.replyTo && message.replyTo.sender) {
      const replySender = await User.findById(message.replyTo.sender).select('_id username name');
      if (replySender) {
        message.replyTo.sender = replySender.toObject();
      }
    }
    
    return message;
  }));
  return populatedMessages;
}

方案2：使用Mongoose的set方法修改字段 #

若需保留Document实例，可使用set方法绕过直接赋值的类型校验（注意：若后续要保存文档，可能因Schema类型不匹配报错，仅适用于返回前端的场景）：

// 在原函数中替换赋值语句
message.set('replyTo', replyMessage);

额外注意 #

检查日志中replyMessage的ID与目标message.replyTo的ID是否匹配，确保处理的是同一条消息，避免数据混淆。

内容的提问来源于stack exchange，提问作者komal sai