如何根据可应用于其中一个文档的条件插入新文档

Question

我有一个集合，其中存储用户的唯一 uID，并且还必须手动注册它们。注册时，我在数组中传递多个 ID，现在我想查询注册时提供的任何 uID 是否与注册集合中的现有 uID 匹配，我想返回这些文档或只是返回一个真值，该值将告诉用户存在，以便我中止注册并为用户分配另一个 ID。

const uniqueIDs = [ "123", "156", "148"]

如果需要，我可以运行映射函数来获取这些值作为数组对象，例如

[
{
    uID : "123",
},
{
    uID : "156",
},
{
    uID : "148",
},
]

现有集合中的示例：

{
    uID : "123",
    country : "abc"
},
{
    uID : "486",
    country : "def"
},
{
    uID : "958",
    country : "jkl"
},

现在我的问题是我可以从映射数组或 uiD 数组查询我的集合，以便找到现有文档吗？ 如果没有找到文档，我可以通过 insertMany 之类的函数从这个唯一的 uID 插入多个文档吗？

Answer 1

有一个解决方案，尽管我想说这更像是一种黑客行为而不是常见做法。

您可以执行以下操作：

db.collection.aggregate([
  {
    $facet: {
      addition: [
        {$limit: 1},
        {$project: {
            _id: 0,
            data: [
              {
                uID: "333",
                country: "defrr"
              },
              {
                uID: "222",
                country: "jklyy"
              }
            ]
          }
        }
      ],
      found: [
        {$match: {"uID": {$in: ["123", "156", "148"]}}}
      ]
    }
  },
  {
    $project: {
      data: {
        $cond: [{$lt: [{$size: "$found"}, 1]},
          {$arrayElemAt: ["$addition", 0]}, {data:'errorCase'}
        ]
      }
    }
  },
  {$unwind: "$data.data"},
  {$replaceRoot: {newRoot: "$data.data"}
  },
  {
    $merge: {
      into: "collection",
      on: "uID"
    }
  }
])

这是一个聚合管道，末尾有一个 $merge 步骤，以便将新文档插入到集合中。 $facet 步骤构建您想要插入的文档并检查条件。带有 $cond 的 $project 使用数据覆盖它，如果条件找到现有的 uID，该数据将返回错误。

这种黑客攻击有其局限性。首先，在运行之前，您的集合中至少需要有一个文档，因为它使用它作为创建新文档的基础。其次，为了使用 $merge 步骤，您需要在 uID 上创建唯一索引。

最后，如果列表中没有带有 uID 的文档，它将插入新文档。如果列表中存在带有 uID 的文档，则会返回错误。