Scala中函数的使用

川长思鸟来 2022-09-03 10:19 190阅读 0赞

#### 函数的定义 ####

在Scala中定义函数需要使用 def 关键字，函数包括`函数名`、`参数`、`函数体`

Scala要求必须给出函数所有参数的类型，但是函数返回值的类型不是必须的，因为Scala可以自己根据函数体中的表达式推断出返回值类型。

函数中最后一行代码的返回值就是整个函数的返回值，不需要使用return，这一点与Java不同，java中函 数的返回值是必须要使用return的

下面来实现一个单行函数和多行函数:

*  单行函数

scala> def sayHello(name: String) = print("Hello, " + name)
    sayHello: (name: String)Unit
    
    scala> sayHello("Scala")
    Hello, Scala

*  多行函数

scala> :paste
    // Entering paste mode (ctrl-D to finish)
    
    def sayHello(name: String, age: Int) = { 
      println("My name is "+name+",age is "+age)
    age }
    
    // Exiting paste mode, now interpreting.
    
    sayHello: (name: String, age: Int)Int
    
    scala> sayHello("Scala",18)
    My name is Scala,age is 18
    res120: Int = 18

#### 函数的参数 ####

*  默认参数  
    在Scala中，有时候我们调用某些函数时，不希望给出参数的具体值，而是希望使用参数自身默认的值，此时就需要在定义函数时使用默认参数。  
    如果给出的参数不够，则会从左往右依次应用参数。

scala> def sayHello(fName: String, mName: String = "mid", lName: String = "last")
    sayHello: (fName: String, mName: String, lName: String)String
    
    scala> sayHello("zhang","san")
    res122: String = zhang san last

大家还记得Java中实现默认参数该怎么实现吗?  
假设是要设置name和age这两个参数的默认值  
java的写法:

public void sayHello(String name, int age) { 
    	if(name == null) { 
    		name = "defaultName"
    	}
      	if(age == 0) { 
        	age = 18
    	}
      	System.out.println("Hello, " + name + ", your age is " + age)
    }

*  带名参数  
    在调用函数时，也可以不按照函数定义的参数顺序来传递参数，而是使用带名参数的方式来传递。

scala> def sayHello(fName: String, mName: String = "mid", lName: String = "last")
    sayHello: (fName: String, mName: String, lName: String)String
    
    scala> sayHello(fName = "Mick", lName = "Tom", mName = "Jack")
    res127: String = Mick Jack Tom

*  可变参数  
    在Scala中，有时我们需要将函数定义为参数个数可变的形式，则此时可以使用变长参数来定义函数。

scala> :paste
    // Entering paste mode (ctrl-D to finish)
    
    def sum(nums: Int*) = { 
      var res = 0
      for (num <- nums) res += num
      res
    }
    
    // Exiting paste mode, now interpreting.
    
    sum: (nums: Int*)Int
    
    scala> sum(1,2,3,4,5)
    res129: Int = 15

#### 特殊的函数-过程 ####

在Scala中，定义函数时，如果函数体直接在花括号里面而没有使用=连接，则函数的返回值类型就是Unit，这样的函数称之为`过程`

过程通常用于不需要返回值的函数

过程还有一种写法，就是将函数的返回值类型显式定义为Unit

比较一下这四种写法的区别:

def sayHello(name: String) = "Hello, " + name
    def sayHello(name: String): String = "Hello, " + name
    def sayHello(name: String) {   "Hello, " + name }
    def sayHello(name: String): Unit = "Hello, " + name

前面两种写法的效果是一样的，都是函数  
后面两种写法的效果是一样的，都是过程

#### lazy ####

Scala提供了lazy特性，如果将一个变量声明为lazy，则只有在第一次使用该变量时，变量对应的表达式才会发生计算

什么场景下需要使用lazy特性呢? 这种特性对于特别耗时的操作特别有用，比如打开文件这个操作。

scala> import scala.io.Source._
    import scala.io.Source._
    
    scala> lazy val lines = fromFile("~\test.txt").mkString
    lines: String = <lazy>

即使`~\test.txt`文件不存在，代码也不会报错，只有变量使用时才会报错，这就是lazy这个特性。

scala> lines
    java.io.FileNotFoundException: ~\test.txt (file not found) 
    .......