Question

大家好日子！

我试图找到解决方案3天，但无法找到任何解决方案。希望你们能帮助我。

数据

{       
 'item': 'pen-1', 
 'colors': [ 'yellow', 'green', 'blue', 'red', 'pink', 'purple' ] //total: 6
  },
  {
 'item': 'pen-2', 
 'colors': [ 'green', 'blue','pink', 'purple' ] //total: 4
 }

这是我到目前为止所做的查询：

var col = ['yellow', 'red', 'blue'];
db.getCollection('data').aggregate([
            { $match: { colors : { $in : col } } },
            { $group: { _id: { 'item': '$item',  colors: { $size : '$colors' } } } }
          ])

和结果

{
    "_id" : {
        "item" : "pen-1",
        "colors" : 6
    }
}
{
    "_id" : {
        "item" : "pen-2",
        "colors" : 4
    }
}

我想做什么：

退回物品，返回总数（匹配的颜色/颜色）所以它会是这样的：

"_id" : {
        "item" : "pen-1",
        "rank": 0.5 // total colors(6) / that match(3)
    }

 "_id" : {
        "item" : "pen-2",
        "rank": 0.25 // total colors(4) / that match(1)
    }

Answer 1

您可以使用 $setIntersection 运算符，它接受两个或多个数组并返回一个数组，其中包含每个输入数组中出现的元素。然后，您可以使用 $size 返回结果数组中的元素数，以获得匹配的总颜色。

如果使用MongoDB 3.4及更新版本，那么 $addFields 将允许您在不明确投影其余字段的情况下向文档添加额外字段，否则 {{3} 就足够了。

以下示例显示了这一点：

var col = ['yellow', 'red', 'blue'];
db.getCollection('data').aggregate([
    { "$match": { "colors": { "$in": col } } },
    {
        "$addFields": {
            "rank": {
                "$divide": [                        
                    { "$size": { "$setIntersection": ["$colors", col] } },
                    { "$size": "$colors" }
                ]
            }
        }
    },
    { "$sort": { "rank": 1 } }            
])

示例输出

/* 1 */
{
    "_id" : ObjectId("58c645eaf3d1c82da16279a6"),
    "item" : "pen-2",
    "colors" : [ 
        "green", 
        "blue", 
        "pink", 
        "purple"
    ],
    "rank" : 0.25
}

/* 2 */
{
    "_id" : ObjectId("58c645eaf3d1c82da16279a5"),
    "item" : "pen-1",
    "colors" : [ 
        "yellow", 
        "green", 
        "blue", 
        "red", 
        "pink", 
        "purple"
    ],
    "rank" : 0.5
}

Answer 2

以下是从v.3.4开始在MongoDB中执行此操作的方法：

var colors = ['red', 'pink'];

db.getCollection('test').aggregate([
        //Filtering out the records where colors array is empty
        //To avoiding dividing by zero.
	{
		$match: {
			'colors.0': {$exists: true}
		}
	},
	{
		$addFields: {
			matchingColors: {
				$filter: {
					input: '$colors',
					as: 'color',
					cond: {'$in': ['$$color', colors]}
				}
			}
		}
	},
	{
		$project: {
			item: '$item',
			score: {$divide: [{$size: '$matchingColors'}, {$size: '$colors'}]}
		}
	}
]);

具有排名/搜索计数的Mongodb

2 个答案: