如何在Mongoose嵌套Map结构中实现file_name的Upsert操作

针对你这种files为Map类型、需要基于嵌套file_name字段做更新/插入的场景，直接用常规的upsert配置没法直接定位深层字段，得结合Map的特性来处理，下面给两种可行方案：

方案一：分两步操作（逻辑清晰，易维护） #

先尝试更新匹配file_name的文件对象，若未找到则插入新的键值对到files中：

1. 定义操作参数 #

const accountId = "你的目标账户ID";
const targetFileName = "要匹配的文件名";
// 要更新的字段
const updateFields = {
  content: "更新后的文件内容",
  isPublished: true
};
// 新文件的完整数据
const newFile = {
  file_name: targetFileName,
  creation_date: new Date(),
  isPublished: false,
  content: "新文件初始内容"
};
// 生成新的文件ID（对应Map的键）
const newFileId = new mongoose.Types.ObjectId().toString();

2. 执行更新操作 #

用arrayFilters定位到files中匹配file_name的对象，更新指定字段：

const updatedAccount = await Account.findOneAndUpdate(
  {
    _id: accountId,
    "files.*.file_name": targetFileName
  },
  {
    $set: {
      "files.$[entry].content": updateFields.content,
      "files.$[entry].isPublished": updateFields.isPublished
    }
  },
  {
    arrayFilters: [{ "entry.file_name": targetFileName }],
    new: true
  }
);

3. 若未找到则插入新文件 #

如果更新结果为空，说明该file_name不存在，直接往files中添加新的键值对：

if (!updatedAccount) {
  await Account.findOneAndUpdate(
    { _id: accountId },
    {
      $set: {
        [`files.${newFileId}`]: newFile
      }
    },
    { new: true }
  );
}

方案二：聚合管道一次操作（无需两次查询） #

利用MongoDB的聚合管道，在一次操作中完成存在判断、更新或插入：

const result = await Account.findOneAndUpdate(
  { _id: accountId },
  [
    {
      $set: {
        files: {
          $cond: {
            // 判断目标file_name是否存在于当前files中
            if: {
              $in: [
                targetFileName,
                {
                  $map: {
                    input: { $objectToArray: "$files" }, // 把Map转成键值对数组
                    as: "fileItem",
                    in: "$$fileItem.v.file_name" // 提取所有file_name
                  }
                }
              ]
            },
            // 存在则更新对应文件的字段
            then: {
              $arrayToObject: {
                $map: {
                  input: { $objectToArray: "$files" },
                  as: "fileItem",
                  in: {
                    k: "$$fileItem.k",
                    v: {
                      $cond: {
                        if: { $eq: ["$$fileItem.v.file_name", targetFileName] },
                        then: { ...$$fileItem.v, ...updateFields }, // 合并更新字段
                        else: "$$fileItem.v"
                      }
                    }
                  }
                }
              }
            },
            // 不存在则添加新的键值对到files
            else: {
              $mergeObjects: [
                "$files",
                { [newFileId]: newFile }
              ]
            }
          }
        }
      }
    }
  ],
  { new: true }
);

关键注意事项 #

• 如果你需要保证同一个账户下file_name唯一，必须添加复合唯一索引：

  accountSchema.index({ "_id": 1, "files.file_name": 1 }, { unique: true });
  

• 原Schema中files的unique: true配置无效，因为Map类型的unique约束不作用于内部字段，需要用上面的复合索引替代。

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