scala - 我可以让 Scala 在这里推断 Option 类型吗？

Question

我想调用一个通用函数 f[X](...)，在我的例子中 X 恰好是 Option[Y]。我尝试在函数需要 X 的地方传递 Some[...] 和 None，但 Scala 坚持 X 是 Some[Y] 类型。

  def flattenOptionMap[A, B](input : Map[A, Option[B]]) : Option[Map[A, B]] = {
    input.foldLeft[Option[Map[A,B]]] (Some(Map.empty)) {
      case (_, (_, None)) => None
      case (None, (_, _)) => None
      case (Some(acc), (key, Some(value))) => Some(acc + (key -> value))
    }
  }

在这个例子中，我必须明确指定 Option[Map[A,B]] 应该用作 foldLeft 的通用类型。所有必要的类型信息都已包含在上下文中，并且在我看来，过于频繁地键入诸如 Option[Map[A,B]] 之类的繁琐类型会大大降低代码的可读性。

有没有办法让 Scala 推断类型，或者避免复制粘贴整个类型？

Answer 1

当您使用 Option(Map.empty[A, B]) 作为您的 foldLeft 的起始值时，Scala 将推断出正确的类型，正如我在评论(以及他的回答中的 beefyhalo)。

我想补充一点，如果您愿意使用 Scalaz，您可以只使用 sequence 函数。

import scalaz._
import Scalaz._ 

val mapSome = Map(1 -> Some("a"), 2 -> Some("b"))
val mapNone = Map(1 -> Some("a"), 2 -> Some("b"), 3 -> None)

mapSome.sequence
flattenOptionMap(mapSome)
// Option[Map[Int,String]] = Some(Map(1 -> a, 2 -> b))

mapNone.sequence
flattenOptionMap(mapNone)
// Option[Map[Int,String]] = None