使用Mongoose findByIdAndUpdate更新父对象嵌套数组对象的部分字段

模型代码 #

const mandateSchema = new mongoose.Schema({
    _id: {
        type: String,
        required: true,
    },
    customer: {
        type: String,
        required: true,
        index: true
    },
    tariffs: {
        gas: {
            product_name: {
                type: String
            },
            code: {
                type: String
            },
            default_coefficients: {
                type: Boolean,
                default: true
            },
            assigned_eans: [{
                eanID: {
                    type: String,
                    ref: Ean,
                },
                status: {
                    type: String,
                    required: false
                },
                start_date: {
                    type: Number,
                },
                end_date: {
                    type: Number,
                },
                segment: {
                    type: String,
                    enum: ['b2c', 'soho', 'b2b', 'corporate', 'industry', 'direction']
                },
                stop_reason: {
                    type: String
                },
                stop_note: {
                    type: String
                },
            }]
        },
        electricity: {
            product_name: {
                type: String
            },
            code: {
                type: String
            },
            default_coefficients: {
                type: Boolean,
                default: true
            },
            assigned_eans: [{
                eanID: {
                    type: String,
                    ref: Ean,
                },
                start_date: {
                    type: Number,
                },
                end_date: {
                    type: Number,
                },
                segment: {
                    type: String,
                    enum: ['b2c', 'soho', 'b2b', 'corporate', 'industry', 'direction']
                },
                stop_reason: {
                    type: String
                },
                stop_note: {
                    type: String
                },
            }]
        }
    },
    start_date: {
        type: mongoose.Schema.Types.Mixed,
        required: false
    },
    end_date: {
        type: mongoose.Schema.Types.Mixed,
        required: false
    },
    end_from_ip: {
        type: String,
        required: false
    },
    stop_reason: {
        type: String,
        required: false
    },
    stop_reason_description: {
        type: String,
        required: false
    },
    sign_date: {
        type: Number,
        required: false
    },
    sign_token: {
        type: String,
        required: false
    },
    sign_from_ip: {
        type: String,
        required: false
    }
});

需求 #

根据给定的_id和eanID，更新匹配该eanID的对象，设置新的status、end_date、stop_reason以及可选的stop_note，同时保留该对象已有的start_date和eanID。目前已写出EnergyMandate.findByIdAndUpdate(energyMandateId，但不知道如何定位到tariffs.gas和tariffs.electricity中匹配的对象完成更新。

解决方案 #

方法一：使用数组过滤器批量更新

利用MongoDB的数组更新操作符，结合arrayFilters定位两个嵌套数组中的目标元素，一次完成更新：

const updateResult = await EnergyMandate.findByIdAndUpdate(
  energyMandateId,
  {
    $set: {
      // 更新gas数组中匹配eanID的元素
      "tariffs.gas.assigned_eans.$[gasEan].status": newStatus,
      "tariffs.gas.assigned_eans.$[gasEan].end_date": newEndDate,
      "tariffs.gas.assigned_eans.$[gasEan].stop_reason": stopReason,
      // 可选更新stop_note
      ...(stopNote && { "tariffs.gas.assigned_eans.$[gasEan].stop_note": stopNote }),
      // 更新electricity数组中匹配eanID的元素
      "tariffs.electricity.assigned_eans.$[elecEan].status": newStatus,
      "tariffs.electricity.assigned_eans.$[elecEan].end_date": newEndDate,
      "tariffs.electricity.assigned_eans.$[elecEan].stop_reason": stopReason,
      // 可选更新stop_note
      ...(stopNote && { "tariffs.electricity.assigned_eans.$[elecEan].stop_note": stopNote })
    }
  },
  {
    arrayFilters: [
      { "gasEan.eanID": eanID },
      { "elecEan.eanID": eanID }
    ],
    new: true // 返回更新后的文档，方便验证结果
  }
);

• arrayFilters定义两个过滤器，分别匹配gas和electricity数组中eanID符合条件的元素，通过占位符$[gasEan]和$[elecEan]在$set中定位对应元素。
• 可选的stop_note通过扩展运算符处理，仅当参数存在时才添加更新字段。

方法二：先查询再手动修改（适合复杂逻辑场景）

如果需要额外校验或只更新其中一个数组的匹配项，可以先查询文档，找到目标元素后修改再保存：

const mandate = await EnergyMandate.findById(energyMandateId);
if (!mandate) {
  throw new Error("未找到对应记录");
}

// 处理gas数组的匹配项
const gasEanIndex = mandate.tariffs.gas.assigned_eans.findIndex(item => item.eanID === eanID);
if (gasEanIndex !== -1) {
  const target = mandate.tariffs.gas.assigned_eans[gasEanIndex];
  target.status = newStatus;
  target.end_date = newEndDate;
  target.stop_reason = stopReason;
  if (stopNote) target.stop_note = stopNote;
}

// 处理electricity数组的匹配项
const elecEanIndex = mandate.tariffs.electricity.assigned_eans.findIndex(item => item.eanID === eanID);
if (elecEanIndex !== -1) {
  const target = mandate.tariffs.electricity.assigned_eans[elecEanIndex];
  target.status = newStatus;
  target.end_date = newEndDate;
  target.stop_reason = stopReason;
  if (stopNote) target.stop_note = stopNote;
}

const updatedMandate = await mandate.save();

• 这种方式适合需要在更新前做额外逻辑判断的场景，比如仅更新找到的第一个匹配项，或者添加权限校验等。

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