scala - Scala 中的魔术 PartialFunction

Question

我不认为这段代码应该有效，但它确实有效(在 Scala 2.10 中):

scala>     ((i: Int) => i.toString match {
     |        case s if s.length == 2 => "A two digit number"
     |        case s if s.length == 3 => "A three digit number"
     |     }): PartialFunction[Int,String]
res0: PartialFunction[Int,String] = <function1>

// other interactions omitted

scala> res0.orElse(PartialFunction((i: Int) => i.toString))
res5: PartialFunction[Int,String] = <function1>

scala> res5(1)
res6: String = 1

它是如何工作的？我希望 MatchError 被抛入 res0。

Scala 语言规范似乎没有明确说明 res0 应该如何解释。

Answer 1

诀窍是编译器不会将您的定义解释为转换为部分函数的总函数——它实际上首先创建了一个部分函数。您可以通过注意 res0.isDefinedAt(1) == false 来验证。

如果您实际将全函数转换为偏函数，您将获得预期的行为:

scala> PartialFunction((i: Int) => i.toString match {
     |       case s if s.length == 2 => "A two digit number"
     |       case s if s.length == 3 => "A three digit number"
     |     })
res0: PartialFunction[Int,String] = <function1>

scala> res0 orElse ({ case i => i.toString }: PartialFunction[Int, String])
res1: PartialFunction[Int,String] = <function1>

scala> res1(1)
scala.MatchError: 1 (of class java.lang.String)
// ...

在此示例中，PartialFunction.apply 将其参数视为一个总函数，因此有关其定义位置的任何信息都将丢失。