从多个文档中获取一个数组字段并将元素合并到一个对象中

时间:2019-10-13 14:06:47

标签: javascript mongodb aggregation-framework

例如,我有一个users集合。每个用户文档都有posts数组,其中包含他们的帖子。我想这样查询这个集合:

从用户名数组Nstart中的每个用户获取2个从索引Nend开始到['user1', 'user2']结束的帖子,并返回一个合并的帖子数组,按post.postedOn int的升序排列

如果这是集合:

{
  username: "user1",
  posts: [
    { id: 1, text: "123" ... },
    { id: 2, text: "456" ... },
    { id: 3, text: "789" ... },
    ...
  ]
},
{
  username: "user2",
  posts: [
    { id: 1, text: "abc" ... },
    { id: 2, text: "def" ... },
    { id: 3, text: "ghi" ... },
    ...
  ]
}
...

我想要这个:

{
  posts: {
    { id: 1, text: "123" ... },
    { id: 2, text: "456" ... },
    { id: 1, text: "abc" ... },
    { id: 2, text: "def" ... },
  }
}

我尝试多次使用aggregate。这是我的查询之一:

db.getCollection("users").aggregate([{ $match: { username: { $in: ["admin", "admin2"] } } }, { $project: { posts: 1 } }])

我得到两个用户对象,每个用户对象包含他们的所有帖子,而不是一个我以后可以操作的帖子数组。

我想要达到的最终结果是:

输入

  • 用户:['user1', 'user2']
  • 开始帖子索引:4
  • 结尾帖子索引:8

结果

所选用户发布的帖子。总数为8(每个用户的ID为4-8)。

1 个答案:

答案 0 :(得分:0)

这需要使用聚合运算符$slice$push分别选择所需的post元素并将其添加到数组中(按索引)。

例如posts集合有三个文档(我在这里省略了_id字段):

{
       "username" : "user2",
       "posts" : [
               {
                       "id" : 1,
                       "text" : "123"
               },
               {
                       "id" : 2,
                       "text" : "456"
               },
               {
                       "id" : 3,
                       "text" : "789"
               },
               {
                       "id" : 4,
                       "text" : "000"
               }
       ]
}
{
       "username" : "user1",
       "posts" : [
               {
                       "id" : 1,
                       "text" : "abc"
               },
               {
                       "id" : 2,
                       "text" : "def"
               },
               {
                       "id" : 3,
                       "text" : "ghi"
               },
               {
                       "id" : 4,
                       "text" : "zzz"
               }
       ]
}
{
        "username" : "user99",
        "posts" : [ ]
}

在mongo shell中:

> var users = [ "user1", "user2" ]
> var startIx = 1, noOfElements = 2

> db.posts.aggregate([ 
  { $project: { _id: 0, username: 1, posts: 1, usersMatch: { $in: [ "$username",  users ] } } },
  { $match: { usersMatch: true } },
  { $addFields: { posts: { $slice: [ "$posts", startIx, noOfElements ] } } },
  { $group: { _id: null, postsOfSelectedUsers: { $push: "$posts"} } },
  { $project: { _id: 0, selectedUsers: users, postsOfSelectedUsers: 1 } } 
])

输出:

{
        "postsOfSelectedUsers" : [
                [
                        {
                                "id" : 2,
                                "text" : "456"
                        },
                        {
                                "id" : 3,
                                "text" : "789"
                        }
                ],
                [
                        {
                                "id" : 2,
                                "text" : "def"
                        },
                        {
                                "id" : 3,
                                "text" : "ghi"
                        }
                ]
        ],
        "selectedUsers" : [
                "user1",
                "user2"
        ]
}

注意:这是一个解决方案,但是可以完善或重新设计(我认为)。