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

Question

我想调用一个通用函数f[X](...)，在我的例子中X恰好是Option[Y]。我尝试将Some[...] 和None 都传递给函数需要X，但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 将推断出我在 cmets 中所写的正确类型（以及他的回答中的 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

Answer 2

如果您的目标是提高可读性，并从编译器那里获得帮助来推断类型，请尝试：

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

甚至更好：

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

或者甚至更好（在我看来）：Peter Neyens 的回答。使用 Traverse 实例中的 sequence。