Scala 偏函数：PartialFunction

Scala 中的 PartialFunction 是一种特殊的函数类型，它只对部分输入值有定义。与普通函数（Function1[A, B]）不同，偏函数允许你明确声明“哪些输入我能处理，哪些我不管”。这种特性在模式匹配、异常处理和事件分发等场景中非常实用。

1. 理解偏函数的核心概念

创建一个偏函数时，你不是写一个对所有输入都返回结果的函数，而是定义一组“条件-行为”规则。只有当输入满足某个条件时，偏函数才会执行对应逻辑；否则，它直接“不响应”。

在 Scala 中，偏函数通过 PartialFunction[A, B] 类型表示，其中：

• A 是输入类型
• B 是输出类型

判断一个值是否被偏函数覆盖，使用 isDefinedAt(x) 方法。调用偏函数时，如果传入未定义的值，会抛出 MatchError。

2. 定义偏函数的两种方式

方式一：使用花括号和 case 语句（推荐）

这是最常见、最简洁的写法：

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

这段代码定义了一个偏函数：只处理偶数，将其除以 2；奇数不在其定义域内。

方式二：继承 PartialFunction 并实现抽象方法

val safeSqrt = new PartialFunction[Double, Double] {
  def apply(x: Double): Double = math.sqrt(x)
  def isDefinedAt(x: Double): Boolean = x >= 0
}

这种方式显式实现了 apply 和 isDefinedAt 方法，适用于逻辑较复杂的场景。

3. 使用偏函数的关键操作

1. 检查输入是否在定义域内：
   调用 isDefinedAt(value)。例如：
   divideByTwo.isDefinedAt(4) 返回 true，而 divideByTwo.isDefinedAt(3) 返回 false。

2. 安全调用偏函数：
   不要直接 divideByTwo(3)，这会抛异常。应先检查或使用 lift 方法转换为 Option：

   val result: Option[Int] = divideByTwo.lift(3) // 返回 None

3. 组合多个偏函数：
   使用 orElse 将多个偏函数连接成一个更大的偏函数：

   val handleEven: PartialFunction[Int, String] = {
     case x if x % 2 == 0 => s"$x is even"
      }
   
      val handleOdd: PartialFunction[Int, String] = {
        case x if x % 2 == 1 => s"$x is odd"
   }
   
   val handleAll = handleEven orElse handleOdd

   现在 handleAll 对所有整数都有定义。

4. 偏函数在集合操作中的典型应用

Scala 集合的 collect 方法专门接受偏函数，只处理匹配的元素，自动跳过不匹配项：

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

val doubledEvens = numbers.collect {
  case x if x % 2 == 0 => x * 2
}
// 结果: List(4, 8)

对比 map，它要求对每个元素都必须有返回值；而 collect 利用偏函数天然支持“选择性处理”，代码更简洁、意图更清晰。

5. 偏函数与普通函数的区别

虽然语法相似，但偏函数和普通函数在类型和行为上有本质区别：

注意：任何 PartialFunction 都是 Function1 的子类型，因此可以传递给期望普通函数的地方，但反过来不行。

6. 实战：构建健壮的事件处理器

假设你正在开发一个聊天机器人，需要根据消息类型执行不同操作：

sealed trait Message
case class Text(content: String) extends Message
case class Image(url: String) extends Message
case class Unknown(payload: String) extends Message

val botHandler: PartialFunction[Message, String] = {
  case Text(content) if content.startsWith("/help") =>
    "可用命令: /help, /image"
  case Image(url) if url.endsWith(".jpg") =>
    s"收到图片: $url"
}

val fallbackHandler: PartialFunction[Message, String] = {
  case _ => "无法处理该消息"
}

val fullHandler = botHandler orElse fallbackHandler

调用 fullHandler(Text("/help")) 返回帮助信息；调用 fullHandler(Unknown("...")) 触发兜底逻辑。这种设计让核心逻辑清晰分离，扩展性强。

7. 避免常见陷阱

1. 不要忽略 isDefinedAt 的一致性：
   如果你在 apply 中处理了某种输入，isDefinedAt 必须返回 true。否则会导致不可预测行为。使用 case 语法定义偏函数可自动保证一致性。

2. 慎用副作用：
   偏函数常用于 collect，而 collect 可能被多次调用（如在并行集合中）。确保偏函数是纯函数，避免依赖外部状态。

3. 性能考量：
   每个 case 子句都会生成一个 if 判断。如果分支很多，考虑是否用 Map 或其他结构替代。

// 示例：安全解析用户输入
val parseCommand: PartialFunction[String, Unit] = {
  case cmd if cmd == "start" => println("Starting...")
  case cmd if cmd == "stop"  => println("Stopping...")
}

val input = scala.io.StdIn.readLine()
if (parseCommand.isDefinedAt(input)) {
  parseCommand(input)
} else {
  println("Unknown command")
}