MongoDB无需循环批量更新文档 实现标签计数自增及缺失自动插入

你要实现的批量标签计数+不存在自动创建的需求，可以通过MongoDB的bulkWrite批量操作接口实现，全程仅需1次数据库请求，无需提前读库，性能远高于循环调用update_one。 #

核心原理 #

MongoDB的update_many的upsert能力有局限：仅当查询条件没有匹配到任何文档时，才会插入1条新文档，无法为$in数组中每个未匹配到的标签单独生成新文档。而bulkWrite可以把多个单文档更新操作打包成单次请求发送给数据库，同时保留每个操作独立的upsert逻辑。
另外有个特性可以直接利用：当upsert设为True时，新创建的文档会自动包含查询条件里的字段和对应值，不需要手动写$set指定tag字段。 #

实现代码 #

from pymongo import UpdateOne

# 构造批量操作列表
operations = [
    UpdateOne(
        filter={"tag": tag},
        update={"$inc": {"num": 1}},
        upsert=True
    ) for tag in json["tags"]
]

# 单次请求执行所有操作
tags_collection.bulk_write(operations)

逻辑说明 #

• 已存在的标签：匹配到对应文档，直接执行num字段自增1
• 不存在的标签：触发upsert自动创建新文档，tag字段自动填充为查询时的标签值，num字段因为$inc操作会被初始化为1，完全符合需求

注意：如果你的json["tags"]可能存在重复标签，且要求重复标签仅计数1次，可以先对标签数组去重：

unique_tags = list(set(json["tags"]))
# 再用unique_tags构造operations数组

内容的提问来源于stack exchange，提问作者Tom3652