如何编写MongoDB聚合查询获取社交媒体用户及好友帖子并解决重复问题

问题原因 #

原有查询出现重复的核心原因是多次$unwind操作产生了笛卡尔积：比如用户有N个好友、M个个人帖子，就会生成N*M条中间文档，拼接数组后自然出现大量重复数据，哪怕用$addToSet也可能因为嵌套字段差异导致去重不全，且额外增加了很多无效计算，性能极差。

修正后的聚合查询 #

优化思路是先一次性拼接出所有需要拉取帖子的用户ID（当前用户+所有好友），仅用一次关联查询直接拿到所有符合要求的帖子，完全避免中间冗余数据：

db
.getDb()
.collection(collections.USERS)
.aggregate([
  // 匹配当前用户
  {
    $match: {
      _id: ObjectId(currentUserId),
    }
  },
  // 构造目标用户ID列表：当前用户ID + 所有好友ID
  {
    $project: {
      targetUserIds: {
        $concatArrays: [ ["$_id"], "$friends" ]
      },
      _id: 0
    }
  },
  // 一次关联查询出所有目标用户发布的帖子
  {
    $lookup: {
      from: "posts",
      localField: "targetUserIds",
      foreignField: "userId",
      as: "allPosts"
    }
  },
  // 拆分数组后按发布时间倒序
  {
    $unwind: "$allPosts"
  },
  {
    $sort: {
      "allPosts.createdAt": -1
    }
  },
  // 重新组装为有序的帖子数组
  {
    $group: {
      _id: null,
      allPosts: {
        $push: "$allPosts"
      }
    }
  },
  {
    $project: {
      _id: 0,
      allPosts: 1
    }
  }
])
.toArray();

数据库结构优化建议 #

• 取消user文档中的posts冗余数组：post文档已经通过userId字段关联了发布者，额外在user里存posts数组会增加数据不一致风险，且用户发帖量上升后会导致user文档体积持续膨胀，影响查询性能。
• 为posts集合添加联合索引：新增userId + createdAt的联合索引，上述关联查询和排序操作都会直接命中索引，不需要额外计算，性能提升非常明显。
• 大用户量下拆分好友关系：如果单用户好友量上限超过500，建议将好友关系单独抽为friend集合，存储双向关联数据，避免user文档因为好友列表过大导致体积超标，也方便后续做好友分页、关系状态判断等扩展功能。
• 冗余字段更新策略优化：post文档中存储authorName、avatar的设计是合理的，可以避免查询帖子时二次关联user表，建议搭配异步任务，用户修改昵称/头像时批量更新其发布的所有帖子对应的冗余字段，平衡一致性和查询性能。
• 首页Feed缓存优化：用户量达到万级以上时，不要每次刷首页都实时聚合查询，可以用Redis为每个用户缓存最近1000条Feed的帖子ID，用户刷新时直接拿ID批量查帖子详情，大幅降低数据库压力。

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