MongoDB跨集合字段比对与数据修正方案咨询

需求说明 #

需要将vehicles集合中的正确energy_type同步到bookings集合的vehicle_energy_type字段，同时当车辆类型为electric时，将battery_level统一设置为10（无论原字段是否存在或正确）。

解决方案代码 #

直接使用MongoDB 4.2+支持的更新管道实现批量同步与修复：

db.bookings.updateMany(
  {}, // 匹配所有预订记录，如需仅处理异常记录可修改过滤条件
  [
    {
      $lookup: {
        from: "vehicles",
        localField: "vehicle_id",
        foreignField: "_id",
        as: "vehicle_info"
      }
    },
    {
      $unwind: {
        path: "$vehicle_info",
        preserveNullAndEmptyArrays: false // 仅处理能关联到车辆信息的记录
      }
    },
    {
      $set: {
        // 强制同步车辆类型为vehicles集合中的正确值
        vehicle_energy_type: "$vehicle_info.energy_type",
        // 仅当车辆类型为electric时设置battery_level为10，其他情况保留原字段值
        battery_level: {
          $cond: {
            if: { $eq: ["$vehicle_info.energy_type", "electric"] },
            then: 10,
            else: "$battery_level"
          }
        }
      }
    },
    {
      $unset: "vehicle_info" // 清理临时关联字段
    }
  ]
)

代码解释 #

1. updateMany：对目标集合执行批量更新，第一个参数为空对象表示处理所有记录。若只想修复异常数据，可改为：

   { $or: [{ vehicle_energy_type: { $ne: "electric" } }, { battery_level: { $exists: false } }] }
   

2. $lookup：通过vehicle_id关联vehicles集合，获取车辆的正确能源类型信息。
3. $unwind：展开关联结果数组，确保每条预订记录对应唯一车辆信息。
4. $set：更新核心逻辑，同步正确的车辆类型，并根据类型设置电池电量。
5. $unset：移除临时生成的vehicle_info字段，保持集合结构整洁。

验证结果 #

执行更新后，查询bookings集合：

db.bookings.find()

返回结果将符合预期：

{ "_id" : 1, "vehicle_id" : "1", "vehicle_energy_type" : "electric", "battery_level" : 10 }
{ "_id" : 2, "vehicle_id" : "1", "vehicle_energy_type" : "electric", "battery_level" : 10 }
{ "_id" : 3, "vehicle_id" : "1", "vehicle_energy_type" : "electric", "battery_level" : 10 }

原聚合操作的问题 #

你之前的聚合仅筛选出vehicle_energy_type与车辆信息匹配的记录，没有对数据进行修改，无法实现数据修复的目的。而使用更新管道结合$lookup可以直接完成数据同步与修复。

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