如何基于单一差异属性值获取MongoDB产品变体的关联组合？

我在MongoDB里设计了带属性变体的产品Schema，拿iPhone 15举例，产品包含memory、color这类属性及对应变体集合。现在需求很明确：拿到某个特定变体（比如64GB黑色款）后，针对每个属性键，筛选出仅该属性值不同的关联变体——比如针对color属性，要筛出内存64GB、颜色不同的所有变体；针对memory属性，要筛出颜色为黑色、内存不同的所有变体。

先定义产品Schema示例 #

const productSchema = new mongoose.Schema({
  name: String,
  variants: [{
    memory: String,
    color: String,
    price: Number,
    stock: Number
    // 其他变体属性按需添加
  }]
});

const Product = mongoose.model('Product', productSchema);

基础实现：逐个属性筛选关联变体 #

假设已经从数据库拿到了目标变体targetVariant，咱们可以遍历属性键，生成对应筛选条件：

// 示例目标变体：64GB黑色款
const targetVariant = { memory: '64GB', color: 'Black', price: 599, stock: 100 };
// 指定要处理的属性键
const variantKeys = ['memory', 'color'];

const relatedVariants = {};

for (const key of variantKeys) {
  // 构建筛选条件：除当前属性外，其他属性值和目标变体完全一致，当前属性值不等于目标值
  const filter = {
    variants: {
      $elemMatch: Object.fromEntries(
        Object.entries(targetVariant)
          .filter(([k]) => k !== key)
          .concat([[key, { $ne: targetVariant[key] }]])
      )
    }
  };

  // 查询并提取匹配的变体
  const product = await Product.findOne(filter, { 'variants.$': 1 });
  relatedVariants[key] = product?.variants || [];
}

// 结果：relatedVariants.color 是64GB其他颜色的变体，relatedVariants.memory是黑色其他内存的变体
console.log(relatedVariants);

高效优化：用聚合管道一次性分组 #

如果变体属性多、数据量大，用聚合管道能一次性完成筛选和分组，效率更高：

const pipeline = [
  // 先匹配包含目标变体的产品
  { $match: { 'variants': targetVariant } },
  // 展开变体数组，方便逐个处理
  { $unwind: '$variants' },
  // 筛选仅单个属性和目标变体不同的条目
  { $match: {
    $expr: {
      $eq: [
        { $size: {
          $filter: {
            input: { $objectToArray: '$variants' },
            cond: { $ne: ['$$this.v', targetVariant[$$this.k]] }
          }
        }},
        1
      ]
    }
  }},
  // 按差异属性分组，把同属性差异的变体归为一组
  { $group: {
    _id: {
      $arrayElemAt: [
        { $filter: {
          input: { $objectToArray: '$variants' },
          cond: { $ne: ['$$this.v', targetVariant[$$this.k]] }
        }},
        0
      ].k
    },
    variants: { $push: '$variants' }
  }},
  // 转换为更易读的键值对格式
  { $replaceRoot: {
    newRoot: { $arrayToObject: [[{ k: '$_id', v: '$variants' }]] }
  }}
];

const result = await Product.aggregate(pipeline);
console.log(result[0]); // 直接得到按属性分组的关联变体集合

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