我在MongoDB集合中有许多用户记录(> 10000),可以按分数desc + time asc + bonus desc进行排序。如何使用Mongoose根据此排序获取列表中一个用户的排名?假设索引已正确构建。
答案 0 :(得分:13)
计算排序顺序中此用户之前的用户数。我将从简单(非复合排序)的情况开始,因为复合案例中的查询更复杂,即使这个想法完全相同。
> db.test.drop()
> for (var i = 0; i < 10; i++) db.test.insert({ "x" : i })
> db.test.find({ }, { "_id" : 0 }).sort({ "x" : -1 }).limit(5)
{ "x" : 9 }
{ "x" : 8 }
{ "x" : 7 }
{ "x" : 6 }
{ "x" : 5 }
对于此订单,文档{ "x" : i }
的排名是{ "x" : j }
i < j
> var rank = function(id) {
var i = db.test.findOne({ "_id" : id }).x
return db.test.count({ "x" : { "$gt" : i } })
}
> var id = db.test.findOne({ "x" : 5 }).id
> rank(id)
4
排名将基于0.同样,如果您要计算排序{ "x" : i }
中文档{ "x" : 1 }
的排名,则可以计算文档{ "x" : j }
的数量i > j
。
对于复合排序,相同的过程有效,但实现起来比较棘手,因为复合索引中的顺序是词典,即排序{ "a" : 1, "b" : 1}
,(a, b) < (c, d)
如果{{1} }或a < c
和a = c
,因此我们需要一个更复杂的查询来表达这种情况。以下是复合索引的示例:
b < d
要查找文档> db.test.drop()
> for (var i = 0; i < 3; i++) {
for (var j = 0; j < 3; j++) {
db.test.insert({ "x" : i, "y" : j })
}
}
> db.test.find({}, { "_id" : 0 }).sort({ "x" : 1, "y" : -1 })
{ "x" : 0, "y" : 2 }
{ "x" : 0, "y" : 1 }
{ "x" : 0, "y" : 0 }
{ "x" : 1, "y" : 2 }
{ "x" : 1, "y" : 1 }
{ "x" : 1, "y" : 0 }
{ "x" : 2, "y" : 2 }
{ "x" : 2, "y" : 1 }
{ "x" : 2, "y" : 0 }
的排名,您需要在{ "x" : i, "y" : j }
的订单{ "x" : a, "y" : b }
中找到{ "x" : 1, "y" : -1 }
的文档数量(i, j) < (a, b)
。根据排序规范,这相当于条件i < a
或i = a
和j > b
:
> var rank = function(id) {
var doc = db.test.findOne(id)
var i = doc.x
var j = doc.y
return db.test.count({
"$or" : [
{ "x" : { "$lt" : i } },
{ "x" : i, "y" : { "$gt" : j } }
]
})
}
> id = db.test.findOne({ "x" : 1, "y" : 1 })._id
> rank(id)
4
最后,在你的三部分复合指数的情况下
{ "score" : -1, "time" : 1, "bonus" : -1 }
rank
函数将是
> var rank = function(id) {
var doc = db.test.findOne(id)
var score = doc.score
var time = doc.time
var bonus = doc.bonus
return db.test.count({
"$or" : [
{ "score" : { "$gt" : score } },
{ "score" : score, "time" : { "$lt" : time } },
{ "score" : score, "time" : time, "bonus" : { "$gt" : bonus } }
]
})
}