MongoDB时间序列集合如何添加传感器类型一致性约束？

在MongoDB里，你可以用文档验证器或者辅助集合+唯一索引的方式，确保同一个传感器始终对应同一类型。

方法一：用文档验证器（推荐直接在时间序列集合里实现） #

MongoDB 5.0及以上版本支持给时间序列集合加文档验证器，你可以自定义规则，检查插入/更新的文档是否符合“同一传感器对应唯一类型”的要求。

创建带验证器的时间序列集合 #

db.createCollection("sensor_readings", {
  timeseries: {
    timeField: "timestamp",
    metaField: "metadata"
  },
  // 自定义验证规则
  validator: {
    $expr: {
      $eq: [
        // 查询当前传感器已存在的类型数量（只能是0或1种）
        {
          $size: {
            $distinct: {
              input: "$$ROOT.metadata.type",
              cond: { $eq: ["$$ROOT.metadata.sensor", "$metadata.sensor"] }
            }
          }
        },
        // 新传感器允许0种（无历史记录），已有传感器必须是1种且和当前文档类型一致
        { $cond: [{ $exists: { $query: { "metadata.sensor": "$metadata.sensor" } } }, 1, 0] }
      ]
    }
  },
  // 严格验证所有插入/更新操作
  validationLevel: "strict",
  // 违反规则直接拒绝操作并报错
  validationAction: "error"
})

这个验证器的逻辑很简单：

• 如果是第一次插入某个传感器的数据，随便指定类型都没问题；
• 如果这个传感器已经有历史数据，新插入的文档类型必须和之前的完全一致，不然操作会直接失败。

方法二：用辅助集合+唯一索引（适合对性能要求更高的场景） #

要是担心文档验证器的性能开销，你可以单独建一个辅助集合存传感器和类型的映射，用唯一索引强制约束，再关联时间序列集合做检查。

步骤1：创建辅助集合并加唯一索引 #

// 创建传感器类型映射集合
db.createCollection("sensor_type_mappings")
// 加唯一索引，确保每个传感器只能对应一个类型
db.sensor_type_mappings.createIndex({ sensor: 1 }, { unique: true })

步骤2：插入传感器类型映射 #

第一次插入某传感器的时间序列数据前，先往辅助集合里插一条映射记录：

db.sensor_type_mappings.insertOne({ sensor: "A", type: "temperature" })

步骤3：给时间序列集合加验证器关联检查 #

修改时间序列集合的验证规则，确保插入的数据和辅助集合里的映射一致：

db.runCommand({
  collMod: "sensor_readings",
  validator: {
    $expr: {
      $eq: [
        "$metadata.type",
        {
          $arrayElemAt: [
            {
              $lookup: {
                from: "sensor_type_mappings",
                localField: "metadata.sensor",
                foreignField: "sensor",
                as: "sensor_type"
              }
            },
            0
          ].sensor_type.type
        }
      ]
    }
  }
})

这样一来，任何和辅助集合映射不符的时间序列数据都会被拒绝插入。

注意事项 #

• 文档验证器每次插入/更新都会执行检查，数据量大的话可能有性能损耗；
• 时间序列集合的验证器只在MongoDB 5.0及以上版本支持；
• 用辅助集合的话，要确保先维护好映射关系再插时间序列数据，也可以用MongoDB触发器自动同步映射。

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