Scala基础—偏函数的示例

来源:互联网 发布:mmd制作软件最新版 编辑:程序博客网 时间:2024/05/20 00:39

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.Dateobject 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 TestMon Dec 02 12:52:41 CST 2013----message1Mon Dec 02 12:52:41 CST 2013----message2Mon Dec 02 12:52:41 CST 2013----message3

实例中,log() 方法接收两个参数:date 和 message。我们在程序执行时调用了三次,参数 date 值都相同,message 不同。

我们可以使用偏应用函数优化以上方法,绑定第一个 date 参数,第二个参数使用下划线(_)替换缺失的参数列表,并把这个新的函数值的索引的赋给变量。以上实例修改如下:

import java.util.Dateobject 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 TestMon Dec 02 12:53:56 CST 2013----message1Mon Dec 02 12:53:56 CST 2013----message2Mon Dec 02 12:53:56 CST 2013----message3
参考:https://wizardforcel.gitbooks.io/w3school-scala/content/10-8.html