如何从聚合结果更新 MongoDB 集合，在数组中设置值答案

【问题标题】：How to update MongoDB collection from aggregation results, set values within array如何从聚合结果更新 MongoDB 集合，在数组中设置值
【发布时间】：2021-03-07 19:50:14
【问题描述】：

使用 PyMongo，我将一个集合 (refer_docs) 中的 MongoDB 文档与另一个集合 (test) 中的文档进行内部连接，并且我想找到那些文档，其中连接的每个文档中的字段值不t 匹配，我想更新这些值。 refer_docs 文档应该说明它指向的测试文档具有正确的文档类型。

我可以找到这些文档，但我想在不循环遍历聚合结果列表并一次运行一个更新的情况下进行更新。

是否有 Mongo/PyMongo 方法可以在更新中使用聚合管道作为查询，或者在聚合（）中使用 $set？还是？

管道作为集合与文档之间的连接开始，如下例所示：

import pymongo
mongo_client = MongoClient('localhost:27017')
osm_db = mongo_client.osm
refer_docs_col = osm_db["refer_docs"]
test_col = osm_db["test"]

# test collection:
test_col.insert_many(
[
 {'_id': '611868136', 'doc_type': 'way'},
 {'_id': '5792648632', 'doc_type': 'node'},
 {'_id': '611868133', 'doc_type': 'node'},
 {'_id': '1', 'doc_type': 'node'}
]
)

# refer_docs collection:
refer_docs_col.insert_many(
[
 {'_id': '8483444',
  'refs': [{'ref': '611868136', 'ref_type': 'way'},
           {'ref': '5792648632', 'ref_type': 'node'},
           {'ref': '611868133', 'ref_type': 'way'}],
  'doc_type': 'relation'}
]
)

现在，这是管道和更新集合的一次尝试，它本质上是循环遍历结果，但它不起作用：

pipeline = [
    { "$unwind" : "$refs" },
    {
        "$lookup" : {
            "from" : "test",
            "localField" : "refs.ref",
            "foreignField" : "_id",
            "as" : "ref_doc"
        }
    },
    { "$match" : { "ref_doc" : { "$ne" : [] } } },
    { "$unwind" : "$ref_doc"},
    { "$project" : { "_id" : 1, "refs" : 1, "ref_doc.doc_type" : 1, 
                     "cmp" : { "$cmp" : [ "$refs.ref_type",
                                          "$ref_doc.doc_type" ] } } },
    { "$match" : { "cmp" : { "$ne" : 0 } } },
]

result = [
 refer_docs_col.find_one_and_update( doc,
    { "$set" : { "refs.ref_type" : "$ref_doc.doc_type" } } ) \
 for doc in refer_docs_col.aggregate(pipeline)
]

refer_docs_col.find_one( { "_id" : "8483444" } )

这不起作用，但我想查看文档已更新，以便 ref "611586133" 的 ref_type 现在为 "node"：

{'_id': '8483444',
 'refs': [{'ref': '611868136', 'ref_type': 'way'},
          {'ref': '5792648632', 'ref_type': 'node'},
          {'ref': '611868133', 'ref_type': 'node'}],
 'doc_type': 'relation'}

【问题讨论】：

标签： mongodb aggregation-framework pymongo updates findandmodify

【解决方案1】：

这行得通。

这里我用"refs" : mismatch["refs"]在过滤器中查找数组元素，"refs.$"设置找到的数组元素。

for mismatch in result: # Loop aggregation result docs
    filtr = { "_id" : mismatch["_id"] ,
              "refs" : mismatch["refs"]}
    update = { "$set" : { "refs.$" : { 
        "ref" : mismatch["ref_doc"]["_id"],
        "ref_type" : mismatch["ref_doc"]["doc_type"] } } }
    doc = refer_docs_col.update_one(filtr, update)

refer_docs_col.find_one( { "_id" : "8483444" } )

输出：

{'_id': '8483444',
 'refs': [{'ref': '611868136', 'ref_type': 'way'},
  {'ref': '5792648632', 'ref_type': 'node'},
  {'ref': '611868133', 'ref_type': 'node'}],
 'doc_type': 'relation'}

【讨论】：