Преобразовать список в карту элемента -> список (элемент) в Scala

Question

У меня есть список документов, владельцем документа является Пользователь.

Какой самый элегантный способ преобразования этого списка в Карту пользователей в список документов, которыми они владеют??

Так, например, у меня есть:

"doc1" owned by user "John"
"doc2" owned by user "Frank"
"doc3" owned by user "John"

Я должен закончить с картой:

"John" -> List("doc1", "doc3"), "Frank" -> List("doc2")

Я могу придумать один способ, который будетзахватить всех уникальных пользователей из документов и для каждого из них отфильтровать список документов, чтобы они были теми, которыми они владеют, но мне интересно, есть ли способ, который использует фиксированное количество проходов по списку, чтобы предотвратить любые проблемы с производительностью, еслисписок большой.

Answer 1

Использовать groupBy:

scala> case class Doc(id: String, owner: String)
defined class Doc

scala> List(Doc("doc1", "John"), Doc("doc2", "Frank"), Doc("doc3", "John"))
res0: List[Doc] = List(Doc(doc1,John), Doc(doc2,Frank), Doc(doc3,John))

scala> res0.groupBy(_.owner)
res1: scala.collection.immutable.Map[String,List[Doc]] = Map(
  Frank -> List(Doc(doc2,Frank)), John -> List(Doc(doc1,John), Doc(doc3,John)))