MongoDB 将 $cond 与更新 ($inc) 结合使用

Question

有什么方法可以在更新中使用 $cond 和 ($set, $inc, ...) 运算符吗？ (MongoDB 4.2) 如果条件成立，我想通过 $inc 更新文档中的字段 "myDataInt" ，否则保持原样：

db.mycoll.update(
    {"_id" : "5e9e5da03da783817d231dc4"},
    {"$inc" : {
       "my_data_sum" : {
           "$cond" : [
               {
                  "$ne" : ["snapshot_time", new_snapshot_time)]
               },myDataInt, 0]
           ]
       }
    },
    {upsert=True, multi=False}
)

然而，这在 pymongo 中给出了一个错误：

raise WriteError(error.get("errmsg"), error.get("code"), error)
pymongo.errors.WriteError: The dollar ($) prefixed field '$cond' in 'my_data_sum.$cond' is not valid for storage.

在这种情况下，有什么办法可以避免在更新前使用 find() 吗？

更新：

如果我使用 Joe 提到的方法，PyMongo (v3.10.1) 将引发异常，因为在 update_many() 中使用 'list' 而不是 'dict':

from pymongo import MongoClient

db = MongoClient()['mydb']

db.mycoll.update_many(
    {"_id" : "5e9e5da03da783817d231dc4"},
    [{"$set" : {
       "my_data_sum" : {
           "$sum": [
               "$my_data_sum",
               {"$cond" : [
                   {"$ne" : ["snapshot_time", new_snapshot_time]},
                   myDataInt, 
                   0
               ]}
           ]
       }
    }}],

    upsert:true
)

最终出现此错误：

  File "/usr/local/lib64/python3.6/site-packages/pymongo/collection.py", line 1076, in update_many session=session),
  File "/usr/local/lib64/python3.6/site-packages/pymongo/collection.py", line 856, in _update_retryable _update, session)
  File "/usr/local/lib64/python3.6/site-packages/pymongo/mongo_client.py", line 1491, in _retryable_write return self._retry_with_session(retryable, func, s, None)
  File "/usr/local/lib64/python3.6/site-packages/pymongo/mongo_client.py", line 1384, in _retry_with_session return func(session, sock_info, retryable)
  File "/usr/local/lib64/python3.6/site-packages/pymongo/collection.py", line 852, in _update retryable_write=retryable_write)
  File "/usr/local/lib64/python3.6/site-packages/pymongo/collection.py", line 823, in _update _check_write_command_response(result)
  File "/usr/local/lib64/python3.6/site-packages/pymongo/helpers.py", line 221, in _check_write_command_response _raise_last_write_error(write_errors)
  File "/usr/local/lib64/python3.6/site-packages/pymongo/helpers.py", line 203, in _raise_last_write_error raise WriteError(error.get("errmsg"), error.get("code"), error)
pymongo.errors.WriteError: Modifiers operate on fields but we found type array instead. For example: {$mod: {<field>: ...}} not {$set: [ { $set: { my_data_sum: { $sum: [ "$my_data_sum", { $cond: [ { $ne: [ "$snapshot_time", 1586910283 ] }, 1073741824, 0 ] } ] } } } ]}

Answer 1

如果您使用的是 MongoDB 4.2，则可以将聚合运算符用于更新。 $inc 不是聚合运算符，但 $sum 是。要指定管道，将数组作为第二个参数传递给 update:

db.coll.update(
    {"_id" : "5e9e5da03da783817d231dc4"},
    [{"$set" : {
       "my_data_sum" : {
           "$sum": [
               "$my_data_sum",
               {"$cond" : [
                   {"$ne" : ["snapshot_time", new_snapshot_time]},
                   myDataInt, 
                   0
               ]}
           ]
       }
    }}],
    {upsert:true, multi:false}
)

Answer 2

花了点时间上网查了一下，发现PyMongo中Collection对象的update_many()、update_one()、update()方法不接受类型列表作为参数支持MongoDB 4.2+ 中更新操作的新聚合管道功能。（至少这个选项在 PyMongo v3.10 中还不可用。）

但是，看起来我可以在 PyMongo 中使用数据库对象的 command 方法，它是 (MongoDB runCommand) 的一个实例，它对我来说工作得很好：

from pymongo import MongoClient

db = MongoClient()['mydb']

result = db.command(
    {
        "update" : "mycoll",
        "updates" : [{
            "q" : {"_id" : "5e9e5da03da783817d231dc4"},
            "u" : [
                {"$set" : {
                   "my_data_sum" : {
                       "$sum": [
                           "$my_data_sum",
                           {"$cond" : [
                               {"$ne" : ["snapshot_time", new_snapshot_time]},
                               myDataInt,
                               0
                           ]}
                       ]
                   }
                }}
            ],
            "upsert" : True,
            "multi" : True
        }],
        "ordered": False
    }
)

数据库对象的command方法获取所有需要的命令的dict对象作为它的第一个参数，然后Aggregation Pipeline的列表可以包含在dict对象中（q是更新查询，u定义了要更新的字段。

result 是来自 MongoDB 的 Ack 消息字典，其中包含 'nModified'、'upserted' 、'writeErrors'.

Answer 3

https://mongoplayground.net/p/1AklFKuhFi6

[
  {
    "id": 1,
    "like": 3
  },
  {
    "id": 2,
    "like": 1
  }
]

让值 = 1，

如果你想递增然后使用

值 = -1 * 值

db.collection.aggregate([
  {
    "$match": {
      "id": 1
    }
  },
  {
    "$set": {
      "count": {
        $cond: {
          if: {
            $gt: [
              "$like",
              0
            ]
          },
          then: {
            "$subtract": [
              "$like",
               value
            ]
          },
          else: 0
        }
      }
    }
  }
])

MongoDB 将 $cond 与更新 ($inc) 结合使用

MongoDB using $cond with Update ($inc)

python

mongodb

pymongo

mongodb-update