本文作者:林伟兵,叩丁狼高级讲师。原创文章,转载请注明出处。
4. 方法&函数的定义
在scala中,函数是一等公民,但是一般在开发中,函数和方法是可以相互代替的。
4.1 方法的定义
-
一个完整方法的定义如下
scala> def add(a:Int, b:Int):Int = { | return a+b | } add: (a: Int, b: Int)Int scala> add(3,5) res1: Int = 8
-
方法的最后一句为返回值的话,可以将return 去掉, 如果一个方法体只有一句代码,大括号可以去掉。
scala> def times(x:Int,y:Int) = x * y times: (x: Int, y: Int)Int scala> times(3,5) res0: Int = 15
-
如果一个方法没有返回值,其返回类型为Unit , 并且“=”号可以去掉。
scala> def log(msg:String){ | println(msg) | } log: (msg: String)Unit scala> log("测试") 测试
返回值的类型也可以通过智能推导来完成。
4.2 方法转函数的实现
在方法名后加上“ _ ” ,可以将方法转换成函数,如下代码,表示函数有2个参数,类型都是Int; 并且有 Int 返回值。
在命令行中,系统会将转换好的函数赋值给res5 ; 可以通过res5调用该函数。
scala> add2 _
res5: (Int, Int) => Int =
scala> res5(8,8)
res6: Int = 16
4.3 函数的定义
函数的定义可以在方法的基础上进行转换,比如下面代码介绍了函数和方法的定义:
## 方法的定义:def 方法名():ReturnType => {}
scala> def times(x:Int, y:Int) => x*y
## 函数的定义: 去掉了def和方法名 这个代码块需要被别人调用,所以要用一个变量来引用。
scala> val times=(x:Int, y:Int) => x*y
times: (Int, Int) => Int =
在函数使用过程中,还有另一个常见的操作,就是将函数作为另一个函数的参数来使用,如下代码:
var money =100
def calMoneyLeft():Int = {
println("函数参数调用中")
money
}
def cost(x: => Int){
money -=50
println("函数参数调用之前")
println(x)
}
cost(calMoneyLeft())
4.4 可变参数
对于Java来说,可变参数的声明的使用如下:
public class Test {
public static void foo(String... args){
}
public static void bar(int... args){
}
//我们常见的main函数就是一个带有可变参数的函数
public static void main(String... args){
//可变参数在调用的时候可以传递一个或者多个参数,也可以不传递
foo("Hello world","Hello Java");
bar(1,2,3,4);
bar();
}
}
scala的可变参数与Java的差不多,只不过把 “…” 改为在类型后面加上“*” 使用如下:
object MultipleArguments {
def main(args:Array[String]):Unit = {
println("sum = "+calcSumWithQueue(1,3,5,9,7))
}
def calcSumWithQueue(numbers: Int*):Int = {
var sum = 0
for (element <- numbers){
sum+= element
}
sum
}
}
可变参数的位置:
object MultipleArguments {
//这种写法没毛病
def foo(arg1: Int,arg2: Double*):Unit = { }
//可变参数后面不能再出现其他参数
//def foo2(arg1: Double*,arg2: Int):Unit = { }
//不允许出现多个类型的可变参数
//def foo3(arg1: Double*,arg2: Int*):Unit = { }
//总结:一个函数中只能有一个可变参数,并且只能放在参数列表最后
}
可变参数的传参问题:
object MultipleArguments {
def main(args: Array[String]): Unit = {
//可变参数的实参为Range的时候,需要添加:_*来进行数据的转换
println("sum = " + calcSumWithQueue(1 to 10: _*))
//可变参数的实参为Array时,需要添加:_*来进行数据的转换
val arr = Array("Hadoop", "Spark", "Storm")
printArr(arr: _*)
}
def calcSumWithQueue(numbers: Int*): Int = {
var sum = 0
for (element <- numbers) {
sum += element
}
sum
}
def printArr(arr: String*) {
arr.foreach(ele => println(ele))
}
}
4.5 参数的默认值
在函数声明的时候可以为参数设置默认值;在调用的时候将真实的值传递进去,也可以根据参数的名称进行传值:
object ArgWithDefaultVal {
def main(args: Array[String]):Unit = {
recordUser()
recordUser("张三",1.65d)
//在方法被调用的时候,指定要传入的参数名,这种叫做命名参数
recordUser(height = 1.55d,name = "李四")
}
/*
* 用户信息的录入
* */
def recordUser(name: String="匿名",height: Double=1.70d):Unit = {
println("录入用户记录 name:"+name+",height:"+height)
}
}
假如一个方法没有任何参数,那么在调用的时候,括号是可以去掉的。
想获取更多技术视频,请前往叩丁狼官网:http://www.wolfcode.cn/openClassWeb_listDetail.html