MongoDB的聚集與多個陣列

Question

我有以下兩個項目插入到集合中「幀」：

frame1 = { 
      "number": 1, 

      "hobjects": [ { "htype": 1, "weight": 50 }, 
         { "htype": 2, "weight": 220 }, 
         { "htype": 2, "weight": 290 }, 
         { "htype": 3, "weight": 450 } ], 

      "sobjects": [ { "stype": 1, "size": 10.0 }, 
         { "stype": 2, "size": 5.1 }, 
         { "stype": 2, "size": 6.5 } ], 
      } 

frame2 = { 
      "number": 2, 

      "hobjects": [ { "htype": 1, "weight": 61 }, 
         { "htype": 2, "weight": 210 }, 
         { "htype": 2, "weight": 250 } ], 

      "sobjects": [ { "stype": 1, "size": 12.1 }, 
         { "stype": 2, "size": 4.9 }, 
         { "stype": 2, "size": 6.2 }, 
         { "stype": 2, "size": 5.7 } ], 
      } 

frames.insert(frame1) 
frames.insert(frame2) 

現在我想要做在部分幀數據的查詢：

query = { "hobjects.htype": 3, "sobjects.stype": 2 } 
db.frames.find(query) 

其結果在：

{u'_id': ObjectId('545b6ea7b9ad9a03462d743b'), u'hobjects': [{u'htype': 1, u'weight': 50}, {u'htype': 2, u'weight': 220}, {u'htype': 2, u'weight': 290}, {u'htype': 3, u'weight': 450}], u'number': 1, u'sobjects': [{u'stype': 1, u'size': 10.0}, {u'stype': 2, u'size': 5.1}, {u'stype': 2, u'size': 6.5}]} 

而不是我真正想要的。我想有收集過濾，根據查詢，使我得到以下的結果，而不是：

{u'_id': ObjectId('545b6ea7b9ad9a03462d743b'), u'hobjects': [{u'htype': 3, u'weight': 450}], u'number': 1, u'sobjects': [{u'stype': 2, u'size': 5.1}, {u'stype': 2, u'size': 6.5}]} 

我發現的唯一的解決方案涉及平倉聚集和集合分組：

query = { "hobjects.htype": 3, "sobjects.stype": 2 } 
db.frames.aggregate([ 
    { "$match": query }, 
    { "$unwind": "$hobjects" }, 
    { "$match": dict((key, value) for key, value in query.iteritems() if "hobjects." in key) }, 
    { "$group": { "_id": "$_id", "number": { "$first": "$number" } , "hobjects": { "$push": "$hobjects" }, "sobjects": { "$first": "$sobjects" } } }, 
    { "$unwind": "$sobjects" }, 
    { "$match": dict((key, value) for key, value in query.iteritems() if "sobjects." in key) }, 
    { "$group": { "_id": "$_id", "number": { "$first": "$number" } , "hobjects": { "$first": "$hobjects" }, "sobjects": { "$push": "$sobjects" } } }, 
    ]) 

我想這就是不是一種非常有效和靈活的查詢方式。我想知道是否還有其他選擇？

Answer 1

只要你的服務器是MongoDB的2.6或更高版本，那麼你總是可以做到這一點：

db.frames.aggregate([ 

    // Still helps to match the documents by conditions to filter 
    { "$match": { 
     "hobjects.htype": 3, "sobjects.stype": 2 
    }}, 

    // Now filter inline using $map and $setDifference 
    { "$project": { 
     "number": 1, 
     "hobjects": { 
      "$setDifference": [ 
       { "$map": { 
        "input": "$hobjects", 
        "as": "el", 
        "in": { 
         "$cond": [ 
          { "$eq": [ "$$el.htype", 3 ] }, 
          "$$el", 
          false 
         ] 
        } 
       }}, 
       [false] 
      ] 
     }, 
     "sobjects": { 
      "$setDifference": [ 
       { "$map": { 
        "input": "$sobjects", 
        "as": "el", 
        "in": { 
         "$cond": [ 
          { "$eq": [ "$$el.stype", 2 ] }, 
          "$$el", 
          false 
         ] 
        } 
       }}, 
       [false] 
      ] 
     } 
    }} 
]) 

這裏美中不足的是，基本的投影之類的東西$elemMatch只能目前的第一元素匹配一個匹配條件的數組。因此，爲了做更多的事情，您需要某種形式的高級操作，這些操作僅適用於諸如聚合框架之類的事物。

$setDiffence和$map運營商給你一個「內聯」的方式來處理數組和實際上在單個文檔中的「集」。這比使用$unwind更有效，特別是在涉及大型數組的情況下。

JavaScript表示法我知道（主要在註釋中），但它基本上與python相同。

Answer 2

在聚集可以幫助你

db.frames.aggregate({"$unwind":"$hobjects"},{"$unwind":"$sobjects"},{"$match":{"hobjects.htype": 3, "sobjects.stype": 2}},{"$group":{"_id":"$_id","u'hobjects":{"$first":"$hobjects"},"u'number":{"$first":"$number"},"u'sobjects":{"$push":"$sobjects"}}})