以原子方式更新多个文档并返回它们

Question

在MongoDB中，我正在寻找一种以原子方式更新多个文档并在一次调用中返回所有更新的文档的方法。

我们可以在MongoDB中执行以下所有操作：

• 原子更新一个文档并返回更新后的文档： findAndModify或findOneAndUpdate
• 以原子方式更新多个文档： update(...{multi: true}或updateMany
• 查询并返回多个文档： find

我不喜欢更新多个文档并将它们全部返回的方法。 有办法吗？ 我使用Mongoose作为查询包。

Answer 1

以原子方式更新多个文档： update(...{multi: true}或updateMany

毫无疑问，这是错误的：

在 MongoDB 中，写入操作，例如 db.collection.update()、db.collection.findAndModify()、db.collection.remove()，在单个文档级别上是原子的。

---

在 MongoDB 中，写入操作在单个文档级别上是原子的，即使该操作修改了单个文档中的多个嵌入文档。

但是，您可以通过“使用两阶段提交方法”模拟事务以原子更新多个文档，该方法在此处进行了详细描述。

您还可以查看$isolated运算符，它“防止影响多个文档的写入操作在写入第一个文档后让步给其他读取或写入” ，但它“不为写入提供“全有或全无”原子性操作”

总而言之，在 mongodb 级别（也不是驱动程序）是不可能的，但您可以在应用程序级别模拟它，因此返回您需要的内容。

Answer 2

MongoDB v3.6 引入了会话，这使得这成为可能： https ://docs.mongodb.com/manual/reference/method/Session/。 我使用猫鼬，这使得使用它们非常简单。 这样的事情可能适用于您的情况：

const session = await mongoose.startSession();

let result;

// NOTE `withTransaction` expects a promise so I'm using `async`
await session.withTransaction(async () => {
  await SomeModel.update({
    foo: "bar"
  }, {
    $set: {
      quux:"baz"
    }
  }).session(session);

  result = await SomeModel.find({
    foo: bar
  }).session(session);
});

return result;

Answer 3

我测试了updateMany。

测试1：

使用 updateMany (pull) 更新 40K 文档，在执行过程中，突然关闭 db，然后一些节点通过（数据被拉出），一些失败（数据未在树中的某些级别 5 节点中拉出），重新启动 db 并再次运行 updateMany ，所有通过并且所有数据现在都是正确的。

测试2：

在字段上创建唯一索引，插入一些数据，在 updateMany 方法中，一些文档会因为唯一键冲突而失败。

我的 test2 结果是：零文档已更新。

---

function insertData() {
  const dataSource = app.models.Entity.getDataSource();
  return new Promise((resolve, reject) => {
    dataSource.connector.connect((err, db) => {
      if (err) {
        reject(new Error('.... error'));
        return;
      }
      const entityCollection = db.collection('Entity');
      // Create index
      entityCollection.createIndex({ age: 1 }, { unique: true })
      .then(() => {
        // Insert data
        const data = [
          {
            id: uuid.v4(),
            age: 1,
            type: 'test',
          },
          {
            id: uuid.v4(),
            age: 2,
            type: 'test',
          },
          {
            id: uuid.v4(),
            age: 3,
            type: 'test',
          },
          {
            id: uuid.v4(),
            age: 4,
            type: 'test',
          },
          {
            id: uuid.v4(),
            age: 5,
            type: 'test',
          },
          {
            id: uuid.v4(),
            age: 6,
            type: 'test',
          },
        ];
        return insertData(data);
      })
      .then(() => {
        resolve();
      })
      .catch((err2) => {
        reject(err2);
      });
    });
  });
}

function updateAge() {
  const dataSource = app.models.Entity.getDataSource();
  return new Promise((resolve, reject) => {
    dataSource.connector.connect((err, db) => {
      if (err) {
        reject(new Error('...error'));
        return;
      }
      const entityCollection = db.collection('Entity');
      entityCollection.updateMany(
        { age: { $gt: 0 } },
        { $mul: { age: 2 } },
      ).then(() => {
        resolve();
      })
      .catch((err2) => {
        logger.error(`ERROR is ${err2}`);
        reject(err2);
      });
    });
  });
}

---

测试结果是：零个文档被更新。 "msg":"ERROR is MongoError: E11000 duplicate key error collection: content-base.Entity index: age_1 dup key: { : 2 }","v":1} 1) updateMany test

0 通过 (114ms) 1 失败