猫鼬-创建文档(如果不存在),否则,在两种情况下均会更新并返回文档


71

我正在寻找一种将代码的一部分重构为更短,更简单的方法,但是我对Mongoose不太了解,我不确定该如何进行。

我正在尝试检查集合中是否存在文档,如果不存在,则创建它。如果确实存在,我需要对其进行更新。无论哪种情况,我都需要随后访问文档的内容。

到目前为止,我设法做到的是查询集合中的特定文档,如果找不到,则创建一个新文档。如果找到它,我将对其进行更新(当前使用日期作为虚拟数据)。从那里,我可以从我的初始find操作中访问找到的文档,也可以访问新保存的文档,并且此方法有效,但是必须有更好的方法来完成我要执行的操作。

这是我的工作代码,没有其他功能。

var query = Model.find({
    /* query */
}).lean().limit(1);

// Find the document
query.exec(function(error, result) {
    if (error) { throw error; }
    // If the document doesn't exist
    if (!result.length) {
        // Create a new one
        var model = new Model(); //use the defaults in the schema
        model.save(function(error) {
            if (error) { throw error; }
            // do something with the document here
        });
    }
    // If the document does exist
    else {
        // Update it
        var query = { /* query */ },
            update = {},
            options = {};

        Model.update(query, update, options, function(error) {
            if (error) { throw error; }
            // do the same something with the document here
            // in this case, using result[0] from the topmost query
        });
    }
});

我研究了findOneAndUpdate其他相关方法,但不确定它们是否适合我的用例,或者我是否了解如何正确使用它们。谁能指出我正确的方向?

(大概)相关问题:


编辑

在搜索过程中,我没有遇到指向我的问题,但是在查看了答案之后,我提出了这个问题。在我看来,它当然更漂亮,并且可以正常工作,因此,除非我做的非常糟糕,否则我认为我的问题可能会结束。

我将不胜感激我的解决方案上的任何其他投入。

// Setup stuff
var query = { /* query */ },
    update = { expire: new Date() },
    options = { upsert: true };

// Find the document
Model.findOneAndUpdate(query, update, options, function(error, result) {
    if (!error) {
        // If the document doesn't exist
        if (!result) {
            // Create it
            result = new Model();
        }
        // Save the document
        result.save(function(error) {
            if (!error) {
                // Do something with the document
            } else {
                throw error;
            }
        });
    }
});

1
Que:Mongoose.js:如何实现创建或更新?答:stackoverflow.com/questions/7267102/…–
阿洛克·德什瓦尔

老实说,我现在感觉很愚蠢。我在搜索时没有找到该问题,但回想起来,答案似乎很容易掌握。感谢您的帮助!
康纳2015年

Answers:


130

您正在寻找newoption参数。该new选项返回新创建的文档(如果创建了新文档)。像这样使用它:

var query = {},
    update = { expire: new Date() },
    options = { upsert: true, new: true, setDefaultsOnInsert: true };

// Find the document
Model.findOneAndUpdate(query, update, options, function(error, result) {
    if (error) return;

    // do something with the document
});

由于upsert创建文档(如果找不到文档),因此您无需手动创建另一个文档。


2
除未使用在我的架构中定义的默认值填充新创建的文档外,这似乎可行。很混乱。您知道有什么原因吗?
Connor 2015年

2
@Connor是的,它是猫鼬的默认功能,尽管有一个解决方案。检查我更新的答案。

有点奇怪的怪癖,但是我认为开发人员有他们的理由。这是解决我的问题的绝佳解决方案,因此感谢您的帮助!
康纳

1
在这种情况下,您如何确定发送哪个状态代码?200 =更新。201年创建
jhonny lopez

如果我要更新的对象具有ID列表,而我想向其中添加一个新ID,该怎么办?将替换或添加到列表中吗?
shinzou

14

由于您希望将代码的一部分重构得更短,更简单,

  1. 采用 async / await
  2. .findOneAndUpdate()按照此答案的建议使用

let query = { /* query */ };
let update = {expire: new Date()};
let options = {upsert: true, new: true, setDefaultsOnInsert: true};
let model = await Model.findOneAndUpdate(query, update, options);

0
///This is simple example explaining findByIDAndUpdate from my code added with try catch block to catch errors
try{
const options = {
            upsert: true,
            new: true,
            setDefaultsOnInsert: true
        };
        const query = {
            $set: {
                description: req.body.description,
                title: req.body.title
            }
        };
        const survey = await Survey.findByIdAndUpdate(
            req.params.id,
            query,
            options
        ).populate("questions");
}catch(e){
console.log(e)
}

-1

这是一个例子:

const mongoose = require('mongoose');
mongoose.connect('mongodb://localhost/rsvp', {useNewUrlParser: true, useUnifiedTopology: true});

const db = mongoose.connection;

db.on('error', () => {
  console.log('mongoose connection error');
});

db.once('open', () => {
  console.log('mongoose connected successfully');
});

const rsvpSchema = mongoose.Schema({
  firstName: String,
  lastName: String,
  email: String,
  guests: Number
});

const Rsvp = mongoose.model('Rsvp', rsvpSchema);


// This is the part you will need... In this example, if first and last name match, update email and guest number. Otherwise, create a new document. The key is to learn to put "upsert" as the "options" for the argument.
const findRsvpAndUpdate = (result, callback) => {

  Rsvp.findOneAndUpdate({firstName: result.firstName, lastName: result.lastName}, result, { upsert: true }, (err, results) => {
    if (err) {
      callback(err);
    } else {
      callback(null, results);
    }
  })
};


// From your server index.js file, call this...
app.post('/rsvps', (req, res) => {
  findRsvpAndUpdate(req.body, (error, result) => {
    if (error) {
      res.status(500).send(error);
    } else {
      res.status(200).send(result);
    }
  })
});

By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.