如何使用 mongoose updateMany 中间件提高性能？

Question

已解决：底部的解决方案

我有以下代码逐个元素更新：

//registerCustomers.js
const CustomerRegistrationCode = require("../models/CustomerRegistrationCode");

const setRegCodesToUsed = async (regCodes) => {
  for (let regCode of regCodes) {
    await setRegCodeToUsed(regCode._id);
  }
};

const setRegCodeToUsed = async (id) => {
  await CustomerRegistrationCode.findByIdAndUpdate(id, { used: true });
};

代码运行良好，但速度较慢，我想一次更新许多 (1000) 个 CustomerRegistrationCode。 我查看了 updateMany 中间件功能，但在网上和官方文档上发现的信息不多。我将代码更改为以下内容，但不知道如何进一步。

//registerCustomers.js
const setRegCodesToUsed = async (regCodes) => {
  await CustomerRegistrationCode.updateMany(regCodes);
}

//CustomerRegistrationCode.js
CustomerRegistrationCodeSchema.pre('updateMany', async function (next, a) {
  console.log('amount arguments: ', arguments.length); //is 2
  console.log(arguments); //both parameters are functions.
  next();
});

用 1000 个不同的 id 更新许多 CustomerRegistrationCode 的最佳方法是什么？

解决方案，感谢 Murat Colyaran

const setRegCodesToUsed = async (regCodes) => {
  const ids = [];
  regCodes.map(code => ids.push(code._id));
  await setRegCodeToUsed(ids);
};

const setRegCodeToUsed = async (ids) => {
  await CustomerRegistrationCode.updateMany(
    { _id: { $in: ids } },
    { used: true }
  );
};

Answer 1

这应该可行：

//registerCustomers.js
const CustomerRegistrationCode = require("../models/CustomerRegistrationCode");

const setRegCodesToUsed = async (regCodes) => {
    let ids = [];
    regCodes.map((code) => ids.push(code._id.toString()));
    await setRegCodeToUsed(ids);
};

const setRegCodeToUsed = async (ids) => {
    await CustomerRegistrationCode.updateMany(
        { 
            id : { $in: ids }
        },
        {
            used: true
        }
    );
};

我们只是解析 id 并使用 $in 发送批量请求，而不是为每条记录发送查询