将flatMap/map chain替换为for-comprehension

Question

我正在尝试使用for-comprehension来处理以下内容：

val s: Seq[Option[Int]] = //...
val t: Option[Int] = //...

s.map(sv => t.flatMap(tv => sv.map(_ == tv))) 

我试过这个：

val r: Seq[Option[Boolean]] = for(
  sv <- s;
  tv <- t;
  svv <- sv
) yield svv == tv //Seq[Boolean] does not conform to Seq[Option[Boolean]]

有没有一种用for-comprehension简洁地编写代码的方法？

Answer 1

无论如何都不鼓励在同一个for-comprehension中使用不同的集合(这里很难拒绝使用M-word )，而且它在一般情况下不会起作用。只有在Scala为这种情况定义了一些隐式转换(因为它们非常常见)的情况下，选项序列才会起作用。

最好使用两个单独的for comprehensions，一个用于序列，另一个用于选项。

val s: Seq[Option[Int]] = List(Some(1), None, Some(2))
val t: Option[Int] = Some(2)

val result = for {
  elem <- s
} yield for {
  evalue <- elem
  tvalue <- t
} yield evalue == tvalue

println(result) // List(Some(false), None, Some(true))

Answer 2

我可能会混合一些东西来保持它的简单和简短：

val r: Seq[Option[Boolean]] =
  for (
    opt <- s;
    v <- t
  ) yield opt.map(_ == v)