Question

我在Scala中定义了两个版本的PartialFunction。

  val doubleEvens1: PartialFunction[Int, Int] = {
    case x if x % 2 == 0 => x * 2
  }

  val doubleEvens2 = new PartialFunction[Int, Int] {
    override def isDefinedAt(x: Int): Boolean = x % 2 == 0

    override def apply(v1: Int): Int = v1 * 2
  }

和一个清单：

val list = List(1, 2, 3, 4, 5)

但是他们的行为对于未定义的值是不同的：

// doubleEvens1
println(doubleEvens1.isDefinedAt(3)) // false
println(list.collect(doubleEvens1))  // List(4, 8)
println(doubleEvens1(3))             // scala.MatchError
println(list.map(doubleEvens1))      // scala.MatchError

// doubleEvens2
println(doubleEvens2.isDefinedAt(3)) // false
println(list.collect(doubleEvens2))  // List(4, 8)
println(doubleEvens2(3))             // 6
println(list.map(doubleEvens2))      // List(2, 4, 6, 8, 10)

我猜doubleEvens1应该是合理的，因为它符合数学定义。但是doubleEvens2的行为是出于目的设计的吗？或者我在代码片段中遗漏了什么？

Answer 1

如果您实现自己的apply方法而不检查isDefinedAt，那么显然您可以使用任何输入调用该方法，而不会抛出异常。这在文档中明确说明：

调用者有责任在调用isDefinedAt之前调用apply，因为如果isDefinedAt为false，则无法保证apply会抛出异常以指示错误的情况。如果未抛出异常，则评估可能会导致任意值。

恰好，从模式匹配创建的部分函数将从其isDefinedAt方法实现中调用apply本身并抛出异常。

因此，如果要复制该行为，则必须执行此操作：

val doubleEvens2 = new PartialFunction[Int, Int] {
  override def isDefinedAt(x: Int): Boolean = x % 2 == 0

  override def apply(x: Int): Int = 
    if (isDefinedAt(x)) x * 2 else throw new MatchError(x.toString)
}

缺点当然是可能会多次调用某些代码（isDefined），如this question中所述。

PartialFunction如何处理Scala中未定义的值？

1 个答案: