MongoDb aggregation framework to group elements of inner array

Question

I'm using MongoDB aggregation framework trying to transform each document:

{
  "all": [
    {
      "type": "A",
      "id": "1"
    },
    {
      "type": "A",
      "id": "1"
    },
    {
      "type": "B",
      "id": "2"
    },
    {
      "type": "A",
      "id": "3"
    }
  ]
}

into this:

{
  "unique_type_A": [ "3", "1" ]
}

(final result is a collection of n documents with unique_type_A field)

The calculation consists of returning in an array all the uniques types of entities of type A.

I got stuck with $group step, anyone knows how to do it?

bvrakvs · Accepted Answer · 2019-10-11 00:02:43Z

2

To apply this logic to each document, you can use the following;

db.collection.aggregate([
  {
    $unwind: "$all"
  },
  {
    $match: {
      "all.type": "A"
    }
  },
  {
    $group: {
      _id: {
        "type": "$all.type",
        "oldId": "$_id"
      },
      unique_type_A: {
        $addToSet: "$all.id"
      }
    }
  },
  {
    $project: {
      _id: 0
    }
  }
])

Where we first $unwind, to be able to filter and play with each member of all array. Then we just filter the non type:"A" members. The $group stage has the difference with a complex _id, where we utilize the _id of $unwind result, which refers back to the original document, so that we can group the results per original document. Collecting the id from all array with $addToSet to keep only unique values, and voilà!

And here is the result per document;

[
  {
    "unique_type_A": [
      "3",
      "1"
    ]
  },
  {
    "unique_type_A": [
      "4",
      "11",
      "5"
    ]
  }
]

Check the code interactively on Mongoplayground

edited Oct 11, 2019 at 0:02

answered Oct 10, 2019 at 23:36

bvrakvs

14.7k4 gold badges55 silver badges96 bronze badges

Sign up to request clarification or add additional context in comments.

9 Comments

italktothewind Over a year ago

I edited the question because I think I was not clear enough. I'm not expecting a single final aggregation, but to transform each document (final result is n documents). BTW that online tool is just great.

bvrakvs Over a year ago

@italktothewind updated the answer =) yeah I know it is pretty great online tool!

italktothewind Over a year ago

Thanks for your answer! I though that maybe we can use something to apply grouping inside the document (without splitting it and grouping it again)

bvrakvs Over a year ago

@italktothewind there are always alternatives, but it will become an hard-to-maintain code, I prefer the readable pipeline unless there is a performance issue etc.

italktothewind Over a year ago

In fact there is a performance issue, I'm applying this to 100 million documents haha

|

Collectives™ on Stack Overflow

MongoDb aggregation framework to group elements of inner array

1 Answer 1

9 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

9 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related