[问题]Nodejs mongoose查询修改的问题

各位用过mongoose有这方面经验或者提示的大神们，给小弟一点提示呀。先看几个案例：

查询对象并修改保存

Model.findOne(function(err,doc){
    doc.xxx = 'xxx'; //修改，假设他有个属性是xxx
    doc.save();        //此时就能直接修改了
});

根据情况同时修改多个对象并保存

Model.find(funciton(err,docs){
    docs.forEach(function(doc){
        switch(条件){
            case 1:doc.xxx = 1;break;
            case 2:doc.xxx = 2;break;
            //.......
            //.......
            //.......
            default:doc.xxx = 0;break;
        }
        doc.save();
    });
});

统一修改多个对象并保存

Model.update(查询条件,{$set:{xxx:'xxx'}},function(err,docs){
})

有没有把第二、三种情况联合在一起的，既能一句话搞定，又能按照多种情况进行更新。

eggper 1楼

当然可以。针对您的需求，我们可以结合第二种和第三种情况，实现既能批量处理又能根据不同条件进行更新的需求。这里提供一个综合示例，展示如何使用 Mongoose 进行批量查询，并根据不同的条件更新文档。

示例代码

const mongoose = require('mongoose');
const Schema = mongoose.Schema;

// 假设我们有一个简单的模型
const UserSchema = new Schema({
    name: String,
    status: { type: String, default: 'active' },
});

const User = mongoose.model('User', UserSchema);

// 连接 MongoDB 数据库
mongoose.connect('mongodb://localhost/test', { useNewUrlParser: true, useUnifiedTopology: true });

async function updateUser() {
    try {
        const users = await User.find({});  // 查询所有用户

        for (const user of users) {
            let updateData = {};

            switch (user.status) {
                case 'active':
                    updateData = { status: 'inactive' };
                    break;
                case 'inactive':
                    updateData = { status: 'archived' };
                    break;
                case 'archived':
                    updateData = { status: 'deleted' };
                    break;
                default:
                    updateData = { status: 'unknown' };
            }

            await User.findByIdAndUpdate(user._id, { $set: updateData });  // 更新每个用户
        }

        console.log('All users updated successfully.');
    } catch (err) {
        console.error('Error updating users:', err);
    }
}

updateUser();

解释

模型定义：我们首先定义了一个简单的 User 模型，包含 name 和 status 字段。
数据库连接：使用 mongoose.connect 方法连接到 MongoDB 数据库。
查询所有用户：使用 User.find({}) 查询所有的用户记录。
循环处理每个用户：遍历查询结果中的每一个用户，根据其当前的 status 字段值决定如何更新。
更新用户：使用 User.findByIdAndUpdate 方法来更新用户的 status 字段，并设置为新的状态值。
错误处理：通过 try...catch 块捕获并处理可能发生的错误。

这种方法不仅实现了批量查询，还能根据不同的条件灵活地更新数据，满足了您的需求。

yuanlaile 2楼

你可以使用 async/await 和 Promise.all 来结合第二种和第三种情况，这样可以一次性处理多个文档，并且根据不同条件进行更新。以下是一个示例代码：

const mongoose = require('mongoose');
const Model = mongoose.model('YourModel');

async function updateDocuments() {
    try {
        const docs = await Model.find();  // 查询所有文档

        const updatePromises = docs.map(async (doc) => {
            let updatedDoc = doc;
            switch (true) {  // 使用 switch-case 来决定如何更新
                case (条件1): 
                    updatedDoc.xxx = 'value1';
                    break;
                case (条件2): 
                    updatedDoc.xxx = 'value2';
                    break;
                // 其他条件...
                default: 
                    updatedDoc.xxx = 'defaultValue';
                    break;
            }
            return await updatedDoc.save();  // 保存更新后的文档
        });

        await Promise.all(updatePromises);  // 等待所有更新操作完成
        console.log('所有文档已更新');
    } catch (err) {
        console.error('更新过程中发生错误', err);
    }
}

updateDocuments();

这段代码首先查询所有的文档，然后根据不同的条件来更新这些文档，并最终使用 Promise.all 确保所有更新操作都完成。这样既可以处理多个文档，又可以根据不同条件来更新它们。