spark scala中 => 符号的使用场景

2019-09-25 17:00:57 | 编辑

作为一个java程序员,对scala虽然很亲和,但是有些语法特别是符号=>的使用还是很茫然,所以特意总结了一下spark scala中 => 符号的使用场景。

1.表示函数的返回类型(Function Type)

scala> def double(x: Int): Int = x*2
 double: (x: Int)Int
 //定义一个函数变量: 
 scala> var x : (Int) => Int = double
 x: Int => Int = <function1>
 //调用
 scala> x(2)
 res1: Int = 4

函数double的类型就是 (x: Int) => Int 或者 Int => Int。左边是参数类型,右边是方法返回值类型。
备注: 当函数只有一个参数的时候,函数类型里面括起来函数参数的括号是可以省略的。

 

2.匿名函数

//通过匿名函数定义一个函数变量xx
 scala> var xx = (x: Int) => x + 1
 xx: Int => Int = <function1>
 //给一个高阶函数,传递一个函数:
 scala> val newList = List(1,2,3).map { (x: Int) => x * 2 }
 newList: List[Int] = List(2, 4, 6)

这种场景还是挺常见的,匿名函数定义, 左边是参数 右边是函数实现体 (x: Int)=>{}

 

3.模式匹配(case语句)

scala> val x = 10; val y = 20
x: Int = 10
y: Int = 20

scala> val max = x > y match {
  case true => x
  case false => y
}
max: Int = 20

在模式匹配 match 和 try-catch 都用 “=>” 表示输出的结果或返回的值

 

4.By-Name Parameters(传名参数)

传名参数在函数调用前表达式不会被求值,而是会被包裹成一个匿名函数作为函数参数传递下去,例如参数类型为无参函数的参数就是传名参数。

 scala> def doubles(x: => Int) = {
     println("Now doubling " + x)
     x*2   
 }
 doubles: (x: => Int)Int
 //调用函数
 scala> doubles(3)
 Now doubling 3
 res2: Int = 6

 scala> def f(x: Int): Int = {
   println(s"Calling f($x)")
   x
 }
 f: (x: Int)Int
 //调用函数
 scala> doubles(f(3))
 Calling f(3)
 Now doubling 3
 Calling f(3)
 res9: Int = 6

对于函数doubles而言,它的参数x就是by-name的。如果调用doubles的时候,直接给个普通的值或者非函数变量。那么doubles的执行结果就跟普通的函数没有区别。但是当把一个返回值为Int类型的函数,例如f(2),传递给doubles的时候。那么f(2)会被先计算出返回值2,返回值2传入doubles参与运算。运算完成以后,f(2)会被doubles在执行以后,再调用一遍。

 

登录后即可回复 登录 | 注册
    
关注编程学问公众号