返回首页 Scala 课堂

Scala 课堂:模式匹配和函数组合

这里我们转载 Twitter 的 Scala 课堂 ,转载的内容基本来自 Twitter 的 Scala 课堂中文翻译,部分有小改动.

函数组合

让我们创建两个函数:


scala> def f(s: String) = "f(" + s + ")"
f: (s: String)String

scala>  def g(s: String) = "g(" + s + ")"
g: (s: String)String

compose

compose 组合其他函数形成一个新的函数 f(g(x))


scala> val fComposeG = f _ compose g _
fComposeG: String => String = <function1>

scala>  fComposeG("yay")
res0: String = f(g(yay))

andThen

andThen 和 compose 很像,但是调用顺序是先调用第一个函数,然后调用第二个,即 g(f(x))


scala> val fAndThenG = f _ andThen g _
fAndThenG: String => String = <function1>

scala>  fAndThenG("yay")
res1: String = g(f(yay))

柯里化 vs 偏应用

理解 PartialFunction (偏函数)

对给定的输入参数类型,函数可接受该类型的任何值。换句话说,一个(Int) => String 的函数可以接收任意 Int 值,并返回一个字符串。 对给定的输入参数类型,偏函数只能接受该类型的某些特定的值。一个定义为(Int) => String 的偏函数可能不能接受所有 Int 值为输入。 isDefinedAt 是 PartialFunction 的一个方法,用来确定 PartialFunction 是否能接受一个给定的参数。 注意 偏函数 PartialFunction 和我们前面提到的部分应用函数是无关的。


scala> val one: PartialFunction[Int, String] = { case 1 => "one" }
one: PartialFunction[Int,String] = <function1>

scala> one.isDefinedAt(1)
res2: Boolean = true

scala>  one.isDefinedAt(2)
res3: Boolean = false

您可以调用一个偏函数。


scala> one(1)
res4: String = one

PartialFunctions 可以使用 orElse 组成新的函数,得到的PartialFunction 反映了是否对给定参数进行了定义。


scala>  val two: PartialFunction[Int, String] = { case 2 => "two" }
two: PartialFunction[Int,String] = <function1>

scala>  val three: PartialFunction[Int, String] = { case 3 => "three" }
three: PartialFunction[Int,String] = <function1>

scala> val wildcard: PartialFunction[Int, String] = { case _ => "something else" }
wildcard: PartialFunction[Int,String] = <function1>

scala> val partial = one orElse two orElse three orElse wildcard
partial: PartialFunction[Int,String] = <function1>

scala>  partial(5)
res5: String = something else

scala>  partial(3)
res6: String = three

scala>  partial(2)
res7: String = two

scala>  partial(1)
res8: String = one

scala>  partial(0)
res9: String = something else

case 之谜

我们看到一些新奇的东西。我们在通常应该使用函数的地方看到了一个 case 语句。


scala> case class PhoneExt(name: String, ext: Int)
defined class PhoneExt

scala>  val extensions = List(PhoneExt("steve", 100), PhoneExt("robey", 200))
extensions: List[PhoneExt] = List(PhoneExt(steve,100), PhoneExt(robey,200))

scala>  extensions.filter { case PhoneExt(name, extension) => extension < 200 }
res10: List[PhoneExt] = List(PhoneExt(steve,100))

为什么这段代码可以工作? filter 使用一个函数。在这个例子中是一个谓词函数(PhoneExt) => Boolean 。 PartialFunction 是 Function 的子类型,所以 filter 也可以使用 PartialFunction!