MongoDB 聚合。 $group 还收集非分组字段的数据

Question

我有一个功能，用户可以发布 cmets，其他用户可以发布对 cme​​ts 的 emoji 反应。

db.emotes 的示例数据：

{ "_id" : 'a', "commentId" : 'commentA', "userId" : "userA", "emote" : "emoteA" }
{ "_id" : 'b', "commentId" : 'commentA', "userId" : "userB", "emote" : "emoteA" }
{ "_id" : 'c', "commentId" : 'commentA', "userId" : "userC", "emote" : "emoteB" }

对于commentA，2 位用户回复了emoteA，1 位用户回复了emoteB。

假设userA 登录，所以我获取当前视图的cmets，以及每个评论的表情。

async listEmotesForCommentId (commentId) {
    const emotes = db.get().collection('emotes')
    return emotes.aggregate([
        { $match: { commentId: new ObjectID(commentId) } },
        {
            $group: {
                _id: '$emote',
                count: { $sum: 1 }
            }
        },
        {
            $project: {
                _id: 0,
                emote: '$_id',
                count: 1,
            }
        }
    ]).toArray()
}

//
const emotes = await listEmotesForCommentId('commentA')
/*
[
  { count: 2, emote: 'emoteA' },
  { count: 1, emote: 'emoteB' },
]
*/

现在，问题来了。如果请求用户对该评论做出反应，我想要一个字段hasReacted: <boolean>。所以现在函数签名是listEmotesForCommentId (commentId, userId)，我想要的结果是

const emotes = await listEmotesForCommentId('commentA', 'userA')
/*
[
  { count: 2, emote: 'emoteA', hasReacted: true },
  { count: 1, emote: 'emoteB', hasReacted: false },
]
*/

这个结果意味着userA 是对emoteA 做出反应的用户之一，现在我的 UI 可以显示它。我不知道如何使用 mongo 聚合来做到这一点。我也不知道如何用 SQL 来做，因为一旦你对数据进行分组，你就失去了对非分组字段的访问权限。

Answer 1

你可以试试这个查询：

emotes.aggregate([
  { $match: { commentId: new ObjectID(commentId) } },
  {
    $group: {
      _id: "$emote",
      count: { $sum: 1 },
      users: {
         $push: "$userId" // Push all users to users field
      }
    }
  },
  {
    $project: {
      _id: 0,
      emote: "$_id",
      count: 1,
      /** Conditional check if pass'd in value 'userA' exists in 'users' array before assigning a value to hasReacted field */
      hasReacted: { $cond: [{ $in: ["userA", "$users"] }, true, false] }
    }
  }
]);

测试： MongoDB-Playground