javascript - Mongo聚合管道更新或推送

Question

我有一个 MongoDB 模型，其中包含作为对象的成员数组。

const guestSchema = new mongoose.Schema({
  salutation: {
    type: String,
  },
  members: [membersSchema],
  user: {
    type: mongoose.Schema.ObjectId,
    ref: 'User',
  },
});

成员架构:

const membersSchema = new mongoose.Schema({
  name: String,
  status: {
    type: String, 
    enum: ['regular', 'helper'],
    default: 'regular',
  },
});

我想实现在给定 ID 的文档存在的情况下进行更新，或者在数组中文档的 ID 不存在的情况下推送到数组。我使用聚合管道，但是我无法实现将新文档推送到数组。为什么我不能像这样使用push after else 语句。

const subDocumentToUpsert = { 'name': mem.name, 'status': mem.status, '_id': ObjectId(mem.id)}
      const subDocumentNoID = { 'name': mem.name, 'status': mem.status}


await Guest.findOneAndUpdate(
        { "_id": req.params.id },
        [
          {
            $set: {
              members: {
                $cond: {
                  if: { $in: [subDocumentToUpsert._id, '$members._id'] },
                  then: {
                    $map: {
                      input: '$members',
                      as: 'sub_document',
                      in: {
                        $cond: {
                          if: { $eq: ['$$sub_document._id', subDocumentToUpsert._id] },
                          then: subDocumentToUpsert,
                          else: '$$sub_document',
                        },
                      },
                    },
                  },
                  else: { 
                        $push: {
                          subDocumentNoID
                        }, 
                      },
                  },
                },
              },
            },
          },
      ]);

最好的方法是什么？谢谢

Answer 1

您可以执行以下操作:

db.collection.update({
 _id: {
  $in: [
    1,
    2
  ]
 }
},
[
{
$set: {
  members: {
    $cond: {
      if: {
        $in: [
          5,
          "$members._id"
        ]
      },
      then: {
        $map: {
          input: "$members",
          as: "sub",
          in: {
            $cond: {
              if: {
                $eq: [
                  "$$sub._id",
                  5
                ]
              },
              then: {
                _id: 5,
                status: "regular_updated",
                name: "Negan_updated"
              },
              else: "$$sub"
            },
            
          },
          
        },
        
      },
      else: {
        $concatArrays: [
          "$members",
          [
            {
              _id: 5,
              status: "regular_upserted",
              name: "Negan_upserted"
            }
          ]
        ]
      }    
    }  
  }
 }
}    
}
],
{
 multi: true
})

解释:

1. 检查子对象中是否存在 _id:5，并通过 $map/$cond 仅更新具有 _id:5 的对象。

2. 如果没有 _id:5，请使用 $concatArrays 将新对象添加到数组中。

Playground