Scala隐含用于包含映射的选件

Question

我试图写下以下隐式：

implicit class ExtractOrElse[K, V](o: Option[Map[K, V]]) {
  def extractOrElse(key: K)(f: => V): V = { if (o.isDefined) o.get(key) else f }
}

我想以这种方式使用：

normalizationContexts.extractOrElse(shardId)(defaultNormalizationContext)

避免使用笨拙的语法（normalizationContexts是Option[Map[String, NormzalitionContext]]）。

另外，我补充一点是有意的，只有一个默认值：如果Option isEmpty，它将使用它，但是如果Option isDefined，则不会更改Map的行为，并且会抛出一个例外如果找不到密钥 - 因此在这种情况下不会使用默认值，这都是故意的。

但是，在单位测试中通过None传递时，我会遇到错误：

assertEquals(None.extractOrElse('a')(0), 0)

导致：

Error:(165, 37) type mismatch;
 found   : Char('a')
 required: K
  assertEquals(None.extractOrElse('a')(0), 0)

我意识到None不是参数，因为它被定义为：

case object None extends Option[Nothing] {
  def isEmpty = true
  def get = throw new NoSuchElementException("None.get")

做这项工作的最佳方法是什么？

Answer 1

而不是None.extractOrElse(...)，尝试Option.empty[Map[Char, Int]].extractOrElse(...)。

如果您始终在测试用例中使用相同的类型，则还可以在规格类中创建类型的别名以减少混乱：

type OpMap = Option[Map[Char, Int]]
// ...
assertEquals(Option.empty[OpMap].extractOrElse('a')(0), 0)

以防万一，您可以使用flatMap和getOrElse在不编写新方法的情况下实现相同的事情：

val n = Option.empty[Map[String, Int]]
val s = Some(Map("x" → 1, "y" → 2))
n.flatMap(_.get("x")).getOrElse(3)  // 3
s.flatMap(_.get("x")).getOrElse(3)  // 1
s.flatMap(_.get("z")).getOrElse(3)  // 3

Answer 2

类型系统没有足够的有关K和V类型的信息。在您的None为Some[A]的情况下，A的类型是什么。

当我创建一个具有显式类型的示例时，代码按预期工作：

// Like this
val e = new ExtractOrElse(Option.empty[Map[Char, Int]])
e.extractOrElse('a')(0) // Equals 0
// Or like this
val e = new ExtractOrElse[Char, Int](None)
println(e.extractOrElse('a')(0))
// Or like this
val m: Option[Map[Char, Int]] = None
val e = new ExtractOrElse(m)
println(e.extractOrElse('a')(0))