偏函数是一个一元方程,PartialFunction[A, B]
输入A类型,返回一个B类型。
定义一个偏函数:
val fn1: PartialFunction[Int, Int] = {
case x: Int if x < 10 => x + 1
}
Assert.assertEquals(6, fn1(5))
try {
fn1(11)
} catch {
case _: MatchError =>
Assert.assertTrue(true)
}
接收一个小于10的数x,返回x + 1。
常用的方法:
1)orElse
对偏函数进行扩展:
val fn2: PartialFunction[Int, Int] = fn1.orElse {
case x: Int => x + 5
}
Assert.assertEquals(16, fn2(11))
原本大于等于10的数会抛异常,现在也可以处理了。
2)andThen
对偏函数的结果进行处理:
val fn3 = fn1.andThen(x => 2 * x)
Assert.assertEquals((5 + 1) * 2, fn3(5))
3)lift
将偏函数的结果包装成Option对象,异常时为None
val fn4 = fn1.lift
Assert.assertEquals(None, fn4(11))
Assert.assertEquals(Some(6), fn4(5))
4)常常跟科里化结合起来
def fn5(x: Int)(f: PartialFunction[Int, Int]): Int = {
f(x)
}
val res: Int = fn5(10) {
case x: Int => x + 5
}
Assert.assertEquals(15, res)