Question

我有这样的类结构

abstract class A
class B extends A
class C extends A
class D extends A
class E extends A

我有各种实例的集合，例如：

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

我的问题是，是否有一种优雅的方法可以从List中过滤掉一些子类？

假设我想要除B和C以外的所有实例。我可以用一堆isInstanceOf来做，或者像这样使用collect：

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

这有效，但感觉很尴尬，主要是因为滤镜和演员。有更优雅的方式吗？首选的代码较少，如果我添加更多A的子类，我希望有一个不需要更新的解决方案。

Answer 1

collect可用于过滤定义函数的值：

获取A类的所有值：

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

获取除B和C以外的所有值：

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

Answer 2

flatMap那个狗屎！（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)

在您的情况下，您获得了List[ScalaObject]，因为ScalaObject是None，D和E的最小上限。

Answer 3

将问题表达为问题似乎是解决问题的一种很好的方法:)我的问题实际上提供了答案 - 只需按子类型过滤：

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

按类型过滤Scala列表

3 个答案: