如何在MongoDB中为字典字段添加最小可用整数键

可以实现，MongoDB 4.2及以上版本支持聚合管道式更新，不需要把文档读取到客户端处理，全程在数据库侧原子完成，比先查后写的方案并发安全性更高。

核心实现思路 #

和你提供的Python逻辑完全一致：

1. 提取key字段下的所有键，转为整数数组
2. 生成从0到最大现有键+1的连续整数序列
3. 取两个数组的差集，第一个元素就是最小未使用的整数键
4. 将新的键值对合并到原key对象中

具体操作代码 #

Mongo Shell 写法 #

db.你的集合名.updateOne(
  // 替换为你要更新的文档匹配条件，比如按_id匹配
  { _id: ObjectId("替换为目标文档的_id") },
  // 聚合更新管道
  [
    // 第一步：提取所有现有键转为整数数组
    {
      $set: {
        exist_keys: {
          $map: {
            input: { $objectToArray: "$key" },
            in: { $toInt: "$$this.k" }
          }
        }
      }
    },
    // 第二步：生成连续整数范围
    {
      $set: {
        range: { $range: [0, { $add: [{ $max: "$exist_keys" }, 2] }] }
      }
    },
    // 第三步：计算最小可用键，兼容key为空的场景
    {
      $set: {
        new_key: {
          $cond: {
            if: { $eq: [{ $size: "$exist_keys" }, 0] },
            then: 0,
            else: { $first: { $setDifference: ["$range", "$exist_keys"] } }
          }
        }
      }
    },
    // 第四步：合并新键值对，删除临时字段
    {
      $set: {
        key: {
          $mergeObjects: [
            "$key",
            { $arrayToObject: [[{ k: { $toString: "$new_key" }, v: "替换为你要插入的自定义值" }]] }
          ]
        },
        exist_keys: "$$REMOVE",
        range: "$$REMOVE",
        new_key: "$$REMOVE"
      }
    }
  ]
)

Python Pymongo 写法 #

如果你用Python操作MongoDB，可以直接用下面的代码，不需要自己读取文档计算：

from pymongo import MongoClient

client = MongoClient("你的MongoDB连接地址")
db = client["你的库名"]
coll = db["你的集合名"]

# 自定义要插入的value
insert_value = 200

coll.update_one(
  {"_id": "替换为目标文档的_id"},
  [
    {
      "$set": {
        "exist_keys": {
          "$map": {
            "input": {"$objectToArray": "$key"},
            "in": {"$toInt": "$$this.k"}
          }
        }
      }
    },
    {
      "$set": {
        "range": {"$range": [0, {"$add": [{"$max": "$exist_keys"}, 2]}]}
      }
    },
    {
      "$set": {
        "new_key": {
          "$cond": {
            "if": {"$eq": [{"$size": "$exist_keys"}, 0]},
            "then": 0,
            "else": {"$first": {"$setDifference": ["$range", "$exist_keys"]}}
          }
        }
      }
    },
    {
      "$set": {
        "key": {
          "$mergeObjects": [
            "$key",
            {"$arrayToObject": [[{"k": {"$toString": "$new_key"}, "v": insert_value}]]}
          ]
        },
        "exist_keys": "$$REMOVE",
        "range": "$$REMOVE",
        "new_key": "$$REMOVE"
      }
    }
  ]
)

注意事项 #

• 低于4.2版本的MongoDB不支持聚合管道更新，只能先把文档读到客户端计算后再写回。
• 高并发写入场景下，建议搭配writeConcern: { w: "majority" }使用，进一步保证数据一致性。
• 如果key字段下的键数量超过1万，会有轻微性能损耗，普通业务场景无需担心。

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