与序列匹配的数组中的文档

Question

考虑以下 3 个文档（简化）：

{
    "name" : "Alex Ham",
    "opponents" : [ 
        {
            "location" : "west",
            "position" : 5
        },
        {
            "location" : "north",
            "position" : 7
        },
        {
            "location" : "east",
            "position" : 2
        }
    ]
},
{
    "name" : "John Flex",
    "opponents" : [ 
        {
            "location" : "north",
            "position" : 9
        },
        {
            "location" : "north",
            "position" : 4
        },
        {
            "location" : "south",
            "position" : 2
        }
    ]
},
{
    "name" : "Adam Will",
    "opponents" : [ 
        {
            "location" : "east",
            "position" : 6
        },
        {
            "location" : "south",
            "position" : 8
        }
    ]
}

我需要按数组opponents 中的项目序列匹配文档。我将有整数，例如 572、942、68 等。整数的每个数字代表opponents 数组中的position。

例如，对于 572，我需要匹配第一个文档 (Alex Ham)，因为如果您查看 opponents 数组，项目的序列及其 position 字段分别为 5、7 和 2。

对于 942，我必须匹配第二个文档 (John Flex)，对于 68，我必须匹配第三个文档 (Adam Will)，依此类推。

考虑到拥有大量数据（几百万个文档），并考虑到性能，我如何构建一个聚合管道来匹配上述案例的文档。

Answer 1

我认为在这种情况下您不需要使用聚合。要通过聚合完成此任务，您需要遍历整个集合并将所有文档投影到一个新表单，其中包含一个可以与您的输入匹配的值。虽然聚合速度很快，但还是不够好。

我建议将您的输入格式化为可以与文档匹配的表单：

const val = 572;

const arr = val.toString().split('');
const size = arr.length;

const selector = arr.map((v, i) => {
  return {
    [`opponents.${i}.position`]: parseInt(v, 10),
  };
}).reduce((acc, cur) => {
  return Object.assign(acc, cur);
}, {});

selector.opponents = {
  $size: size,
};
console.log(selector);

现在使用这个新的选择器，您可以使用普通的.find 操作来获取文档：

collection.find(arr);

Answer 2

请通过下面解决问题的管道。基本逻辑是$unwind在保留索引的同时，使用索引计算整数的位值，使用公式position * (pow(10, size - index - 1))

[
{$project: {name: 1, opponents: 1, sz: {$size: '$opponents'}}},
{$unwind: {path: '$opponents', includeArrayIndex: 'i'}},
{$project: {name: 1, opponents: 1, 
    num: {$multiply: [{$pow: [10, {$subtract: [{$subtract: ['$sz', 1]}, '$i']}]}, '$opponents.position']}
    }},
{$group: {
    _id: '$_id', 
    num: {$sum: '$num'}, 
    name: {$first: '$name'}, 
    opponents: {$push: '$opponents'}
}},
{$match: {num: 572}},
]