Обновить поле MongoDB, используя значение другого поля - PullRequest
303 голосов
/ 20 октября 2010

В MongoDB возможно ли обновить значение поля, используя значение из другого поля?Эквивалентный SQL будет выглядеть примерно так:

UPDATE Person SET Name = FirstName + ' ' + LastName

И псевдокод MongoDB будет:

db.person.update( {}, { $set : { name : firstName + ' ' + lastName } );

Ответы [ 8 ]

232 голосов
/ 20 января 2013

Вы должны пройти через.Для вашего конкретного случая:

db.person.find().snapshot().forEach(
    function (elem) {
        db.person.update(
            {
                _id: elem._id
            },
            {
                $set: {
                    name: elem.firstname + ' ' + elem.lastname
                }
            }
        );
    }
);
155 голосов
/ 17 мая 2016

Лучший способ сделать это в версии 4.2+, которая позволяет использовать конвейер агрегации в документе обновления и updateOne, updateMany или update метод сбора. Обратите внимание, что последний не рекомендуется использовать в большинстве, если не во всех драйверах языков.

MongoDB 4.2 +

Версия 4.2 также представила $set оператор стадии конвейера, который является псевдонимом $addFields. Я буду использовать $set здесь, как это карты с тем, что мы пытаемся достичь.

db.collection.<update method>(
    {},
    [
        {"$set": {"name": { "$concat": ["$firstName", " ", "$lastName"]}}}
    ]
)

MongoDB 3.4 +

В версии 3.4+ вы можете использовать $addFields и $out операторов конвейера агрегации.

db.collection.aggregate(
    [
        { "$addFields": { 
            "name": { "$concat": [ "$firstName", " ", "$lastName" ] } 
        }},
        { "$out": "collection" }
    ]
)

Обратите внимание, что не обновляет вашу коллекцию, а заменяет существующую коллекцию или создает новую. Также для операций обновления, требующих «приведение типов», вам потребуется обработка на стороне клиента, и, в зависимости от операции, вам может потребоваться использовать метод find() вместо .aggreate().

MongoDB 3.2 и 3.0

То, как мы это делаем, заключается в $project наших документах и ​​использовании оператора агрегирования строк $concat для возврата объединенной строки. Оттуда вы затем итерируете курсор и используете оператор обновления $set, чтобы добавить новое поле в ваши документы, используя массовые операции для максимальной эффективности.

Запрос агрегации:

var cursor = db.collection.aggregate([ 
    { "$project":  { 
        "name": { "$concat": [ "$firstName", " ", "$lastName" ] } 
    }}
])

MongoDB 3.2 или новее

из этого вам нужно использовать метод bulkWrite.

var requests = [];
cursor.forEach(document => { 
    requests.push( { 
        'updateOne': {
            'filter': { '_id': document._id },
            'update': { '$set': { 'name': document.name } }
        }
    });
    if (requests.length === 500) {
        //Execute per 500 operations and re-init
        db.collection.bulkWrite(requests);
        requests = [];
    }
});

if(requests.length > 0) {
     db.collection.bulkWrite(requests);
}

MongoDB 2.6 и 3.0

В этой версии вам нужно использовать устаревший Bulk API и связанные с ним методы .

var bulk = db.collection.initializeUnorderedBulkOp();
var count = 0;

cursor.snapshot().forEach(function(document) { 
    bulk.find({ '_id': document._id }).updateOne( {
        '$set': { 'name': document.name }
    });
    count++;
    if(count%500 === 0) {
        // Excecute per 500 operations and re-init
        bulk.execute();
        bulk = db.collection.initializeUnorderedBulkOp();
    }
})

// clean up queues
if(count > 0) {
    bulk.execute();
}

MongoDB 2.4

cursor["result"].forEach(function(document) {
    db.collection.update(
        { "_id": document._id }, 
        { "$set": { "name": document.name } }
    );
})
103 голосов
/ 20 октября 2010

Очевидно, что существует способ сделать это эффективно, начиная с MongoDB 3.4, см. ответ стивейна .


Устаревший ответ ниже

Вы не можете ссылаться на сам документ в обновлении (пока).Вам нужно будет перебирать документы и обновлять каждый документ с помощью функции.См. этот ответ для примера или этот для серверной стороны eval().

41 голосов
/ 11 февраля 2015

Для базы данных с высокой активностью вы можете столкнуться с проблемами, когда ваши обновления влияют на активно меняющиеся записи, и по этой причине я рекомендую использовать snapshot ()

db.person.find().snapshot().forEach( function (hombre) {
    hombre.name = hombre.firstName + ' ' + hombre.lastName; 
    db.person.save(hombre); 
});

http://docs.mongodb.org/manual/reference/method/cursor.snapshot/

9 голосов
/ 03 апреля 2015

Я попробовал описанное выше решение, но оказалось, что оно не подходит для больших объемов данных.Затем я обнаружил функцию потока:

MongoClient.connect("...", function(err, db){
    var c = db.collection('yourCollection');
    var s = c.find({/* your query */}).stream();
    s.on('data', function(doc){
        c.update({_id: doc._id}, {$set: {name : doc.firstName + ' ' + doc.lastName}}, function(err, result) { /* result == true? */} }
    });
    s.on('end', function(){
        // stream can end before all your updates do if you have a lot
    })
})
3 голосов
/ 15 января 2019

Относительно этого ответа , функция моментального снимка устарела в версии 3.6, согласно этому обновлению . Таким образом, в версии 3.6 и выше можно выполнить операцию следующим образом:

db.person.find().forEach(
    function (elem) {
        db.person.update(
            {
                _id: elem._id
            },
            {
                $set: {
                    name: elem.firstname + ' ' + elem.lastname
                }
            }
        );
    }
);
2 голосов
/ 08 июня 2016

Вот что мы придумали для копирования одного поля в другое для ~ 150_000 записей.Это заняло около 6 минут, но все еще значительно менее ресурсоемко, чем это было бы для создания экземпляра и итерации по тому же числу объектов ruby.

0 голосов
/ 11 июня 2019

Начиная с Mongo 4.2, db.collection.update() можно принять конвейер агрегации, наконец, разрешив обновление / создание поля на основе другого поля:

// { firstName: "Hello", lastName: "World" }
db.collection.update(
  {},
  [{ $set: { name: { $concat: [ "$firstName", " ", "$lastName" ] } } }],
  { multi: true }
)
// { "firstName" : "Hello", "lastName" : "World", "name" : "Hello World" }
  • Первая часть {} - это запрос на совпадение, отфильтровывающий, какие документы обновлять (в нашем случае все документы).

  • Вторая часть [{ $set: { name: { ... } }] - это конвейер агрегации обновлений (обратите внимание на квадратные скобки, обозначающие использование конвейера агрегации).$set - это новый оператор агрегирования с псевдонимом $addFields.

  • Не забудьте { multi: true }, иначе будет только первый соответствующий документобновлено.

...