在Scala中组合两个Option [List [String]]

Sal*_*ran 6 scala

我有两个选项

val opt1 = Some(List("Sal", "Salil"))
val opt2 = Some(List("Sal2", "Salil2"))
Run Code Online (Sandbox Code Playgroud)

opt1或opt2可以是None.如果任何一个是None,那么我想要一个包含在另一个中的List的选项.如果两者都是None,则应返回None.

如果两者都是Some,那么有些列表包含来自两个列表的元素,如下所示:

Some(List(Sal, Salil, Sal2, Salil2))
Run Code Online (Sandbox Code Playgroud)

我知道我可以手工做到这一点,但有一种优雅的方式来做到这一点吗?如果其中一个选项为"无",则"推理"不起作用.

Alv*_*sco 9

Option((opt1 ++ opt2).flatten.toList).filter(_.nonEmpty)
Run Code Online (Sandbox Code Playgroud)

  • 如果两个输入都是"Some(List())",则不会返回正确的值.这应该返回`Some(List())`,但你的解决方案返回`None`. (3认同)

Mic*_*jac 5

你可以使用scagz或cat附加的半群很好地做到这一点:

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
Run Code Online (Sandbox Code Playgroud)

否则,使用标准库,您可能需要逐个处理它:

(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
}
Run Code Online (Sandbox Code Playgroud)

或者使用收集方法将它们展平:

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)
Run Code Online (Sandbox Code Playgroud)