前文已经提到Scala变量的用法,现在聊聊函数。在Scala里,函数的定义很简单。例如:
def add(x: Int, y: Int): Int = {
x + y
}
Scala的函数和Python一样,以def开头,然后是函数名add,最后指定函数的参数,这里Scala的类型推断就不灵了,需要在冒号后面指定传入的函数参数类型。add函数接受两个为Int类型的参数,返回一个Int类型的结果参数。花括号里是add函数的函数体,负责函数的具体实现。Scala的函数的返回值,不需要指明return。
当然这是最死板的写法,灵活的Scala衍生出了更多的函数定义方式:
def add(x: Int, y: Int) = {
x + y
}
def add(x: Int, y: Int) = x + y
在讨论完Scala函数的定义以后,那么Scala是如何定义函数是一等对象的? 在函数式编程里,函数本身不仅仅可以被定义和调用,也可以被当作值传递。在此之前先引入lambda函数,Scala里称为函数字面量,如下:
scala> (x:Int) => x + 1
res0: Int => Int = <function1>
虽然这个函数字面量还不能使用,但之前已经提到过了,函数本身是可以被当作值传递的,所以可以把上述的函数自变量,传给一个变量:
val increaseOne = (x: Int) => x + 1
与Python的lambda函数只能接受表达式不一样,Scala的lambda函数可以像一般的函数定义体一样定义。
val increaseOne = (x: Int) => {
x + 1
}
这样的用法推广开来,就演变成了下面的用法:
val lst = List(1, 2, 3, 4, 5)
lst.foreach((x:Int) => println(x))
按照Scala能省则省的性格,上面的可以省略为:
lst.foreach(println)