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

而不是 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