Filtrare un elenco Scala per tipo

Question

ho una struttura di classe come questo

e ho una collezione dei diversi casi, ad esempio:

val xs = List(new D, new B, new E, new E, new C, new B) 

La mia domanda è, c'è un modo elegante per filtrare alcune sottoclassi dalla lista?

Diciamo che voglio tutte le istanze tranne B e C. Posso farlo con un po 'di isInstanceOf di, o utilizzando raccogliere in questo modo:

val ys = (xs collect { 
    case b: B => None 
    case c: C => None 
    case notBorC => notBorC 
}).filter(_ != None).asInstanceOf[List[A]] 

Questo funziona ma ci si sente a disagio, soprattutto a causa del filtro e cast. C'è un modo più elegante? Si preferisce meno codice e mi piacerebbe avere una soluzione che non debba essere aggiornata se aggiungo più sottoclassi di A.

Answer 1

flatMap quella merda! (as they say):

scala> val ys = xs flatMap { 
    | case _: B | _: C => None 
    | case other => Some(other) 
    | } 
ys: List[A] = List(D@7ecdc97b, E@2ce07e6b, E@468bb9d1) 

Nel tuo caso si stavano diventando un List[ScalaObject] perché ScalaObject è l'estremo superiore di None, D e E.

Answer 2

Formulare problemi come domande sembra essere un buon modo per risolverli :) la mia domanda in realtà fornisce la risposta - basta filtrare per sottotipo:

val ys = xs filterNot(List(classOf[B], classOf[C]) contains _.getClass) 

Answer 3

collect può essere utilizzato per filtrare i valori su cui è definita la funzione:

garantita tutti i valori di tipo A:

xs.collect { case a: A => a } 

ottenere tutti i valori eccetto B e C:

xs diff xs.collect { case x@(_: B | _: C) => x }