Scala基础—偏函数的示例

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