MongoDB：不使用unwind扁平化嵌套数组适配updateMany问题排查

问题背景 #

维护的日英词典数据库中，多数文档包含r_ele.re_pri字段，因之前updateMany操作失误，该字段出现两种异常：部分为正常字符串数组，部分为长度为1的嵌套数组。尝试用聚合管道处理但未得到预期结果，具体情况如下：

原始文档示例 #

{"r_ele": [{"re_pri": ["ichi1","news1","nf10"]}]},
{"r_ele": [{"re_pri": [["ichi1"]]}]}

尝试的聚合管道 #

[
  {"$match": {"r_ele.re_pri": {"$exists": 1}}},
  {"$project": {
    "r_ele.re_pri": 1,
    "a": {
      "$reduce": {
        "input": "$r_ele.re_pri",
        "initialValue": [],
        "in": {
          "$cond": {
            "if": {"$and": [{"$eq": [{"$type": "$$this"},"array"]},{"$eq": [{"$size": "$$this"},1]}]},
            "then": {"$concatArrays": "$$this"},
            "else": {"$concatArrays": ["$$this","$$value"]}
          }
        }
      }
    }
  }},
  {"$unwind": "$a"},
  {"$set": {"r_ele.re_pri": "$a"}},
  {"$unset": "a"}
]

实际输出 #

{"a": ["ichi1","news1","nf10"]},
{"a": [["ichi1"]]}

预期输出 #

{"a": ["ichi1","news1","nf10"]},
{"a": ["ichi1"]}

管道错误分析 #

1. $reduce输入对象错误：$r_ele.re_pri返回的是嵌套数组（比如第二个文档中是[[["ichi1"]]]），而非re_pri字段本身的数组。需要先遍历r_ele数组，再处理每个元素的re_pri字段。
2. 扁平化逻辑不彻底：当re_pri是长度为1的嵌套数组时，$concatArrays: $$this仅合并数组元素，但$$this本身是[["ichi1"]]，合并后仍保留嵌套结构，未真正提取内层数组元素。

修改后的聚合管道（适配updateMany） #

直接用管道式更新实现扁平化，遍历r_ele数组并处理每个re_pri：

db.collection.updateMany(
  {"r_ele.re_pri": {"$exists": 1}},
  [
    {"$set": {
      "r_ele": {
        "$map": {
          "input": "$r_ele",
          "as": "ele",
          "in": {
            "$mergeObjects": [
              "$$ele",
              {
                "re_pri": {
                  "$cond": {
                    // 判断re_pri的第一个元素是否为数组（识别嵌套结构）
                    "if": {"$eq": [{"$type": {"$arrayElemAt": ["$$ele.re_pri", 0]}}, "array"]},
                    "then": {"$arrayElemAt": ["$$ele.re_pri", 0]},
                    "else": "$$ele.re_pri"
                  }
                }
              }
            ]
          }
        }
      }
    }}
  ]
)

逻辑说明 #

• $map遍历r_ele数组：针对每个r_ele元素，保留原有字段的同时替换re_pri。
• 嵌套数组判断：通过$arrayElemAt取re_pri的第一个元素，判断其类型是否为数组，以此识别嵌套结构。
• 扁平化处理：如果是嵌套数组，直接提取内层数组；如果是正常数组，保持原样。

通用扁平化方案（支持多层嵌套） #

如果存在更深层级的嵌套（比如[[["ichi1"]]]），可使用递归逻辑处理：

db.collection.updateMany(
  {"r_ele.re_pri": {"$exists": 1}},
  [
    {"$set": {
      "r_ele": {
        "$map": {
          "input": "$r_ele",
          "as": "ele",
          "in": {
            "$mergeObjects": [
              "$$ele",
              {
                "re_pri": {
                  "$function": {
                    "body": function flatten(arr) {
                      return arr.reduce((acc, val) => acc.concat(Array.isArray(val) ? flatten(val) : val), []);
                    },
                    "args": ["$$ele.re_pri"],
                    "lang": "js"
                  }
                }
              }
            ]
          }
        }
      }
    }}
  ]
)

该方案通过JavaScript函数递归展开所有嵌套数组，适用于任意深度的嵌套场景。

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