MongoDB聚合：$set

聚合$set阶段可以为文档添加新的字段。$set输出的文档包含输入文档中的所有现有字段和新添加的字段。$set是$addFields的别名，从MongoDB4.2开始支持。$set和$addFields等价于$project阶段，这两个阶段都等同于 $project 阶段，后者明确指定输入文档中的所有现有字段并添加新字段。

语法

{ $set: { <newField>: <expression>, ... } }

指定要添加的每个字段的名称，并将其值设置为聚合表达式或空对象。

如果新字段的名称与现有字段的名称（包括 _id）相同，$set将用指定表达式的值覆盖该字段的现有值。

使用

• 将新字段追加到现有文档中。可以在聚合操作中包含一个或多个$set阶段。
• 接受对象嵌入，您可以将值设置为聚合表达式或空对象。例如，可接受嵌套对象：{$set：{ a:{ b: { }}}}
• 要在嵌入文档（包括数组中的文档）中添加一个或多个字段，可以使用点符号。

例子

使用两个$set阶段

创建一个scores集合，并添加文档：

db.scores.insertMany([
   { _id: 1, student: "Maya", homework: [ 10, 5, 10 ], quiz: [ 10, 8 ], extraCredit: 0 },
   { _id: 2, student: "Ryan", homework: [ 5, 6, 5 ], quiz: [ 8, 8 ], extraCredit: 8 }
])

以下操作使用了两个$set阶段，在输出文档中加入三个新字段：

db.scores.aggregate( [
   {
     $set: {
        totalHomework: { $sum: "$homework" },
        totalQuiz: { $sum: "$quiz" }
     }
   },
   {
     $set: {
        totalScore: { $add: [ "$totalHomework", "$totalQuiz", "$extraCredit" ] } }
   }
] )

操作返回下面的文档：

{
  "_id" : 1,
  "student" : "Maya",
  "homework" : [ 10, 5, 10 ],
  "quiz" : [ 10, 8 ],
  "extraCredit" : 0,
  "totalHomework" : 25,
  "totalQuiz" : 18,
  "totalScore" : 43
}
{
  "_id" : 2,
  "student" : "Ryan",
  "homework" : [ 5, 6, 5 ],
  "quiz" : [ 8, 8 ],
  "extraCredit" : 8,
  "totalHomework" : 16,
  "totalQuiz" : 16,
  "totalScore" : 40
}

为内嵌文档增加字段

使用点符号为嵌入式文档添加新字段。

创建vehicles集合并添加如下内容：

db.vehicles.insertMany([
   { _id: 1, type: "car", specs: { doors: 4, wheels: 4 } },
   { _id: 2, type: "motorcycle", specs: { doors: 0, wheels: 2 } },
   { _id: 3, type: "jet ski" }
])

下面的聚合操作在嵌入文档specs中添加一个新字段 fuel_type：

db.vehicles.aggregate( [
   { $set: { "specs.fuel_type": "unleaded" } }
] )

操作返回以下结果：

{ _id: 1, type: "car", specs: { doors: 4, wheels: 4, fuel_type: "unleaded" } }
{ _id: 2, type: "motorcycle", specs: { doors: 0, wheels: 2, fuel_type: "unleaded" } }
{ _id: 3, type: "jet ski", specs: { fuel_type: "unleaded" } }

覆盖已有字段

在$set操作中指定一个已存在的字段，原有字段会被替换。创建一个animals集合：

db.animals.insertOne( { _id: 1, dogs: 10, cats: 15 } )

下面的$set聚合操作覆盖了cats字段：

db.animals.aggregate( [
  { $set: { "cats": 20 } }
] )

操作返回以下文件：

{ _id: 1, dogs: 10, cats: 20 }

可以用一个字段替换另一个字段。在下面的示例中，item字段替代了_id字段。创建一个名为fruits的样本集合，其中包含以下文档：

db.fruits.insertMany([
   { "_id" : 1, "item" : "tangerine", "type" : "citrus" },
   { "_id" : 2, "item" : "lemon", "type" : "citrus" },
   { "_id" : 3, "item" : "grapefruit", "type" : "citrus" }
])

下面的聚合操作使用$set将每个文档的_id字段替换为item字段的值，并将item字段替换为字符串"fruit"。

db.fruits.aggregate( [
  { $set: { _id : "$item", item: "fruit" } }
] )

该操作的返回值如下：

{ "_id" : "tangerine", "item" : "fruit", "type" : "citrus" }
{ "_id" : "lemon", "item" : "fruit", "type" : "citrus" }
{ "_id" : "grapefruit", "item" : "fruit", "type" : "citrus" }

向数组中添加元素

创建scores集合，内容如下：

db.scores.insertMany([
   { _id: 1, student: "Maya", homework: [ 10, 5, 10 ], quiz: [ 10, 8 ], extraCredit: 0 },
   { _id: 2, student: "Ryan", homework: [ 5, 6, 5 ], quiz: [ 8, 8 ], extraCredit: 8 }
])

可以使用$set与$concatArrays表达式配合使用，可向现有数组字段添加元素。例如，以下操作使用$set将homework字段替换为一个新数组，新数组的元素是homework数组与数组[7]元素合并后的结果。

db.scores.aggregate([
   { $match: { _id: 1 } },
   { $set: { homework: { $concatArrays: [ "$homework", [ 7 ] ] } } }
])

该操作的返回值如下：

{ "_id" : 1, "student" : "Maya", "homework" : [ 10, 5, 10, 7 ], "quiz" : [ 10, 8 ], "extraCredit" : 0 }

用现有字段重设新字段

创建scores集合，内容如下：

db.scores.insertMany([
   { _id: 1, student: "Maya", homework: [ 10, 5, 10 ], quiz: [ 10, 8 ], extraCredit: 0 },
   { _id: 2, student: "Ryan", homework: [ 5, 6, 5 ], quiz: [ 8, 8 ], extraCredit: 8 }
])

下面的聚合操作为每个文档添加了一个新字段quizAverage，其中包含quiz数组的平均值。

db.scores.aggregate( [
   {
      $set: {
         quizAverage: { $avg: "$quiz" }
      }
   }
] )

db.scores.aggregate( [
   {
      $set: {
         quizAverage: { $avg: "$quiz" }
      }
   }
] )

该操作返回如下文档：

[
   {
      _id: 1,
      student: 'Maya',
      homework: [ 10, 5, 10 ],
      quiz: [ 10, 8 ],
      extraCredit: 0,
      quizAverage: 9
   },
   {
      _id: 2,
      student: 'Ryan',
      homework: [ 5, 6, 5 ],
      quiz: [ 8, 8 ],
      extraCredit: 8,
      quizAverage: 8
   }
]