scala - 如何使用 yield 语句返回不同的集合

Question

根据我有限的知识，我知道编译器会自动继承集合返回类型，并基于它确定要返回的集合类型，所以在下面的代码中我想返回Option[Vector[String]]。

我尝试使用以下代码进行试验，但出现编译错误

type mismatch;  found   : scala.collection.immutable.Vector[String]  required: Option[Vector[String]]   

代码：

def readDocument(v:Option[Vector[String]]) : Option[Vector[String]] =    
{
   for ( a <- v;  
   b <- a )
     yield 
     {
        b
     }
}

Answer 1

scala> for (v <- Some(Vector("abc")); e <- v) yield e
<console>:8: error: type mismatch;
 found   : scala.collection.immutable.Vector[String]
 required: Option[?]
              for (v <- Some(Vector("abc")); e <- v) yield e
                                               ^

scala> for (v <- Some(Vector("abc")); e = v) yield e
res1: Option[scala.collection.immutable.Vector[String]] = Some(Vector(abc))

嵌套x <- xs意味着flatMap并且仅当返回的类型与最外层的类型相同时才会起作用。

Answer 2

for理解已经为您拆箱，所以Option这应该有效

def readDocument(v:Option[Vector[String]]) : Option[Vector[String]] =    
{
  for ( a <- v )
  yield 
  {
    a
  }
}

Answer 3

这个怎么样？

  def readDocument(ovs: Option[Vector[String]]): Option[Vector[String]] =
    for (vs <- ovs) yield for (s <- vs) yield s