scala - 使用 case 语句时，isDefinedAt 方法如何工作？

Question

在this scala偏函数解释，使用case语句如下:

val divide2: PartialFunction[Int, Int] = {
    case d: Int if d != 0 => 42 / d
}

然后它说:

Although this code doesn’t explicitly implement the isDefinedAt method, it works exactly the same as the previous divide function definition

前面的 divide 函数显式定义了 apply 和 isDefined 方法如下:

    val divide = new PartialFunction[Int, Int] {
    def apply(x: Int) = 42 / x
    def isDefinedAt(x: Int) = x != 0
}

我看到编译器在这里施展了某种魔法，但我对这种机制的本质很好奇:守卫是否在后台提取到 isDefined 的主体中(如果 d != 0)？有自己的 guard 的多个案例陈述怎么样？函数的域在 case 语句中定义，因此这一定是编译器在幕后构建 isDefined 实现的方式。

我说得对吗？

Answer 1

首先，解释是错误的，因为所提供的两个示例的工作方式不同:

scala> val a: PartialFunction[Int, Int] = {
     | case i if i != 0 => 42 / i
     | }
a: PartialFunction[Int,Int] = <function1>

scala> val b = new PartialFunction[Int, Int] {
     | def apply(i: Int) = 42 / i
     | def isDefinedAt(i: Int) = i != 0
     | }
b: PartialFunction[Int,Int] = <function1>

scala> a(0)
scala.MatchError: 0 (of class java.lang.Integer)
  at scala.PartialFunction$$anon$1.apply(PartialFunction.scala:253)
  at scala.PartialFunction$$anon$1.apply(PartialFunction.scala:251)
  at $anonfun$1.applyOrElse(<console>:7)
  at $anonfun$1.applyOrElse(<console>:7)
  at scala.runtime.AbstractPartialFunction$mcII$sp.apply$mcII$sp(AbstractPartialFunction.scala:36)
  ... 33 elided

scala> b(0)
java.lang.ArithmeticException: / by zero
  at $anon$1.apply$mcII$sp(<console>:8)
  ... 33 elided

但它的核心是正确的，编译器将 case 子句中的代码转换为 isDefinedAt 方法的 bool 表达式，然后将函数本身作为 apply 方法(因此如果值未定义)

举个一般的例子，一个像这样的函数:

val a = PartialFunction[A, B] {
    case <clause A> => <action A>
    case <clause B> => <action B>
}

编译为:

val a = new PartialFunction[A, B] {
    def isDefinedAt(a: A) = a match {
        case <clause A> => true
        case <clause B> => true
        case _ => false
    }
    def apply(a: A) = a match {
        case <clause A> => <action A>
        case <clause B> => <action B>
    }
}

这是 spec定义了这种行为。