MongoDB findOneAndUpdate扁平化$set数组索引插入为对象问题

这个问题我之前也碰到过——当用带数字索引的扁平化路径执行upsert时，MongoDB会默认把数字索引当作对象的键，而不是数组索引，所以才会出现priceTier变成{ "0": { ... } }这种对象结构，而不是我们想要的数组。

核心原因 #

MongoDB在创建不存在的嵌套字段时，会从左到右解析路径：当它遇到price.priceTier.0.unitSold，发现priceTier不存在，就会默认创建一个对象，然后把0作为这个对象的属性键，而不是自动创建数组。只有当priceTier已经是数组时，数字索引才会被当作数组下标处理。

解决方案：用$setOnInsert提前初始化数组结构 #

既然我们要保留现有的扁平化查询结构，最贴合需求的办法是在upsert时，通过$setOnInsert提前把目标数组路径初始化为数组（里面包含对应数量的空对象，确保索引有效），这样后续的$set就能正确操作数组元素了。

示例代码

db.test.findOneAndUpdate(
  {"saleId" : "7d55acf0-5bc2-11e8-b3e5-c51df4c55083"},
  {
    // 仅在插入新文档时初始化数组，已有文档不会受影响
    $setOnInsert: { "price.priceTier": [{}] },
    // 保留你原本的扁平化$set操作
    $set: {"price.priceTier.0.unitSold": 1}
  },
  {upsert:true}
)

执行后再查询，就能得到你想要的数组结构：

db.test.find()
// 输出结果
{ 
  "_id" : ObjectId("5b0173fbcd90c934727269ac"), 
  "saleId" : "7d55acf0-5bc2-11e8-b3e5-c51df4c55083", 
  "price" : { "priceTier" : [{ "unitSold" : 1 } ]} 
}

通用代码的适配方案 #

因为你用了flat模块生成查询，需要把这个逻辑整合到通用代码里：

1. 解析所有$set中的扁平化路径，提取出带数字索引的数组路径（比如从price.priceTier.0.unitSold中提取price.priceTier和索引0）
2. 对每个数组路径，找到最大的索引值n，然后生成一个长度为n+1的空对象数组（比如索引是2，就生成[{}, {}, {}]）
3. 将这些数组初始化逻辑添加到$setOnInsert中，和原有的$set组合成最终的更新语句

比如如果你的$set包含多个数组索引路径：

const flatSet = {
  "price.priceTier.0.unitSold": 1,
  "price.priceTier.1.unitPrice": 100
};
// 解析后生成$setOnInsert
const setOnInsert = {
  "price.priceTier": [{}, {}]
};
// 组合查询
db.test.findOneAndUpdate(
  {"saleId": "xxx"},
  { $setOnInsert: setOnInsert, $set: flatSet },
  { upsert: true }
);

注意事项 #

• $setOnInsert只会在插入新文档时生效，已有文档的数组结构不会被修改，所以不用担心影响现有数据
• 如果已有文档的目标字段是错误的对象结构（比如之前插入的priceTier是对象），需要先通过聚合更新把对象转成数组，比如用$objectToArray和$map处理，但这属于数据修复场景，不在通用新增/更新的逻辑里

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