如何在 MongoDB $projection 管道中的 array.indexOf 中使用变量名

Question

我有一个类似的收藏

   {
        "LifeTime" : 123.000,
        "Host" : "xxx.com",
    }, 
    {
        "LifeTime" : 213.000,
        "Host" : "yyy.com",
    }, 
    {
        "LifeTime" : 321.000,
        "Host" : "zzz.com",

    } 

和一个预定义的数组，比如

["xxx.com","yyy.com"]

现在我想使用 $let 和 $cond 进行投影，这样如果数组中存在主机，则 LifeTime 将为 0，否则与 LifeTime 相同。

我试过了

LifeTime: {
    $let: {
    vars: {
        hostname: '$Host'
    },
    in: {
        "$cond": [
            {
                "$and": [
                    {
                        "$gte": [
                            ArrayOfPredefinedHost.indexOf('$$hostname'),
                            0
                        ]
                    }
                ]
            },
            0,
            "$LifeTime"
        ]
    }
    }
}

但它没有返回预期的结果。我认为indexOf 试图找到$$hostname 而不是xxx.com,yyy.com & zzz.com。

我该如何做这种复杂的查询？

Answer 1

以下聚合管道将首先对文档进行分组，以允许使用 $addToSet 累加器运算符包含单个主机元素的附加数组字段。然后，这个附加字段将在 $cond 表达式中的下一个 $project 管道阶段使用，利用 $setIsSubset 运算符：

var ArrayOfPredefinedHost = ["xxx.com", "yyy.com"];
db.collection.aggregate([
    {
        "$group": {
            "_id": {
                "id": "$_id",
                "LifeTime": "$LifeTime",
                "Host": "$Host"
            },
            "HostSet": { "$addToSet": "$Host" }
        }
    },
    {
        "$project": {
            "_id": 0,
            "LifeTime": {
                "$cond": [ { "$setIsSubset": [ "$HostSet", ArrayOfPredefinedHost ] }, "$_id.LifeTime", 0]
            },
            "Host": "$_id.Host"
        }
    }
]);

输出：

/* 0 */
{
    "result" : [ 
        {
            "LifeTime" : 0,
            "Host" : "zzz.com"
        }, 
        {
            "LifeTime" : 213,
            "Host" : "yyy.com"
        }, 
        {
            "LifeTime" : 123,
            "Host" : "xxx.com"
        }
    ],
    "ok" : 1
}