MongoDB不同结构集合中不同嵌套层级指定键的重命名方案问询

原生MongoDB实现（推荐，适配大数据量场景） #

要求MongoDB版本≥4.2，支持服务端JavaScript（默认开启），可无视嵌套层级直接批量处理：

固定键名替换（示例：sound→noise） #

针对单个集合执行如下更新，多集合只需批量遍历集合列表即可：

// 定义全局规则
const renameMap = {
  "sound": "noise"
  // 可加多个要替换的键值对
}
const targetCollections = ["Animals", "Events"]

// 递归重命名函数
const renameRecursive = (obj, rule) => {
  Object.keys(obj).forEach(key => {
    if(rule[key]) {
      obj[rule[key]] = obj[key]
      delete obj[key]
      key = rule[key]
    }
    if(typeof obj[key] === 'object' && obj[key] !== null) {
      renameRecursive(obj[key], rule)
    }
  })
  return obj
}

// 批量执行所有集合更新
targetCollections.forEach(collName => {
  db[collName].updateMany(
    {}, // 可添加过滤条件只处理符合要求的文档
    [
      {
        $replaceWith: {
          $function: {
            body: renameRecursive,
            args: ["$$ROOT", renameMap],
            lang: "js"
          }
        }
      }
    ]
  )
})

正则匹配键名替换 #

如果需要匹配符合正则规则的键（比如所有sound_开头的键替换为noise_开头），修改递归函数逻辑即可：

const renameRecursiveByRegex = (obj, regex, replaceStr) => {
  Object.keys(obj).forEach(key => {
    if(regex.test(key)) {
      const newKey = key.replace(regex, replaceStr)
      obj[newKey] = obj[key]
      delete obj[key]
      key = newKey
    }
    if(typeof obj[key] === 'object' && obj[key] !== null) {
      renameRecursiveByRegex(obj[key], regex, replaceStr)
    }
  })
  return obj
}

// 调用示例：把所有sound开头的键替换为noise开头
db.Events.updateMany(
  {},
  [
    {
      $replaceWith: {
        $function: {
          body: renameRecursiveByRegex,
          args: ["$$ROOT", /^sound/, "noise"],
          lang: "js"
        }
      }
    }
  ]
)

PyMongo实现（适配小数据量、禁止开启服务端JS场景） #

直接在客户端遍历文档处理，逻辑和原生实现完全对齐：

from pymongo import MongoClient

# 连接数据库
client = MongoClient("你的MongoDB连接串")
db = client["你的数据库名"]

# 配置规则
RENAME_MAP = {"sound": "noise"}
TARGET_COLLECTIONS = ["Animals", "Events"]

def rename_key_recursive(doc: dict, rename_rule: dict) -> dict:
    for key in list(doc.keys()):
        # 匹配重命名规则
        if key in rename_rule:
            doc[rename_rule[key]] = doc.pop(key)
            key = rename_rule[key]
        # 递归处理嵌套对象
        if isinstance(doc[key], dict):
            rename_key_recursive(doc[key], rename_rule)
        # 兼容处理数组中的嵌套对象
        elif isinstance(doc[key], list):
            for item in doc[key]:
                if isinstance(item, dict):
                    rename_key_recursive(item, rename_rule)
    return doc

# 批量执行
for coll_name in TARGET_COLLECTIONS:
    coll = db[coll_name]
    for doc in coll.find():
        updated_doc = rename_key_recursive(doc, RENAME_MAP)
        coll.replace_one({"_id": doc["_id"]}, updated_doc)

注意事项 #

• 操作前务必全量备份数据，避免误操作导致数据丢失
• 数据量超过100W时优先使用原生MongoDB方案，性能远高于客户端遍历
• 如需处理数组中嵌套的对象，两种实现都已经默认兼容该场景

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