Scala: filtrar una colección de opciones

Question

Decir que tengo una función que comprueba si alguna operación es aplicable a una instancia de A y, si es así, devuelve una instancia de B o Nada:

def checker[A,B](a: A) : Option[B] = ... 

Ahora quiero para formar una nueva colección que contenga todas las instancias válidas de B, descartando los valores Ninguno. El siguiente código parece hacer el trabajo, pero sin duda hay una mejor manera:

val as = List[A](a1, a2, a3, ...) 
    val bs = 
    as 
    .map((a) => checker(a)) // List[A] => List[Option[B]] 
    .filter(_.isDefined)  // List[Option[B]] => List[Option[B]] 
    .map(_.get)    // List[Option[B]] => List[B] 

Gracias!

Answer 1

Esto debe hacerlo:

val bs = as.flatMap(checker) 

Answer 2

La respuesta anterior es correcta, pero si se puede reescribir checker, le sugiero que utilice PartialFunction y collect. Función parcial es una función del tipo A => B que no es necesario definir para todos los valores de A. Aquí está un ejemplo simple:

scala> List(1, 2, 3, 4, "5") collect {case x : Int => x + 42} 
res1: List[Int] = List(43, 44, 45, 46) 

collect toma una instancia de función parcial como argumento y lo aplica a todos los elementos de la colección. En nuestro caso, la función se define solo para Ints y "5" se filtra. Por lo tanto, collect es una combinación de map y filter, que es exactamente tu caso.