如何从MongoDB文档的数组字段中提取指定ID的对象？

问题场景 #

现有MongoDB用户文档结构如下：

{
  _id: ObjectId("630e2a2250283de03b2dc920"),
  fullName: "John",
  phone: "+1234...",
  createdAt: ISODate("2022-08-30T15:17:54.608Z"),
  selectedClients: [
    {
      phone: "+4567...",
      fullName: "Client1",
      _id: ObjectId("630e2a8f8367a2aaac3343b4"),
      createdAt: ISODate("2022-08-30T15:19:43.372Z")
    },
    {
      phone: "+7890...",
      fullName: "Client2",
      _id: ObjectId("630e2b73d42ddc8d622e860f"),
      createdAt: ISODate("2022-08-30T15:23:31.883Z")
    }
  ],
  __v: 0
}

已知用户ID和客户端ID，希望仅获取指定client_id对应的Client对象（比如Client1），但当前代码返回的是完整用户模型，而非目标对象：

exports.removeTrustee = asyncHandler(async (req, res, next) => {
  const user_id = req.params.user_id.split("-")[1];
  const client_id = ObjectId(req.params.client_id.split("-")[1]);

  const client = await User.find({
     selectedClients: { $elemMatch: { _id: client_id } },
  });
});

解决方案 #

注意：你的原始代码缺少用户ID的查询条件，会返回所有包含目标client的用户，必须加上_id: user_id来精准定位目标用户。以下是几种可行的实现方式：

方法1：findOne配合投影（Projection） #

通过MongoDB的投影语法，只返回匹配的数组元素：

exports.removeTrustee = asyncHandler(async (req, res, next) => {
  const user_id = req.params.user_id.split("-")[1];
  const client_id = ObjectId(req.params.client_id.split("-")[1]);

  // 定位用户并仅返回匹配的selectedClients元素
  const result = await User.findOne(
    { 
      _id: user_id,
      selectedClients: { $elemMatch: { _id: client_id } }
    },
    { "selectedClients.$": 1, _id: 0 } // $表示匹配到的第一个数组元素，_id:0排除用户ID
  );

  // 提取目标client对象
  const targetClient = result?.selectedClients?.[0];
  res.status(200).json(targetClient);
});

方法2：聚合管道（Aggregation Pipeline） #

适合更复杂的数组筛选与结构重构：

exports.removeTrustee = asyncHandler(async (req, res, next) => {
  const user_id = req.params.user_id.split("-")[1];
  const client_id = ObjectId(req.params.client_id.split("-")[1]);

  const targetClient = await User.aggregate([
    { $match: { _id: ObjectId(user_id) } }, // 定位目标用户
    { $unwind: "$selectedClients" }, // 展开数组为单个文档
    { $match: { "selectedClients._id": client_id } }, // 筛选目标client
    { 
      $project: { // 自定义返回字段，排除冗余信息
        _id: 0,
        phone: "$selectedClients.phone",
        fullName: "$selectedClients.fullName",
        clientId: "$selectedClients._id",
        createdAt: "$selectedClients.createdAt"
      }
    }
  ]).then(results => results[0]); // 取唯一匹配结果

  res.status(200).json(targetClient);
});

方法3：查询后代码内过滤 #

如果不需要MongoDB端处理，可在JS中直接筛选数组：

exports.removeTrustee = asyncHandler(async (req, res, next) => {
  const user_id = req.params.user_id.split("-")[1];
  const client_id = ObjectId(req.params.client_id.split("-")[1]);

  const user = await User.findOne({ _id: user_id });
  if (!user) {
    return res.status(404).json({ message: "用户不存在" });
  }

  // 在数组中找到目标client（注意要转换为字符串比较ObjectId）
  const targetClient = user.selectedClients.find(
    client => client._id.toString() === client_id.toString()
  );
  res.status(200).json(targetClient);
});

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