Mongo搜索结果与facet计数

时间:2017-04-11 12:31:46

标签: mongodb aggregation-framework faceted-search mgo

我是Mongo的新手,我希望使用mgo驱动程序在Go中实现分面搜索。我需要获得与我的查询匹配的文档以及facet计数。

我当前的实现是执行查询以获取文档,然后使用相同参数执行另一个查询以获取构面计数,但这似乎效率非常低。有一个好方法可以一步到位吗?

例如,如果我有一本书集:

[{
  title: "Book One",
  author: "Author A",
  numPages: 20,
  type: "book"
},
{
  title: "Book Two",
  author: "Author B",
  numPages: 40,
  type: "book"
},
...
...
...
{
  title: "Magazine AA",
  author: "Author A",
  numPages: 10,
  type: "magazine"
}]

首先,我得到符合我查询的文件:

err = books.Find(bson.M{"$and": matches}).All(&results)

然后我使用聚合管道和$ facet重复查询以获得方面计数:

err = Pipe([]bson.M{
    {"$match": bson.M{"$and": matches}},
    {"$facet": bson.M{
        "type":     []bson.M{bson.M{"$sortByCount": "$type"}},
        "author":   []bson.M{bson.M{"$sortByCount": "$author"}, 
    }},
}).All(&facets)

我还看到了$ out,这会让我把我的结果写到临时收藏中,然后我可以用它来确定方面的数量,但我不知道是否还有高效。

1 个答案:

答案 0 :(得分:2)

  

有一个好方法可以一步完成吗?

是的,您可以使用构面搜索来添加未分面的结果。例如:

pipeline := []bson.M{ 
                {"$match": bson.M{"type": bson.M{"$in": []string{"book", "magazine"}}}},
                {"$facet": bson.M{"type": []bson.M{{"$sortByCount":"$type"}}, 
                                  "author": []bson.M{{"$sortByCount":"$author"}},
                                  "unfaceted": []bson.M{{"$match": bson.M{} }},
                                 },
                           },
            }
err = collection.Pipe(pipeline).All(&resp)

以上unfaceted $match没有条件(空),因为第一个$match阶段已经过滤到所需的文档子集。

下面给出了结果:

{
  "type": [
    {
      "_id": "book",
      "count": 2
    },
    {
      "_id": "magazine",
      "count": 1
    }
  ],
  "author": [
    {
      "_id": "Author A",
      "count": 2
    },
    {
      "_id": "Author B",
      "count": 1
    }
  ],
  "unfaceted": [
    {
      "_id": ObjectId(".."),
      "title": "Magazine AA",
      "author": "Author A",
      "numPages": 10,
      "type": "magazine"
    },
    {
      "_id": ObjectId(".."),
      "title": "Book Two",
      "author": "Author B",
      "numPages": 40,
      "type": "book"
    },
    {
      "_id": ObjectId(".."),
      "title": "Book One",
      "author": "Author A",
      "numPages": 20,
      "type": "book"
    }
  ]
}

现在,您可以遍历unfaceted部分,而不是发送单独的查询。有关运算符的更多示例和说明,另请参阅$facet