Generisch erweiternden Scala Kollektionen

Question

Ich habe eine Seq[(A, B)]. Ich wollte eine implizite Methode zu solchen Sammlungen hinzufügen, so dass ich .toMultiMap tun kann, um eine Map[A, Seq[B]] zurück zu bekommen.

Dies war mein erster Versuch:

implicit class PairsExtensions[A, B](t: Traversable[(A, B)]) { 
    def toMultiMap: Map[A, Traversable[B]] = t.groupBy(_._1).mapValues(_.map(_._2)) 
    } 

Aber jetzt das Problem ist, ich für die Werte ein Traversable wieder immer. Ich möchte ein Map[A, Set[B]] zurückbekommen, wenn ich Set[(A, B)].toMultiMap mache. So

, dann habe ich versucht, so etwas wie diese:

implicit class PairsExtensions2[A, B, Repr[_] <: Traversable[(A, B)]](t: TraversableLike[(A, B), Repr[(A, B)]]) { 
    def toMultiMap(implicit bf: CanBuild[B, Repr[B]]): Map[A, Repr[B]] = t.groupBy(_._1).mapValues(_.map(_._2)) 
    } 

Aber es funktioniert nicht:

val m1: Map[Int, Set[String]] = Set.empty[(Int, String)] 
val m2: Map[Int, List[String]] = List.empty[(Int, String)] 

Was ist die Art und Weise, dies zu tun?

Answer 1

Ich denke, das könnte sein, wonach Sie suchen.

import scala.collection.generic.CanBuildFrom 
import scala.language.higherKinds 

implicit class PairsExtensions[A, B, C[X] <: Traversable[X]](t: C[(A, B)]) { 
    def toMultiMap(implicit cbf: CanBuildFrom[Nothing, B, C[B]]): Map[A, C[B]] = 
    t.groupBy(_._1).mapValues(_.map(_._2).to[C]) 
} 

Dies geht ein paar einfachen Tests:

scala> val m1: Map[Int, Set[String]] = Set.empty[(Int, String)].toMultiMap 
m1: Map[Int,Set[String]] = Map() 

scala> val m2: Map[Int, List[String]] = List.empty[(Int, String)].toMultiMap 
m2: Map[Int,List[String]] = Map() 

scala> Seq(('c',4),('x',2),('c',5)).toMultiMap 
res4: Map[Char,Seq[Int]] = Map(x -> Vector(2), c -> Vector(4, 5))