Lọc một danh sách Scala theo loại

Question

Tôi có một cấu trúc lớp như thế này

và tôi có một bộ sưu tập của các trường hợp khác nhau, ví dụ:

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

Câu hỏi của tôi là, liệu có một cách thanh lịch để lọc ra một số lớp con khỏi Danh sách?

Giả sử tôi muốn tất cả các trường hợp ngoại trừ B và C. Tôi có thể làm điều đó với một loạt các isInstanceOf, hoặc bằng cách sử dụng thu thập như thế này:

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

này hoạt động nhưng nó cảm thấy lúng túng, chủ yếu là vì các bộ lọc và đúc. Có cách nào thanh lịch hơn không? Ít mã được ưu tiên và tôi muốn có giải pháp không cần phải cập nhật nếu tôi thêm các lớp con khác của A.

Answer 1

flatMap rằng shit! (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) 

Trong trường hợp của bạn, bạn đã nhận được một List[ScalaObject] vì ScalaObject là đơn giản nhất trên ràng buộc của None, D, và E.

Answer 2

Xây dựng các vấn đề như câu hỏi có vẻ là cách khá tốt để giải quyết chúng :) câu hỏi của tôi thực sự cung cấp câu trả lời - chỉ cần lọc theo kiểu phụ:

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

Answer 3

collect có thể được sử dụng để lọc các giá trị mà hàm được định nghĩa:

Nhận tất cả các giá trị của loại A:

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

Nhận tất cả các giá trị ngoại trừ B và C:

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