2017-03-01 13 views
5

mam dwie opcjeŁącząc dwa Opcja [Lista [String]] w Scala

val opt1 = Some(List("Sal", "Salil")) 
val opt2 = Some(List("Sal2", "Salil2")) 

Tak czy OPT1 lub OPT2 mógłby być brak. Jeśli którykolwiek jest Brak, chcę opcję z Listą zawartą w drugiej. Jeśli oba są Brak, Żaden nie powinien zostać zwrócony.

Jeśli oba są pewne następnie Niektóre z listą zawierającą elementy z obu list, jak pokazano poniżej:

Some(List(Sal, Salil, Sal2, Salil2)) 

wiem, że mogę to zrobić ręcznie, ale jest tam elegancki sposób to zrobić? Zrozumienia nie działają, jeśli jedna z opcji jest Brak.

Odpowiedz

6
Option((opt1 ++ opt2).flatten.toList).filter(_.nonEmpty) 
+2

To nie zwróci poprawną wartość, jeśli oba wejścia są 'Niektórzy (List())'. To powinno zwrócić 'Some (List()), ale twoje rozwiązanie zwraca' None'. –

5

Można to zrobić ładnie użyciu półgrupa dołączanie z scalaz lub kotów:

import scalaz._, Scalaz._ // for cats use `import cats._, implicits._` 

val opt1 = Option(List("Sal", "Salil")) 
val opt2 = Option(List("Sal2", "Salil2")) 

scala> opt1 |+| opt2 
res0: Option[List[String]] = Some(List(Sal, Salil, Sal2, Salil2)) 

scala> opt1 |+| None 
res1: Option[List[String]] = Some(List(Sal, Salil)) 

scala> Option.empty[List[String]] |+| None 
res2: Option[List[String]] = None 

W przeciwnym razie, z biblioteki standardowej może trzeba go obsłużyć przypadek po przypadku:

(opt1, opt2) match { 
    case (Some(a), Some(b)) => Option(a ++ b) 
    case (Some(a), None) => Option(a) 
    case (None, Some(b)) => Option(b) 
    case _ => None 
} 

Lub użyj metod gromadzenia, aby je spłaszczyć:

scala> List(opt1, opt2).flatten.flatten 
res5: List[String] = List(Sal, Salil, Sal2, Salil2) 

scala> List(opt1, None).flatten.flatten 
res6: List[String] = List(Sal, Salil) 
3

Nie sądzę, że istnieje tylko jeden, właściwy, elegancki sposób, aby to osiągnąć. Jest moja propozycja:

val opt1 = Some(List("Sal", "Salil")) 
val opt2 = Some(List("Sal2", "Salil2")) 

def merge(xs: Option[Iterable[_]]*) = xs.flatten.reduceLeftOption(_ ++ _) 

z wynikami:

merge (opt1, opt2) 
res1: Option[Iterable[_]] = Some(List(Sal, Salil, Sal2, Salil2)) 

merge (None, opt2) 
res2: Option[Iterable[_]] = Some(List(Sal2, Salil2)) 

merge (opt1, None) 
res5: Option[Iterable[_]] = Some(List(Sal, Salil)) 

merge (None, None) 
res6: Option[Iterable[_]] = None