使用 Mongoose 查找或创建

Question

我有

Page.findById(pageId).then(page => {
  const pageId = page.id;
   ..
});

我的问题是，如果没有给出页面 id，它应该只取第一个可用的页面给定一些条件，这是由

Page.findOne({}).then(page => {
  const pageId = page.id;
  ..
});

但是如果没有找到页面，它应该创建一个新页面并使用它，这是完成的

Page.create({}).then(page => {
  const pageId = page.id;
  ..
});

但是如何将所有这些组合到尽可能少的行中呢？

我有很多逻辑在里面发生

page => { ... }

所以我很想聪明地做这件事，所以我可以避免这样做

if (pageId) {
  Page.findById(pageId).then(page => {
    const pageId = page.id;
     ..
  });
} else {
  Page.findOne({}).then(page => {
    if (page) {
      const pageId = page.id;
      ..
    } else {
      Page.create({}).then(page => {
        const pageId = page.id;
        ..
      });
    }
  });
}

我在想我也许可以用类似的东西为架构分配一个静态

pageSchema.statics.findOneOrCreate = function (condition, doc, callback) {
  const self = this;
  self.findOne(condition).then(callback).catch((err, result) => {
    self.create(doc).then(callback);
  });
};

Answer 1

根据猫鼬docs：

根据previous SO answer

Model.findByIdAndUpdate()

“找到一个匹配的文档，根据更新参数更新它，传递任何选项，并将找到的文档（如果有的话）返回给回调。”

在选项中设置 upsert 为 true：

upsert: bool - 如果对象不存在则创建它。默认为 false。

Model.findByIdAndUpdate(id, { $set: { name: 'SOME_VALUE' }}, { upsert: true  }, callback)

Answer 2

与 Yosvel Quintero 的回答有关，但对我不起作用：

pageSchema.statics.findOneOrCreate = function findOneOrCreate(condition, callback) {
    const self = this
    self.findOne(condition, (err, result) => {
        return result ? callback(err, result) : self.create(condition, (err, result) => { return callback(err, result) })
    })
}

然后像这样使用它：

Page.findOneOrCreate({ key: 'value' }, (err, page) => {
    // ... code
    console.log(page)
})

Answer 3

承诺async/await 版本。

Page.static('findOneOrCreate', async function findOneOrCreate(condition, doc) {
  const one = await this.findOne(condition);

  return one || this.create(doc);
});

用法

Page.findOneOrCreate({ id: page.id }, page).then(...).catch(...)

或者

async () => {
  const yourPage = await Page.findOneOrCreate({  id: page.id }, page);
}

Answer 4

每个模式都可以为其模型定义实例和static methods。静态与方法几乎相同，但允许定义直接存在于模型中的函数

静态方法findOneOrCreate:

pageSchema.statics.findOneOrCreate = function findOneOrCreate(condition, doc, callback) {
  const self = this;
  self.findOne(condition, (err, result) => {
    return result 
      ? callback(err, result)
      : self.create(doc, (err, result) => {
        return callback(err, result);
      });
  });
};

现在，当您拥有Page 的实例时，您可以调用findOneOrCreate：

Page.findOneOrCreate({id: 'somePageId'}, (err, page) => {
  console.log(page);
});

Answer 5

async/await 的单行解决方案：

const page = Page.findOne({}).then(p => p || p.create({})

Answer 6

如果你不想在模型中添加静态方法，你可以尝试移动一些东西，至少不要让所有这些回调嵌套级别：

function getPageById (callback) {
  Page.findById(pageId).then(page => {
    return callback(null, page);
  });
}

function getFirstPage(callback) {
  Page.findOne({}).then(page => {
    if (page) {
      return callback(null, page);
    }

    return callback();
  });
}

let retrievePage = getFirstPage;
if (pageId) {
  retrievePage = getPageById;
}

retrievePage(function (err, page) {
  if (err) {
    // @todo: handle the error
  }

  if (page && page.id) {
    pageId = page.id;
  } else {
    Page.create({}).then(page => {
      pageId = page.id;
    });
  }
});

Answer 7

试试这个..

 var myfunc = function (pageId) {
  // check for pageId passed or not
 var newId = (typeof pageId == 'undefined') ? {} : {_id:pageId};

 Page.findOne(pageId).then(page => {
 if (page)
 const pageId = page.id;
 else {  // if record not found, create new

    Page.create({}).then(page => {
        const pageId = page.id;
    });
  }
});

 }