我有一个包含这样文件的集合:
[
{
"user_id": 1,
"prefs": [
"item1",
"item2",
"item3",
"item4"
]
},
{
"user_id": 2,
"prefs": [
"item2",
"item5",
"item3"
]
},
{
"user_id": 3,
"prefs": [
"item4",
"item3",
"item7"
]
}
]
我想要的是编写一个聚合,它将获得user_id
和生产者列表,其中包含映射到其列表中相同prefs
个数的所有用户。例如,如果我为user_id = 1
运行聚合,我必须得到:
[
{
"user_id": 2,
"same": 1
},
{
"user_id": 3,
"same": 2
}
]
答案 0 :(得分:4)
您不能在此处使用"user_id": 1
这样简单的输入编写任何查询,但您可以检索该用户的文档,然后将该数据与您要检索的其他文档进行比较:
var doc = db.collection.findOne({ "user_id": 1 });
db.collection.aggregate([
{ "$match": { "user_id": { "$ne": 1 } } },
{ "$project": {
"_id": 0,
"user_id": 1
"same": { "$size": { "$setIntersection": [ "$prefs", doc.prefs ] } }
}}
])
这是一种方法,但与比较客户端中的每个文档也没有太大区别:
function intersect(a,b) {
var t;
if (b.length > a.length) t = b, b = a, a = t;
return a.filter(function(e) {
if (b.indexOf(e) != -1) return true;
});
}
var doc = db.collection.findOne({ "user_id": 1 });
db.collection.find({ "user_id": { "$ne": 1 } }).forEach(function(mydoc) {
printjson({
"user_id": mydoc.user_id,
"same": intersect(mydoc.prefs, doc.prefs).length
});
});
它是一回事。你不是真的"聚合"这里的任何内容,只是将一个文档内容与另一个文档内容进行比较。当然,您可以要求聚合框架执行类似" filter"任何没有相似匹配的东西:
var doc = db.collection.findOne({ "user_id": 1 });
db.collection.aggregate([
{ "$match": { "user_id": { "$ne": 1 } } },
{ "$project": {
"_id": 0,
"user_id": 1
"same": { "$size": { "$setIntersection": [ "$prefs", doc.prefs ] } }
}},
{ "$match": { "same": { "$gt": 0 } }}
])
实际上,在进行投影之前删除任何零数量的文档会更有效:
var doc = db.collection.findOne({ "user_id": 1 });
db.collection.aggregate([
{ "$match": { "user_id": { "$ne": 1 } } },
{ "$redact": {
"$cond": {
"if": { "$gt": [
{ "$size": { "$setIntersection": [ "$prefs", doc.prefs ] } },
0
]},
"then": "$$KEEP",
"else": "$$PRUNE"
}
}},
{ "$project": {
"_id": 0,
"user_id": 1
"same": { "$size": { "$setIntersection": [ "$prefs", doc.prefs ] } }
}}
])
至少那样做服务器处理会有一些意义。
但除此之外,它几乎都是一样的,可能是一个"小"在客户端制定“交叉路口”的更多开销"这里。