如何在Mongoose中为Product模型指定字段关联填充SKU数据

问题分析 #

你的需求是在Product模型的stores.package.sku字段存储Storage模型中嵌套的region.skus子文档ID，查询时自动填充该sku的详细信息。由于sku是Storage集合的嵌套子文档而非独立集合，Mongoose默认的populate无法直接处理这种深层嵌套的引用，需要通过聚合管道或者模型重构来实现。

方案一：基于现有模型结构的聚合管道实现 #

如果不想修改现有数据结构，可以使用MongoDB的聚合管道来关联查询并填充sku信息：

1. 修正ProductSchema的sku字段定义 #

首先明确sku字段为ObjectId类型：

const mongoose = require('mongoose');

const ProductSchema = new mongoose.Schema({
    name:{type:String, required:true, unique:true},
    stores:[{
        name:String,
        package:[{
            name:String,
            sku: { type: mongoose.Schema.Types.ObjectId } // 存储sku的_id
        }]
    }],
},{timestamps : true})
export default mongoose.models.Product || mongoose.model("Product", ProductSchema);

2. 编写聚合查询函数 #

通过聚合管道展开嵌套数组、关联Storage集合并匹配sku：

async function getProductWithSkuDetails(productId) {
  const result = await Product.aggregate([
    // 匹配目标产品
    { $match: { _id: mongoose.Types.ObjectId(productId) } },
    // 关联Storage集合，筛选包含目标sku的仓储数据
    {
      $lookup: {
        from: 'storages',
        let: { skuIds: '$stores.package.sku' },
        pipeline: [
          { $unwind: '$region' },
          { $unwind: '$region.skus' },
          { $match: { $expr: { $in: ['$region.skus._id', '$$skuIds'] } } },
          { $project: { sku: '$region.skus', _id: 0 } }
        ],
        as: 'matchedSkus'
      }
    },
    // 展开stores和package数组
    { $unwind: '$stores' },
    { $unwind: '$stores.package' },
    // 匹配对应的sku详情并替换原字段
    {
      $addFields: {
        'stores.package.sku': {
          $first: {
            $filter: {
              input: '$matchedSkus',
              cond: { $eq: ['$$this.sku._id', '$stores.package.sku'] }
            }
          }
        }
      }
    },
    // 恢复package数组结构
    {
      $group: {
        _id: { productId: '$_id', storeId: '$stores._id' },
        name: { $first: '$name' },
        storeName: { $first: '$stores.name' },
        package: { $push: '$stores.package' },
        createdAt: { $first: '$createdAt' },
        updatedAt: { $first: '$updatedAt' }
      }
    },
    // 恢复stores数组结构
    {
      $group: {
        _id: '$_id.productId',
        name: { $first: '$name' },
        stores: {
          $push: {
            _id: '$_id.storeId',
            name: '$storeName',
            package: '$package'
          }
        },
        createdAt: { $first: '$createdAt' },
        updatedAt: { $first: '$updatedAt' }
      }
    }
  ]);

  return result[0] || null;
}

方案二：重构模型为独立Sku集合（推荐） #

如果可以调整数据结构，将sku抽为独立集合，就能直接使用Mongoose的populate功能，代码更简洁易维护：

1. 创建独立的Sku模型 #

const SkuSchema = new mongoose.Schema({
  name: String,
  variant: [{ name: String }]
}, { timestamps: true });
export default mongoose.models.Sku || mongoose.model("Sku", SkuSchema);

2. 修改Storage模型引用Sku #

const StorageSchema = new mongoose.Schema({
    name: String,
    region: [{
        name: String,
        skus: [{ type: mongoose.Schema.Types.ObjectId, ref: 'Sku' }] // 引用Sku集合
    }]
},{timestamps: true})
export default mongoose.models.Storage || mongoose.model("Storage", storageSchema);

3. 修改Product模型引用Sku #

const ProductSchema = new mongoose.Schema({
    name:{type:String, required:true, unique:true},
    stores:[{
        name:String,
        package:[{
            name:String,
            sku: { type: mongoose.Schema.Types.ObjectId, ref: 'Sku' } // 引用Sku集合
        }]
    }],
},{timestamps : true})
export default mongoose.models.Product || mongoose.model("Product", ProductSchema);

4. 查询时直接使用populate #

// 查询单个产品并填充sku详情
const product = await Product.findById(productId)
  .populate({
    path: 'stores.package.sku',
    model: 'Sku'
  });

// 查询所有产品并填充sku详情
const products = await Product.find()
  .populate({
    path: 'stores.package.sku',
    model: 'Sku'
  });

方案对比 #

• 方案一无需修改现有数据结构，但聚合管道代码较复杂，性能略低于独立集合的populate。
• 方案二结构更清晰，Mongoose原生支持填充，代码简洁，维护成本低，是长期开发的推荐方案。

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