MongoDB：分组，然后计算不同的值

Question

我有一个书单。我想得到作者写的类型的数量，我也想添加这些类型的结果。我的数据库如下所示：

{"_id": ObjectID("1), "title": "Harry Potter", "year": NumberInt(2000), "author": "JK. Rowling", 
"genres": "Fantasy"}, 
"_id": ObjectID("2"), "title": "Harry Potter 99", "year": NumberInt(2020), "author": "JK. Rowling", 
"genres": "Drama"}, "_id": ObjectID("2"), "title": "Harry Potter", "year": NumberInt(2000), "author": "JK. Rowling", 
"genres": "Drama"}, {...}

所以，到目前为止，我的代码如下所示：

phase1 = {$group : {"_id" : "$author"}, "countgenres" : {$sum : 1}}
phase2 = {$addFields : "genres"}}
phase3 = {$sort : {"numgenres" : -1}}

steps = [phase1, phase2, phase3]
db.database.aggregate(steps)

这对我不起作用，所以我希望有人可以帮助我编写正确的代码来执行此操作。结果应如下所示：

{

"_id" : "JK. Rowling",

"countgenres" : 4,

"genres" : [

"War",
"Fantasy",
"Drama",
"Crime"

]
}

谢谢。

Answer 1

你不能直接在 $group 阶段这样做。取而代之的是，您必须使用 $addFields 阶段并使用 $reduce 和 $setUnion 来连接没有双精度的数组。 然后，您可以添加一个具有新数组大小的字段，然后执行 $sort。

db.collection.aggregate([
  {
    $group: {
      _id: "$author",
      genres: {
        $push: "$genres"
      }
    }
  },
  {
    $addFields: {
      genres: {
        "$reduce": {
          "input": "$genres",
          "initialValue": [],
          "in": {
            $setUnion: [
              "$$value",
              "$$this"
            ]
          }
        }
      }
    }
  },
  {
    $addFields: {
      countGenres: {
        $size: "$genres"
      }
    }
  }
])

You can test here

Answer 2

试试这个查询：

db.collection.aggregate([
  {
    "$match": {
      "author": "JK. Rowling"
    }
  },
  {
    "$group": {
      "_id": "$author",
      "genres": {
        "$addToSet": "$genres"
      }
    }
  },
  {
    $addFields: {
      genres: {
        "$reduce": {
          "input": "$genres",
          "initialValue": [],
          "in": {
            $setUnion: [
              "$$value",
              "$$this"
            ]
          }
        }
      }
    }
  },
  {
    "$project": {
      "countgenres": {
        "$size": "$genres"
      },
      "genres": 1
    }
  }
])

第一阶段是作者$match。

然后分组，我使用$addToSet 来避免重复值。
之后，与$addFields 值合并。

分组后，字段genres输出如下：

"genres": [
      [
        "War",
        "Fantasy",
        "Drama",
        "Crime"
      ],
      [
        "War",
        "Fantasy",
        "Drama"
      ]
    ]

合并是必要的$setUnion。

最后一步是计算数组大小并输出你想要的值。

例如here