简体   繁体   中英

Using aggregate $lookup and $mergeObjects

I want to join collection. before, I used only lookup, so that I could get separated field that is joined. but I need to get result similar mysql join. I noticed there is $lookup and $mergeObjects for this action but not working well.

user collection model.

{
    "_id": ObjectId("xxxxxxx"), //this is default id from mongoDB
    "name": 'admin user',
    "email": 'admin@test.com',
    "password": 'xxxxxxxx',
    "roles": [
        {
            "id": 0,
            "approved": true
        },{
            "id": 2,
            "approved": true
        }
    ]
},{
    "_id": ObjectId("xxxxxxx"), //this is default id from mongoDB
    "name": 'userOne',
    "email": 'user@test.com',
    "password": 'xxxxxxxx',
    "roles": [
        {
            "id": 1,
            "approved": true
        }
    ]
}

roles collection model.

{
    "_id": ObjectId("xxxxxxx"), //this is default id from mongoDB
    "id": '0',
    "name": 'administrator'
},{
    "_id": ObjectId("xxxxxxx"), //this is default id from mongoDB
    "id": '0',
    "name": 'employeer'
},{
    "_id": ObjectId("xxxxxxx"), //this is default id from mongoDB
    "id": '0',
    "name": 'freelancer'
} 

after join, I want to get result like below.

{
    "_id": ObjectId("xxxxxxx"), //this is default id from mongoDB
    "name": 'admin user',
    "email": 'admin@test.com',
    "password": 'xxxxxxxx',
    "roles": [
        {
            "id": 0,
            "name": "administrator",    //join result
            "approved": true
        },{
            "id": 2,
            "name": "freelancer",       //join result
            "approved": true
        }
    ]
},{
    "_id": ObjectId("xxxxxxx"), //this is default id from mongoDB
    "name": 'userOne',
    "email": 'user@test.com',
    "password": 'xxxxxxxx',
    "roles": [
        {
            "id": 1,
            "name": "employeer",        //join result
            "approved": true
        }
    ]
}

You can use below aggregation with mongodb 3.4

You need to $unwind the roles array first and then $group to rollback again

db.users.aggregate([
  { "$unwind": "$roles" },
  { "$lookup": {
    "from": "roles",
    "localField": "roles.id",
    "foreignField": "id",
    "as": "roles.role"
  }},
  { "$unwind": "$roles.role" },
  { "$addFields": {
    "roles": { "$mergeObjects": ["$roles.role", "$roles"] }
  }},
  { "$group": {
    "_id": "$_id",
    "email": { "$first": "$email" },
    "password": { "$first": "$password" },
    "roles": { "$push": "$roles" }
  }},
  { "$project": { "roles.role": 0 }}
])

Which is quite simple with the mongodb 3.6 and above

db.users.aggregate([
  { "$unwind": "$roles" },
  { "$lookup": {
    "from": "roles",
    "let": { "roleId": "$roles.id", "approved": "$roles.approved" },
    "pipeline": [
      { "$match": { "$expr": { "$eq": ["$id", "$$roleId"] }}},
      { "$addFields": { "approved": "$$approved" }}
    ],
    "as": "roles"
  }},
  { "$unwind": "$roles" },
  { "$group": {
    "_id": "$_id",
    "email": { "$first": "$email" },
    "password": { "$first": "$password" },
    "roles": { "$push": "$roles" }
  }}
])

Both will give you similar Output

[
  {
    "_id": ObjectId("5a934e000102030405000004"),
    "email": "user@test.com",
    "password": "xxxxxxxx",
    "roles": [
      {
        "_id": ObjectId("5a934e000102030405000001"),
        "approved": true,
        "id": 1,
        "name": "employeer"
      }
    ]
  },
  {
    "_id": ObjectId("5a934e000102030405000003"),
    "email": "admin@test.com",
    "password": "xxxxxxxx",
    "roles": [
      {
        "_id": ObjectId("5a934e000102030405000000"),
        "approved": true,
        "id": 0,
        "name": "administrator"
      },
      {
        "_id": ObjectId("5a934e000102030405000002"),
        "approved": true,
        "id": 2,
        "name": "freelancer"
      }
    ]
  }
]

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM