Как применить $ lookup с условиями в mongodb? - PullRequest
Новая
       35

Как применить $ lookup с условиями в mongodb?

1 голос
/ 24 января 2020

Я использую mongodb базы данных и хочу применить $lookup к 2 collections, но с указанными c условиями. У меня есть один collection с именем Company вот так 

new Schema({
    name: String,
    benefit: String,
    benefitDesc: String,
    company_url: String,
    logoUrl: String,
    coverUrl: String,
    desc: String,
    createdAt: String,
    categoryId: { type: Schema.Types.ObjectId, ref: 'categories' },
})

И еще один collection с именем Referrallinks вот так 

new Schema({
    referral_link: String,
    referral_code: String,
    isLink: Number,
    offer_name: String,
    offer_desc: String,
    user_email: String,
    companyId: { type: Schema.Types.ObjectId, ref: 'companies' },
    addedByAdmin: { type: Boolean, default: true },
    number_of_clicks: Number,
    referral_country: String,
    link_status: String,
    categoryId: { type: Schema.Types.ObjectId, ref: 'categories' },
    number_of_clicks: { type: Number, default: 0 },
    createdAt: String,
    updatedAt: String,
    userId: { type: Schema.Types.ObjectId, ref: 'users' }
})

Теперь, когда я применяю это $lookup

Company.aggregate([{
    $lookup: {
        from: 'referrallinks',
        localField: '_id',
        foreignField: 'companyId',
        as: 'referrals'
    }
}])

Итак, я получаю все компании с referrals как array вот так 

[
  {name : '', benefit : '', benefiDesc : '', referrals : []},
  {name : '', benefit : '', benefiDesc : '', referrals : []},
  {name : '', benefit : '', benefiDesc : '', referrals : []},
  .....
]

Но я хочу, чтобы в каждом массиве company's referrals были только рефералы иметь link_status значение Approved.

** Примечание: ** Я также хочу иметь category объект с company и пробовал эту агрегацию, но она выдает ошибку 

{
  from: "referrallinks",
  let: { company_id: "$_id" },
  pipeline: [{
    $match: {
      $expr: {
       $and: [
        { $eq: [ "$$company_id", "$companyId" ] },
        { $eq: [ "$link_status", "Approved" ] }
       ]
     }
    },
  {
    $lookup : {
    from : "categories",
    let  : {"category_id" : "$categoryId"},
    pipeline : [
      {
        $match : {
          $expr : {
            $eq : ["category_id","$_id"]
          }
        }
      }
    ],
    as : "company"
   }
  }
 }],
 as: "referrals"
}

Как мне этого добиться?

1 Ответ

0 голосов
/ 24 января 2020

Используйте $ lookup с пользовательским конвейером , чтобы указать дополнительное условие фильтрации: 

Company.aggregate([{
    $lookup: {
        from: "referrallinks",
        let: { company_id: "$_id" },
        pipeline: [{
            $match: {
                $expr: {
                    $and: [
                        { $eq: [ "$$company_id", "$companyId" ] },
                        { $eq: [ "$link_status", "Approved" ] }
                    ]
                }
            }
        }],
        as: "referrals"
    }
}])

EDIT: для запуска вложенного $lookup ваш код должен выглядеть следующим образом: 

Company.aggregate([{
    $lookup: {
        from: "referrallinks",
        let: { company_id: "$_id" },
        pipeline: [
            {
                $match: {
                    $expr: {
                        $and: [
                            { $eq: [ "$$company_id", "$companyId" ] },
                            { $eq: [ "$link_status", "Approved" ] }
                        ]
                    }
                }
            },
            {
                $lookup: {
                    from: "categories",
                    localField: "categoryId",
                    foreignField: "_id",
                    as: "company"
                }
            }
        ],
        as: "referrals"
    }
}])
...