package com.spark.scala.basics
/**
* 1.Spark源码中的用法.
*/
object PartialFunctionDemo {
def main(args: Array[String]): Unit = {
val sample = 1 to 10
/* val isEven: PartialFunction[Int, Unit] = {
case x if x % 2 == 0 => println(x + " is Even!")
}
isEven(4)*/
val isEven: PartialFunction[Int, String] = {
case x if x % 2 == 0 => x + " is Evne!"
}
isEven(4)
val evenNumbers = sample collect isEven //collect方法
// evenNumbers.foreach(println)
// evenNumbers.foreach(x => println(x)) //等价于
// evenNumbers.foreach(println(_)) //等价于
// evenNumbers.foreach(println) //等价于
val isOdd: PartialFunction[Int, String] = {
case x if x % 2 == 1 => x + " is odd"
}
val numbers = sample map (isEven orElse (isOdd))
numbers.foreach(println)
}
}
Scala 偏应用函数是一种表达式,你不需要提供函数需要的所有参数,只需要提供部分,或不提供所需参数。
如下实例,我们打印日志信息:
import java.util.Date
object Test {
def main(args: Array[String]) {
val date = new Date
log(date, "message1" )
Thread.sleep(1000)
log(date, "message2" )
Thread.sleep(1000)
log(date, "message3" )
}
def log(date: Date, message: String) = {
println(date + "----" + message)
}
}
执行以上代码,输出结果为:
$ scalac Test.scala
$ scala Test
Mon Dec 02 12:52:41 CST 2013----message1
Mon Dec 02 12:52:41 CST 2013----message2
Mon Dec 02 12:52:41 CST 2013----message3
实例中,log() 方法接收两个参数:date 和 message。我们在程序执行时调用了三次,参数 date 值都相同,message 不同。
我们可以使用偏应用函数优化以上方法,绑定第一个 date 参数,第二个参数使用下划线(_)替换缺失的参数列表,并把这个新的函数值的索引的赋给变量。以上实例修改如下:
import java.util.Date
object Test {
def main(args: Array[String]) {
val date = new Date
val logWithDateBound = log(date, _ : String)
logWithDateBound("message1" )
Thread.sleep(1000)
logWithDateBound("message2" )
Thread.sleep(1000)
logWithDateBound("message3" )
}
def log(date: Date, message: String) = {
println(date + "----" + message)
}
}
执行以上代码,输出结果为:
$ scalac Test.scala
$ scala Test
Mon Dec 02 12:53:56 CST 2013----message1
Mon Dec 02 12:53:56 CST 2013----message2
Mon Dec 02 12:53:56 CST 2013----message3
参考:https://wizardforcel.gitbooks.io/w3school-scala/content/10-8.html