MongoDB-不同，限制和排序以获得更好的结果

Question

我正在尝试开发一个查询，以帮助在MongoDB中的搜索请求中混合结果。 我的收藏的一个示例（非常简化的版本）如下所示。 每个文档都有要查询的位置，清单质量的等级以及插入清单的提供者的名称。

[
  {
    "location": "paris",
    "ranking": "998",
    "provider": "Alpha"
  },
  {
    "location": "paris",
    "ranking": "965",
    "provider": "Alpha"
  },
  {
    "location": "paris",
    "ranking": "945",
    "provider": "Alpha"
  },
  {
    "location": "paris",
    "ranking": "933",
    "provider": "Alpha"
  },
  {
    "location": "paris",
    "ranking": "953",
    "provider": "Alpha"
  },
  {
    "location": "paris",
    "ranking": "983",
    "provider": "Alpha"
  },
  {
    "location": "paris",
    "ranking": "700",
    "provider": "Beta"
  },
  {
    "location": "paris",
    "ranking": "745",
    "provider": "Beta"
  },
  {
    "location": "paris",
    "ranking": "670",
    "provider": "Omega"
  },
  {
    "location": "paris",
    "ranking": "885",
    "provider": "Omega"
  },
  {
    "location": "paris",
    "ranking": "500",
    "provider": "Omega"
  },
  {
    "location": "london",
    "ranking": "600",
    "provider": "Omega"
  },
  {
    "location": "london",
    "ranking": "650",
    "provider": "Beta"
  }
]

如您所见，提供商Alpha具有最多的列表和最佳的排名。 因此，当我搜索巴黎并按排名进行排序时，来自Alpha提供程序的所有列表均排在最前面，而Beta和Omega的排名则滑到了底部。

我想做的就是将每个提供者限制为3个。因此，即使Alphas仍然排名靠前，它们也将被限制为3个，以使Beta和Omega更高。 然后，当使用.skip时，可以在“第2页”上看到其余的Alpha。

如果我要在Python中执行此操作，则一个同步示例将如下所示。

#!/usr/bin/env python
# -*- coding: utf-8 -*-

results = []

providersAvailable = colc.find({'location': 'paris'}).distinct('provider')
for provider in providersAvailable:
    search = colc.find({'provider':provider, 'location': 'paris'}).limit(3)
    results = results + list(search)

return sorted(results, key=lambda k: k['ranking']) 

这是沉重的，耗时的，并且总体来说很糟，尤其是收集了250万份文档。 我怎么能在蒙哥斯方面做到这一点？ 谢谢！

Answer 1

您可以尝试一些服务器端JS，例如。

var providers = db.runCommand({distinct:"colc", key:"provider"}).values
for(p in providers){
   var c = db.colc.find({"provider":providers[p]}).sort({"ranking":-1}).limit(3);
   c.forEach(printjson);
}

但是由于所有JS都被解释了，所以它不是最快的选择。

您可以使用聚合框架，该框架主要是服务器方面的问题。

db.colc.aggregate([ 
    {$match: {"location":"paris"}}, 
    {$group:{_id: { "provider": "$provider", "location":"$location"}, 
             "rankings" : { $addToSet: "$ranking"} } } 
]);

但是您将需要一些客户端代码来从返回数组中选择每个提供程序的排名。

{
    "result" : [
        {
            "_id" : {
                "provider" : "Omega",
                "location" : "paris"
            },
            "rankings" : [
                "500",
                "885",
                "670"
            ]
        },
        {
            "_id" : {
                "provider" : "Beta",
                "location" : "paris"
            },
            "rankings" : [
                "745",
                "700"
            ]
        },
        {
            "_id" : {
                "provider" : "Alpha",
                "location" : "paris"
            },
            "rankings" : [
                "983",
                "953",
                "933",
                "945",
                "965",
                "998"
            ]
        }
    ],
    "ok" : 1
}