nodejs mongoose批量更新

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

const userSchema = new mongoose.Schema({
  name: String,
  age: Number,
  active: Boolean
});

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

// 使用 updateMany 批量更新
User.updateMany(
  { active: true }, // 查询条件
  { $set: { age: 30 } }, // 更新操作
  (err, result) => {
    if (err) return console.error(err);
    console.log(result);
  }
);

// 使用 bulkWrite 批量更新
User.bulkWrite([
  { updateOne: { filter: { name: 'Alice' }, update: { $set: { age: 31 } } } },
  { updateOne: { filter: { name: 'Bob' }, update: { $set: { age: 32 } } } }
], (err, result) => {
  if (err) return console.error(err);
  console.log(result);
});

const batchSize = 1000;
let count = 0;
const updateUsers = async () => {
  const users = await User.find({ active: true }).limit(batchSize);
  if (users.length === 0) return;
  await User.updateMany(
    { _id: { $in: users.map(user => user._id) } },
    { $set: { age: 30 } }
  );
  count += users.length;
  console.log(`Updated ${count} users`);
  await updateUsers();
};

updateUsers().catch(console.error);

const session = await mongoose.startSession();
session.startTransaction();
try {
  await User.updateMany({ active: true }, { $set: { age: 30 } }, { session });
  await session.commitTransaction();
} catch (err) {
  await session.abortTransaction();
  throw err;
} finally {
  session.endSession();
}