为何使用变量会让Scala偏函数的偏特性减弱？

先看问题中的代码示例：

val f1: PartialFunction[Option[Int], Int] = { b =>
  val i = identity(b)
  i match {
    case Some(res) => res
  }
}
val f2: PartialFunction[Option[Int], Int] = { b =>
  identity(b) match {
    case Some(res) => res
  }
}

f1.isDefinedAt(None)
f2.isDefinedAt(None)

不同Scala版本的表现 #

• Scala 2.12.18：f1编译报错，提示：

  type mismatch;
  found : Option[Int] => Int
  required: PartialFunction[Option[Int],Int]
  f2可正常编译。

• Scala 2.13.12 / Scala 3.3.1：
  • f1编译时触发匹配穷尽性警告，调用isDefinedAt(None)返回true；
  • f2无编译警告，调用isDefinedAt(None)返回false。

核心原因：Scala对偏函数的识别规则 #

Scala的偏函数语法糖依赖函数体直接为模式匹配语句的结构，编译器只有在这种情况下，才能自动分析模式分支，生成对应的isDefinedAt逻辑——即判断输入是否能匹配到有效分支。

• 对于f1：函数体先定义了临时变量i，再对i做匹配。这种结构无法被编译器识别为偏函数的模式匹配，只能被当作普通的Function1（单参数函数）处理。
  • Scala 2.12中普通函数不能直接转为PartialFunction，因此编译报错；
  • Scala 2.13及之后版本会自动将普通函数包装为偏函数，但这个自动生成的偏函数的isDefinedAt会永远返回true（编译器无法从临时变量后的匹配逻辑中提取判断条件），同时因为匹配未覆盖None分支，触发穷尽性警告。
• 对于f2：函数体直接是对参数表达式的模式匹配，编译器可以完整识别偏函数结构，根据case Some(res)分支生成正确的isDefinedAt逻辑——仅当输入为Some类型时返回true，None时返回false，且编译器明确知道该偏函数只处理Some情况，因此不会触发穷尽性警告。

内容的提问来源于stack exchange，提问作者Suma