將列表映射到另一個特徵的地圖的最佳方法

Question

可以說我有一個List[TraitA]的對象。 TraitA提供了一個屬性propX : String。 我知道該列表的一個子集也是TraitB的一個實例，但它不提供屬性propX。

因此，例如：

trait TraitA 
{ 
    def propX : String 
} 

trait TraitB 
{ 
    def abc : Int 
} 

一些列表只是extend TraitA的情況下，而另一些extend TraitA with TraitB。我只需要提取那些具有TraitB的實例，但我需要從TraitA保留屬性propX。 propX只能是幾個值，我需要的是一個Map，它根據此值對TraitB的實例進行分組。

所以我需要提取是從List[TraitA]實例TraitB實例的這個子集，因爲他們中的一些TraitA with TraitB，並創建一個Map[String, List[TraitB]]，這裏的關鍵是從TraitApropX。

我一直在擺弄周圍有for內涵，但由於某種原因，我不能yield元組的List[(String, TraitB)]（這我可以再groupBy _.1），可能是因爲第一發電機TraitA類型。

我想這一點，但它抱怨說，預期的類型是List[(String, TraitA)]：

for { 
traitA <- listOfTraitAs 
traitBoption = (traitA match { 
    case traitB : TraitB => Some(traitB) 
    case _ => None 
    }) 
} yield (traitA.propX, traitBoption) 

在另一方面，如果我通過圖案TraitB匹配filter列表中我失去了過濾功能的propX知名度。

實現此目標的最佳方法是什麼？

Answer 1

我不明白爲什麼filter不會爲你工作：

scala> trait TraitA { def propX: String }; trait TraitB { def abc: Int } 
defined trait TraitA 
defined trait TraitB 

scala> class A extends TraitA { def propX = util.Random.nextInt(5).toString() } 
defined class A 

scala> class B extends A with TraitB { def abc = util.Random.nextInt(5) } 
defined class B 

scala> val xs: List[TraitA] = List.fill(15)(if (util.Random.nextBoolean()) new A else new B) 
xs: List[TraitA] = List(A@6b46e91a, B@7c71e0fb, A@1869be91, B@465e2e1c, B@5125545b, A@69c54bfb, B@17ff81fd, A@7af155a, B@77a2cba6, A@60e83ca6, A@2ee5e7fe, B@77e1ecbf, A@117e2d16, A@72c20852, B@20b07a5a) 

scala> xs collect { case a: TraitB => a } groupBy (_.propX) 
res7: scala.collection.immutable.Map[String,List[TraitA with TraitB]] = Map(4 -> List(B@465e2e1c, B@77e1ecbf), 1 -> List(B@77a2cba6), 0 -> List(B@20b07a5a), 2 -> List(B@7c71e0fb, B@17ff81fd), 3 -> List(B@5125545b)) 

即使你鬆的知名度，你總是可以做模式匹配的東西，如{ case a: TraitA with TraitB => }。