Mongoose/Node.js如何更新数组内对象字段不替换整体及简化if判断的方法

核心逻辑是动态构建$set对象，遍历传入的更新字段自动拼接数组定位符前缀，完全不需要冗余的if判断，同时还支持同时更新多个字段。

实现代码 #

const menuId = req.params.menuId;
const userId = userIdFromToken(req);
// 定义允许更新的菜单字段白名单，避免恶意更新非法字段
const ALLOWED_MENU_FIELDS = ['name', 'price', 'description'];
// 过滤出合法的更新字段
const updateFields = Object.entries(req.body).filter(([key]) => ALLOWED_MENU_FIELDS.includes(key));
// 动态构建$set对象
const setObj = {};
updateFields.forEach(([key, value]) => {
  setObj[`menus.$.${key}`] = value;
});
const update = { $set: setObj };

const filter = {
  userId: userId,
  "menus._id": menuId
};
const options = {
  upsert: true,
  new: true
};

Restaurant.findOneAndUpdate(filter, update, options)
  .then(() => {
    return res.status(204).json({ message: "Menus updated" });
  });

方案优势 #

• 完全去掉所有if判断，代码更简洁易维护
• 支持同时更新多个字段，比如传入{price:12, description:'新描述'}也能正常处理，你之前修改后的代码只能处理单个字段的更新，存在多字段更新失效的问题
• 增加了字段白名单校验，避免前端传入非法字段导致的数据安全问题

原修改版代码的缺陷 #

你之前调整后的代码中，遍历req.body的逻辑会覆盖this.menuKey和this.menuValue，如果传入多个更新字段，只会保留最后一个字段的更新，逻辑存在漏洞，本方案不存在该问题。

内容的提问来源于stack exchange，提问作者David