如何按 mongoose 聚合中的子文档字段过滤？

Question

我想过滤支付订单的特定用户的费用。 这是我的模式：

var PaymentOrderSchema = new Schema({
  name: {type: String, required: true},
  description: {type: String},
  amount: {type: Number},
  charges: [{type: Schema.Types.ObjectId, ref: 'Charge'}],
},

var ChargeSchema = new Schema({
  amount: {type: Number},
  user: {type: Schema.Types.ObjectId, ref: 'User'},
  status: {type: String},
  description: {type: String},
  creation_date:{type: Date}
}

我正在尝试使用聚合，但我无法过滤费用。 当我使用 $ne 而不是 $eq 时，它确实会向我退还费用，所以我知道我走在正确的轨道上。 这是我的实际代码：

const paymentOrder = await PaymentOrder.aggregate([
  {$match: {'_id': mongoose.Types.ObjectId(req.query.payment_order_id)}},
  {$project: {
    charges: {$filter: {
      input: '$charges',
      as: 'charge',
      cond: {$eq: ['$$charge.user._id', mongoose.Types.ObjectId(req.user._id)]}
    }}
  }}
]);

谢谢！

Answer 1

如果我正确理解您的要求，那么我们可以将 $lookup 与管道一起使用

像这样的东西

db.paymentOrder.aggregate([
  {
    $match: {
      _id: ObjectId("5a934e000102030405000003") // replace this hard-coded objectId with mongoose.Types.ObjectId(req.query.payment_order_id)
    }
  },
  {
    $lookup: {
      from: "charge",
      let: {
        chargeIds: "$charges"
      },
      pipeline: [
        {
          $match: {
            $expr: {
              $eq: [
                "$user",
                "userId1" // replace this with mongoose.Types.ObjectId(req.user._id)
              ]
            }
          }
        }
      ],
      as: "charges"
    }
  }
])

检查这个Mongo Plaground

希望能帮助到你

Answer 2

聚合在数据库服务器上运行，而引用由客户端的驱动程序扩展，因此在管道中实际看到的值是

DBRef("User",ObjectId(...),"DatabaseName")

可以访问类似于 object

{"$ref":"User", "$id":ObjectId(...), "$db":"DatabaseName"}

但这又引发了一个问题：字段名称不允许以$开头，所以这将引发错误：

{$eq:["$$charge.user.$id",mongoose.Types.ObjectId(req.user._id)]}

因此，您可以跳过聚合中的该步骤并将文档带回 mongoose 进行填充，或者如果您只需要匹配$id ，您可以使用$objectToArray分解 DBRef 以便您可以匹配：

  {$match: {'_id': mongoose.Types.ObjectId(req.query.payment_order_id)}},
  {$unwind: "$charges"},
  {$addFields: {charges:{$objectToArray:"$charges"}}},
  {$match: {charges:{
         $elemMatch:{k:"$id",v:mongoose.Types.ObjectId(req.user._id}
  }}},
  {$group:{_id:"$_id", charges:{$push:{$arrayToObject("$charges"}}}}}

这将返回可以由驱动程序扩展的 DBRef，但这与charges文档的_id匹配，这可能不是您想要的。

然而，一旦你有了收费文档的_id ，你可以使用$lookup然后过滤它们，但是我们还必须处理作为 DBref 的user ：

  {$match: {'_id': mongoose.Types.ObjectId(req.query.payment_order_id)}},
  {$unwind: "$charges"},
  {$addFields: {chargeId:{$filter:{
         input:$objectToArray:"$charges"
         cond:{$eq:["$$this.k","$id"]}
  }}}}, 
  {$lookup:{
       from:"charge",
       localField:"$chargeId.v",
       foreignField:"$_id",
       as: charges
  }},
  {$unwind:"$charges"},
  {$addField:{chargeUser:{$filter:{
        input:{$objectToArray:"$charges.user"},
        cond:{$eq:["$$this.k","$id"]}
  }}}},
  {$match: {"chargeUser.v":mongoose.Types.ObjectId(req.user._id)}},  
  {$project:{
        chargeUser:0, 
        chargeId:0
  },
  {$group:{
         _id:"$_id", 
         document:{$first:"$$ROOT"}, 
         charges:{$push:"$charges"}
  }},
  {$addFields:{"document.charges":"$charges"}},
  {$replaceRoot:{newRoot:"$document"}}