具有两个字段的MongoDB聚合数组

时间:2015-05-14 14:26:45

标签: arrays mongodb mongodb-query

我的车辆系列采用以下架构,所有文章只是一般产品(不包括儿童产品):

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

我有产品系列,其中一些是与仓库sku相关的一般产品,有些产品是" son"这些儿子产品适用于多种通用产品,也与仓库sku的相关:

  
    

一般产品

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

儿童产品

  
{
    "_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")] 
  
    

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

  

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

1 个答案:

答案 0 :(得分:2)

使用聚合游标的map()方法返回一个ObjectIds数组,如下所示:

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需要传入一个函数,所以它恰好也是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))