高阶函数使用函数作为参数或者返回函数作为结果。这是可能的,因为在 Scala 中函数和类一样,是可以单独存在的。其中一个最常见的使用高阶函数的例子是在 Scala 中的集合库中使用的 map
函数 。
val salaries = Seq(20000, 70000, 40000)
val doubleSalary = (x: Int) => x * 2
val newSalaries = salaries.map(doubleSalary) // List(40000, 140000, 80000)
doubleSalary
是一个函数,这个函数需要一个 Int
类型的整数作为参数并且返回 x * 2
。通常像这样的函数形式,=>
左边是函数的参数列表,右边表达式的值是函数的返回值。第三行,函数 doubleSalary
被集合中中的每一个元素调用。
缩短代码,我们可以将这个函数变成匿名函数,直接写在 map 函数的参数列表中。
val salaries = Seq(20000, 70000, 40000)
val newSalaries = salaries.map(x => x * 2) // List(40000, 140000, 80000)
注意:上面的代码中 x
没有声明类型。这是因为编译器可以通过 map 函数需要的类型推断出 x
的类型。
这是另外一个例子:
val salaries = Seq(20000, 70000, 40000)
val newSalaries = salaries.map(_ * 2)
因为 Scala 的编译器早已知道参数的类型(一个 Int 整数),所以,你只需要提供函数的右边。唯一需要注意的一点是你需要使用 _
来代替参数名。
方法强制包装成函数 (Coercing Methods into functions)
在高阶函数中使用方法作为参数也是可以的。因为 Scala 编译器将会强制将方法包装成函数。
case class WeeklyWeatherForecast(temperatures: Seq[Double]) {
private def convertCtoF(temp: Double) = temp * 1.8 + 32
def forecastInFahrenheit: Seq[Double] = temperatures.map(convertCtoF) // <-- passing the method convertCtoF
}
这里的方法 convertCtoF
成功地成为了方法 forecasrInFahrenheit
的参数。因为编译器强制将方法 convertCtoF
包装成了匿名函数 x => convertCtoF(x)
(注意:x
是编译器生成的名字并且保证在此作用域中唯一)。
以函数作为参数的函数 (Functions that accept functions)
使用高阶函数的原因之一是减少冗余代码。如果说你想实现因各种因素而增加工资的函数而不使用高阶函数,它可以就像下面一样:
object SalaryRaiser {
def smallPromotion(salaries: List[Double]): List[Double] =
salaries.map(salary => salary * 1.1)
def greatPromotion(salaries: List[Double]): List[Double] =
salaries.map(salary => salary * math.log(salary))
def hugePromotion(salaries: List[Double]): List[Double] =
salaries.map(salary => salary * salary)
}
执行上面的函数就可以得到你想要的输出。 [4]
注意:上面的三个方法仅仅是乘法的因子有不同。为了简化代码,你可以使用高阶函数提取出这些方法的重复代码,就像下面这样:
object SalaryRaiser {
private def promotion(salaries: List[Double], promotionFunction: Double => Double): List[Double] =
salaries.map(promotionFunction)
def smallPromotion(salaries: List[Double]): List[Double] =
promotion(salaries, salary => salary * 1.1)
def bigPromotion(salaries: List[Double]): List[Double] =
promotion(salaries, salary => salary * math.log(salary))
def hugePromotion(salaries: List[Double]): List[Double] =
promotion(salaries, salary => salary * salary)
}
新的方法 promotion
,通过一个 Double => Double
类型的函数参数(函数需要一个 Double 类型参数并且返回一个 Double 类型的值)来增加工资和返回所有的工资。
以函数作为返回值的函数 (Functions that return functions)
这里有一个以函数作为返回值的例子。
def urlBuilder(ssl: Boolean, domainName: String): (String, String) => String = {
val schema = if (ssl) "https://" else "http://"
(endpoint: String, query: String) => s"$schema$domainName/$endpoint?$query"
}
val domainName = "www.example.com"
def getURL = urlBuilder(ssl=true, domainName)
val endpoint = "users"
val query = "id=1"
val url = getURL(endpoint, query) // "https://www.example.com/users?id=1": String
注意:函数 urlBuilder
的返回值是 (String, String) => String
。意思是将返回一个拥有两个 String 类型参数且返回一个 String 类型的值的匿名函数。在这个例子中,返回的匿名函数为 (endpoint: String, query: String) => s"$schema$domainName/$endpoint?$query"
。