Scala 隐式包含 Map 的 Option

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

类型系统没有关于类型 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))

Answer 2

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

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

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

Scala 隐式包含 Map 的 Option

Scala implicit for Option containing Map

scala

implicit