MongoDB 包含两个字段的聚合数组

Question

我有以下架构的车辆集合，所有文章都只是一般产品（不包括子产品）：

{
    "_id" : ObjectId("554995ac3d77c8320f2f1d2e"),
    "model" : "ILX",
    "year" : 2015,
    "make" : "Acura",
    "motor" : {
        "cylinder" : 4,
        "liters" : "1.5"
    },
    "products" : [
        ObjectId("554f92433d77c803836fefe3"),
        ...
    ]
}

我有产品集合，其中一些是与仓库 sku 相关的一般产品，一些产品是 "son" 适合多个一般产品的产品，这些子产品也与仓库 sku 相关：

general products

{
  "_id" : ObjectId("554b9f223d77c810e8915539"),
  "brand" : "Airtex",
  "product" : "E7113M",
"type" : "Fuel Pump",
"warehouse_sku" : [
    "1_5551536f3d77c870fc388a04",
    "2_55515e163d77c870fc38b00a"
]
}

child product

{
    "_id" : ObjectId("55524d0c3d77c8ba9cb2d9fd"),
    "brand" : "Performance",
    "product" : "P41K",
    "type" : "Repuesto Bomba Gasolina",
    "general_products" : [
        ObjectId("554b9f223d77c810e8915539"),
        ObjectId("554b9f123d77c810e891552f")
    ],
    "warehouse_sku" : [
        "1_555411043d77c8066b3b6720",
        "2_555411073d77c8066b3b6728"
    ]
}

我的问题是获取 warehouse_sku 的一般产品列表（子产品中的 _id 和 general_products）遵循以下模式：1_

我创建了一个具有以下结构的聚合查询：

list_products = db.getCollection('products').aggregate([
...  {$match: {warehouse_sku: /^1\_/}},
...  {$group: { "_id": "$_id" } }
... ])

那个查询成功地给我一个结果：

{ "_id" : ObjectId("55524d0c3d77c8ba9cb2d9fd") }
{ "_id" : ObjectId("554b9f223d77c810e8915539") }

但我需要获得一份通用产品列表，以便我可以在车辆集合中使用 $in。

list_products = [ ObjectId("55524d0c3d77c8ba9cb2d9fd"), ObjectId("554b9f223d77c810e8915539")]

example: db.vehicles.find({products:{$in: list_products}})

最后一个查询我无法实现。

Answer 1

使用聚合游标的 map() 方法 return ObjectId 数组，如下所示：

var pipeline = [
    {$match: {warehouse_sku: /^1\_/}},
    {$group: { "_id": "$_id" } }
],
list_products = db.getCollection('products')
                  .aggregate(pipeline)
                  .map(function(doc){ return doc._id });

find() 光标的 map() 也可以在这里工作：

var query = {'warehouse_sku': /^1\_/},
    list_products = db.getCollection('products')
                      .find(query)
                      .map(function(doc){ return doc._id });

更新

在 pymongo 中，您可以将 lambda 函数与 map 函数一起使用。因为map期望传入一个函数，所以它也恰好是lambda常出现的地方之一：

import re
regx = re.compile("^1\_", re.IGNORECASE)
products_cursor = db.products.find({"warehouse_sku": regx})
list_products = list(map((lambda doc: doc["_id"]), products_cursor))

MongoDB 包含两个字段的聚合数组

MongoDB Aggregate Array with Two Fields

arrays

mongodb

mongodb-query