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的行为不会改变，并且如果找不到密钥则会抛出异常 - 因此在这种情况下不会使用默认值，这都是故意的。 / p>

但是，在单元测试中传递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(...)。

如果您总是对测试用例使用相同的类型，您还可以在specs类中创建一个类型别名，以减少混乱：

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类型的信息。在A为None的情况下，无法知道Some[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))

包含地图

2 个答案: