Genel olarak Scala koleksiyonlarını genişletme

Question

Seq[(A, B)] var. Bu tür koleksiyonlara örtülü bir yöntem eklemek istedim, böylece Map[A, Seq[B]]'u geri almak için .toMultiMap yapabilirim.

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

Ama şimdi sorun hep değerler için Traversable geri almak geçerli:

Bu

benim ilk girişimdi. Set[(A, B)].toMultiMap yaparsam Map[A, Set[B]]'u geri almak istiyorum. Çalışmazsa,

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)) 
    } 

Ama:

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

Bunu yapmanın yolu nedir

Yani, o zaman böyle bir şey denedi?

Answer 1

Bunun peşinde olduğun şey olabileceğini düşünüyorum.

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]) 
} 

Bu birkaç basit testler geçer:

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))