Сортировка и объединение недавно вычисленного поля массива с использованием конвейера агрегации - PullRequest
2 голосов
/ 25 июня 2019

У меня есть коллекция пользователей и коллекция ролей. Роли и роли пользователя has_many хранятся в виде массива идентификаторов в пользовательском документе. Теперь я хочу отсортировать пользователей по значению их ролей. Пользователь может иметь одну или несколько ролей. Если у пользователя более одной роли, я хочу, чтобы они были разделены запятой:

'admin'
admin, marketing'
'user, marketing'

И это именно те данные, которые я хочу отсортировать по буквенно-цифровым символам, например, сортировка по убыванию:

'user, marketing'
'admin, marketing'
'admin'

Вот что я пробовал:

db.users.aggregate([
  { $lookup : { from: 'roles', localField: 'role_ids', foreignField: '_id', as: 'roles' } },
])

Дает мне:

{"_id": ObjectId ("5d0ac05c3ed149604a000003"), "электронная почта": "me@gmail.com", "name": "Administrator", "role_ids": [ ObjectId ( "5d0ac01f3ed149604a000000"), ObjectId ("cd0ab01f3ed129603a000550")], "role": [{"_id": ObjectId ("5d0ac01f3ed149604a000000"), "name": "admin")}, {"_id": ObjectId ("cd0ab01f3ed129603a000550"), "name": "marketing")}]}

Затем я решил использовать $ project для добавления только что вычисленного каскадного поля имен ролей:

db.users.aggregate([
  { $lookup : { from: 'roles', localField: 'role_ids', foreignField: '_id', as: 'roles' } },
  { $project : {_id: 1, email: 1, name: 1, joinedRoles: { $concat: [] }  } }
  { $sort : { joinedRoles: -1 } }
])

Именно эту часть я пытаюсь выяснить:

joinedRoles: { $concat: [] }

1 Ответ

1 голос
/ 25 июня 2019

К сожалению, вы пытаетесь сделать две основные вещи, которые Монго в настоящее время очень плохо делает:

  • сортировка массива
  • объединение массива в виде строки

Тем не менее, мы можем применить несколько этапов, таких как:

// { id: 1, roles: ["marketing", "admin", "user"], b: 42 }
// { id: 2, roles: ["admin"], b: 11 }
// { id: 3, roles: [], b: 1 }
db.collection.aggregate([
  // First thing, let's handle empty arrays by replacing them with
  // an array containing `""`. This way, the next stage (`$unwind`)
  // will not get rid of these empty arrays. This might not be needed
  // if you know you'll always have a role for a user. `roles: []`
  // becomes `roles: [""]` other arrays are left untouched:
  { $addFields: {
    roles: { $cond: {
      if: { $eq: [ { $size: "$roles" }, 0 ] }, then: [""], else: "$roles"
    }}
  }},
  // Transforms each document in as many documents as there are items in
  // `roles`. For instance, `{ id: 1, roles: ["marketing", "admin", "user"], b: 42 }`
  // becomes these 3 documents: `{ id: 1, roles: "marketing", b: 42 }`,
  // `{ id: 1, roles: "admin", b: 42 }` and `{ id: 1, roles: "user", b: 42 }`
  // That's also why we handled empty arrays in the previous stage:
  { $unwind: "$roles" },
  // Now we can alphabetically sort all documents on the `roles` field:
  { $sort: { roles: 1 } },
  // And we can transform back roles to arrays by grouping on `id` using
  // `$push` within a `$group` stage. Since this preserves the order, the
  // `roles` array is now sorted (due to the `$sort` stage applied
  // just before on unwind elements). Note that you have to specify all
  // other fields in your documents that you want to keep using `$first`.
  // At this point you'll get `{ _id: 1, roles: ["admin", "marketing", "user"], b: 42 }`:
  { $group: { _id: "$id", roles: { $push: "$roles" }, b: { $first: "$b" } } },
  // Finally, we can join an array of strings as a string by coupling
  // `$reduce` and `$substr` operations:
  { $addFields: { roles: {
      // The `$substr`ing handles results of `$reduce` such as
      // `", admin, marketing"` by removing the first 2 characters:
      $substr: [
        // `$reduce` joins elements by applying `$concat` as many times as
        // their are items in the array:
        { $reduce: {
          input: "$roles",
          initialValue: "",
          in: { $concat: ["$$value", ", ", "$$this"] }
        }},
        2,
        -1
      ]
  }}},
  // And, you can obviously sort the resulting documents on this field:
  { $sort: { roles: -1 } }
])
// { _id: 3, roles: "", b: 1 }
// { _id: 2, roles: "admin", b: 11 }
// { _id: 1, roles: "admin, marketing, user", b: 42 }

Обратите внимание, что я немного упростил вашу проблему, используя чистый массив ролей для ввода. Чтобы добраться до этой точки, вы можете сначала применить преобразование $map, например:

// { roles: [{ id: 1, role: "admin" }, { id: 2, role: "marketing" }] }
db.collection.aggregate({ $addFields: { roles: { $map: { input: "$roles", as: "role", in: "$$role.role" } } } })
// { roles: [ "admin", "marketing" ] }
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...