如果项不存在,则更新数组中的条目或添加到数组

时间:2013-11-03 18:58:43

标签: mongodb

我的集合的结构如下所示

{ "_id" : "MHBk8q96vpuRYrAdn", 
    "circles" : { 
        "guests" : 3, 
        "properties" : [      
            {       
                "position" : {  "x" : 146, "y" : 70.5207970},  
                "name" : "circle-1" 
            },
            {       
                "position" : {  "x" : 200, "y" : 85},  
                "name" : "circle-2" 
            }  
        ], 
        "tables" : 1 
    } 
}

我需要能够更新circles.properties.position的位置(如果它按名称存在),或者如果不存在则添加新条目。例如,更新“circle-1”的位置,因为它存在,但为“circle-3”添加一个带有名称和位置的新数组项。是否有可能实现这一目标?到目前为止,我只能使用$ push推进数组,并且我已经弄乱了$(查询)运算符但没有成功。感谢。

1 个答案:

答案 0 :(得分:2)

MongoDB doesn't support upserts to arrays起,它可能会很棘手。您可以尝试以下内容:

var query = {};
new_circle = { "position" : {  "x" : -1, "y" : -1}, "name" : "circle-1" };

db.foo.find(query).forEach(
    function(doc) {

        // Find index of 'circle-1'
        i = doc.circles.properties.map(
            function(el) { if (el.name == 'circle-1') return 1; else return -1;}
        ).indexOf(1);

        // Update if circle-1 in circles-properties 
        if (i != -1) {
            doc.circles.properties[i] = new_circle;
        }

        // If not push new
        else {
            doc.circles.properties.push(new_circle);
        }

        db.foo.save(doc);
    }
)

修改

如果您无法使用saveupdate upsert选项替换上面发布的if-else块,请执行以下操作:

if (i != -1) {
    db.foo.update(
        {"_id" : doc._id, "circles.properties.name": "circle-1"},
        {$set: {"circles.properties.$.position": new_circle.position}}
}

else {
    db.foo.update(
        {"_id" : doc._id},
        {$push: {"properties": new_circle }}
    )
}